Island: Register a thread-safe IAgentEventQueue

This commit is contained in:
Mike Salvatore 2022-09-29 12:21:28 -04:00
parent 5d893d64cd
commit 91375cdff2
1 changed files with 12 additions and 2 deletions

View File

@ -15,7 +15,11 @@ from common.agent_event_serializers import (
register_common_agent_event_serializers,
)
from common.aws import AWSInstance
from common.event_queue import IAgentEventQueue, PyPubSubAgentEventQueue
from common.event_queue import (
IAgentEventQueue,
LockingAgentEventQueueDecorator,
PyPubSubAgentEventQueue,
)
from common.utils.file_utils import get_binary_io_sha256_hash
from monkey_island.cc.event_queue import (
IIslandEventQueue,
@ -104,11 +108,17 @@ def _register_conventions(container: DIContainer):
def _register_event_queues(container: DIContainer):
container.register_instance(IAgentEventQueue, container.resolve(PyPubSubAgentEventQueue))
agent_event_queue = container.resolve(PyPubSubAgentEventQueue)
container.register_instance(IAgentEventQueue, _decorate_agent_event_queue(agent_event_queue))
island_event_queue = container.resolve(PyPubSubIslandEventQueue)
container.register_instance(IIslandEventQueue, _decorate_island_event_queue(island_event_queue))
def _decorate_agent_event_queue(agent_event_queue: IAgentEventQueue):
return LockingAgentEventQueueDecorator(agent_event_queue)
def _decorate_island_event_queue(island_event_queue: IIslandEventQueue):
return LockingIslandEventQueueDecorator(island_event_queue)