p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Do small readability changes in batching_telemetry_messenger.py

This commit is contained in:
VakarisZ 2021-06-29 12:09:26 +03:00
parent e34599779b
commit a454449cca
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -19,7 +19,7 @@ class BatchingTelemetryMessenger(ITelemetryMessenger):
def __init__(self, telemetry_messenger: ITelemetryMessenger, period=DEFAULT_PERIOD): def __init__(self, telemetry_messenger: ITelemetryMessenger, period=DEFAULT_PERIOD):
self._queue: queue.Queue[ITelem] = queue.Queue() self._queue: queue.Queue[ITelem] = queue.Queue()
self._thread = BatchingTelemetryMessenger._BatchingTelemetryMessengerThread( self._thread = self._BatchingTelemetryMessengerThread(
self._queue, telemetry_messenger, period self._queue, telemetry_messenger, period
) )
@ -32,8 +32,10 @@ class BatchingTelemetryMessenger(ITelemetryMessenger):
self._queue.put(telemetry) self._queue.put(telemetry)
class _BatchingTelemetryMessengerThread: class _BatchingTelemetryMessengerThread:
def __init__(self, queue: queue.Queue, telemetry_messenger: ITelemetryMessenger, period): def __init__(
self._queue: queue.Queue[ITelem] = queue self, telem_queue: queue.Queue, telemetry_messenger: ITelemetryMessenger, period: int
):
self._queue: queue.Queue[ITelem] = telem_queue
self._telemetry_messenger = telemetry_messenger self._telemetry_messenger = telemetry_messenger
self._period = period self._period = period