forked from p34709852/monkey
agent: Add BatchableTelemMixin
Adds an implementation as a mixin of the two methods specified by IBatchableTelem.
This commit is contained in:
parent
f2a940a4e0
commit
8e40e44263
|
@ -0,0 +1,22 @@
|
||||||
|
from typing import Iterable
|
||||||
|
|
||||||
|
from infection_monkey.telemetry.i_batchable_telem import IBatchableTelem
|
||||||
|
|
||||||
|
|
||||||
|
class BatchableTelemMixin:
|
||||||
|
"""
|
||||||
|
Implements the IBatchableTelem interface methods using a list.
|
||||||
|
"""
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _telemetry_entries(self):
|
||||||
|
if not hasattr(self, "_list"):
|
||||||
|
self._list = []
|
||||||
|
|
||||||
|
return self._list
|
||||||
|
|
||||||
|
def get_telemetry_entries(self) -> Iterable:
|
||||||
|
return self._telemetry_entries
|
||||||
|
|
||||||
|
def add_telemetry_to_batch(self, telemetry: IBatchableTelem):
|
||||||
|
self._telemetry_entries.extend(telemetry.get_telemetry_entries())
|
Loading…
Reference in New Issue