p15693087
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests 1 Projects Releases Wiki Activity

Island: Register IEventQueue instance in DI container

This commit is contained in:
Shreya Malviya 2022-08-09 14:32:33 +05:30
parent d190f220f7
commit 2d248151b3
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monkey/monkey_island/cc/services/initialize.py
View File

@ -3,6 +3,7 @@ import logging
from pathlib import Path from pathlib import Path
from typing import Sequence from typing import Sequence
from pubsub import pub
from pymongo import MongoClient from pymongo import MongoClient
from common import DIContainer from common import DIContainer
@ -13,6 +14,8 @@ from common.agent_configuration import (
) )
from common.aws import AWSInstance from common.aws import AWSInstance
from common.common_consts.telem_categories import TelemCategoryEnum 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.utils.file_utils import get_binary_io_sha256_hash
from common.version import get_version from common.version import get_version
from monkey_island.cc import 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) ILockableEncryptor, RepositoryEncryptor(data_dir / REPOSITORY_KEY_FILE_NAME)
) )
container.register_instance(Version, container.resolve(Version)) container.register_instance(Version, container.resolve(Version))
container.register_instance(IEventQueue, PyPubSubEventQueue(pub))
_register_repositories(container, data_dir) _register_repositories(container, data_dir)
_register_services(container) _register_services(container)