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

agent: Rename _run_batch_thread -> _should_run_batch_thread

This commit is contained in:
Mike Salvatore 2021-06-28 11:33:06 -04:00
parent be6e76757d
commit 0a9c98f061
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
monkey/infection_monkey/telemetry/messengers/batching_telemetry_messenger.py
View File

@ -21,7 +21,7 @@ class BatchingTelemetryMessenger(ITelemetryMessenger):
self._telemetry_messenger = telemetry_messenger self._telemetry_messenger = telemetry_messenger
self._period = period self._period = period
self._run_batch_thread = True self._should_run_batch_thread = True
self._queue: queue.Queue[ITelem] = queue.Queue() self._queue: queue.Queue[ITelem] = queue.Queue()
# TODO: Create a "timer" or "countdown" class and inject an object instead of # TODO: Create a "timer" or "countdown" class and inject an object instead of
# using time.time() # using time.time()
@ -37,7 +37,7 @@ class BatchingTelemetryMessenger(ITelemetryMessenger):
self.stop() self.stop()
def stop(self): def stop(self):
self._run_batch_thread = False self._should_run_batch_thread = False
self._manage_telemetry_batches_thread.join() self._manage_telemetry_batches_thread.join()
def send_telemetry(self, telemetry: ITelem): def send_telemetry(self, telemetry: ITelem):
@ -46,7 +46,7 @@ class BatchingTelemetryMessenger(ITelemetryMessenger):
def _manage_telemetry_batches(self): def _manage_telemetry_batches(self):
self._reset() self._reset()
while self._run_batch_thread: while self._should_run_batch_thread:
try: try:
telemetry = self._queue.get(block=True, timeout=self._period / WAKES_PER_PERIOD) telemetry = self._queue.get(block=True, timeout=self._period / WAKES_PER_PERIOD)