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

Agent: Use threading.Event instead of flag in send_all_events_to_island.py

This commit is contained in:
Shreya Malviya 2022-09-14 12:29:07 +05:30
parent a561195508
commit 5152b9a3cc
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/infection_monkey/send_all_events_to_island.py
View File

@ -48,17 +48,16 @@ class send_all_events_to_island:
self._server_address = server_address self._server_address = server_address
self._time_period = time_period self._time_period = time_period
self._should_run_batch_and_send_thread = True self._stop_batch_and_send_thread = threading.Event()
def start(self): def start(self):
self._should_run_batch_and_send_thread = True
self._batch_and_send_thread = threading.Thread( self._batch_and_send_thread = threading.Thread(
name="SendEventsToIslandInBatchesThread", target=self._manage_event_batches name="SendEventsToIslandInBatchesThread", target=self._manage_event_batches
) )
self._batch_and_send_thread.start() self._batch_and_send_thread.start()
def _manage_event_batches(self): def _manage_event_batches(self):
while self._should_run_batch_and_send_thread: while not self._stop_batch_and_send_thread.is_set():
self._send_events_to_island() self._send_events_to_island()
sleep(self._time_period) sleep(self._time_period)
@ -90,5 +89,5 @@ class send_all_events_to_island:
self._send_events_to_island() self._send_events_to_island()
def stop(self): def stop(self):
self._should_run_batch_and_send_thread = False self._stop_batch_and_send_thread.set()
self._batch_and_send_thread.join() self._batch_and_send_thread.join()