From 2d248151b388291622dc212a40bd4cc741a080e7 Mon Sep 17 00:00:00 2001
From: Shreya Malviya <shreya.malviya@gmail.com>
Date: Tue, 9 Aug 2022 14:32:33 +0530
Subject: [PATCH] Island: Register IEventQueue instance in DI container
---
monkey/monkey_island/cc/services/initialize.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/monkey/monkey_island/cc/services/initialize.py b/monkey/monkey_island/cc/services/initialize.py
index f6ff6d309..eeb488b14 100644
--- a/monkey/monkey_island/cc/services/initialize.py
+++ b/monkey/monkey_island/cc/services/initialize.py
@@ -3,6 +3,7 @@ import logging
from pathlib import Path
from typing import Sequence
+from pubsub import pub
from pymongo import MongoClient
from common import DIContainer
@@ -13,6 +14,8 @@ from common.agent_configuration import (
)
from common.aws import AWSInstance
from common.common_consts.telem_categories import TelemCategoryEnum
+from common.event_queue.i_event_queue import IEventQueue
+from common.event_queue.pypubsub_event_queue import PyPubSubEventQueue
from common.utils.file_utils import get_binary_io_sha256_hash
from common.version import get_version
from monkey_island.cc import Version
@@ -71,6 +74,8 @@ def initialize_services(data_dir: Path) -> DIContainer:
ILockableEncryptor, RepositoryEncryptor(data_dir / REPOSITORY_KEY_FILE_NAME)
)
container.register_instance(Version, container.resolve(Version))
+ container.register_instance(IEventQueue, PyPubSubEventQueue(pub))
+
_register_repositories(container, data_dir)
_register_services(container)