p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Island: Subscribe AgentSignalsService.on_terminate_agents_signal to 

TERMINATE_AGENTS events
This commit is contained in:
Ilija Lazoroski 2022-09-22 17:36:42 +02:00 committed by Mike Salvatore
parent f12e839878
commit 5eeee2a60d
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
monkey/monkey_island/cc/setup/island_event_handlers.py
View File

@ -15,6 +15,7 @@ from monkey_island.cc.repository import (
INodeRepository, INodeRepository,
ISimulationRepository, ISimulationRepository,
) )
from monkey_island.cc.services import AgentSignalsService
from monkey_island.cc.services.database import Database from monkey_island.cc.services.database import Database
@ -25,6 +26,7 @@ def setup_island_event_handlers(container: DIContainer):
_subscribe_reset_agent_configuration_events(island_event_queue, container) _subscribe_reset_agent_configuration_events(island_event_queue, container)
_subscribe_clear_simulation_data_events(island_event_queue, container) _subscribe_clear_simulation_data_events(island_event_queue, container)
_subscribe_set_island_mode_events(island_event_queue, container) _subscribe_set_island_mode_events(island_event_queue, container)
_subscribe_on_terminate_agents_signal(island_event_queue, container)
def _subscribe_agent_registration_events( def _subscribe_agent_registration_events(
@ -74,3 +76,13 @@ def _subscribe_set_island_mode_events(
simulation_repository = container.resolve(ISimulationRepository) simulation_repository = container.resolve(ISimulationRepository)
island_event_queue.subscribe(topic, simulation_repository.set_mode) island_event_queue.subscribe(topic, simulation_repository.set_mode)
def _subscribe_on_terminate_agents_signal(
island_event_queue: IIslandEventQueue, container: DIContainer
):
topic = IslandEventTopic.TERMINATE_AGENTS
agent_signals_service = container.resolve(AgentSignalsService)
island_event_queue.subscribe(topic, agent_signals_service.on_terminate_agents_signal)