Indirect evt_count updates via method call

so that we can stub it for the sentinel and not have a billion failing UTs
This commit is contained in:
Richard van der Hoff 2018-06-22 10:42:28 +01:00
parent b088aafcae
commit 0495fe0035
2 changed files with 12 additions and 1 deletions

View file

@ -91,6 +91,9 @@ class LoggingContext(object):
def add_database_scheduled(self, sched_sec):
pass
def record_event_fetch(self, event_count):
pass
def __nonzero__(self):
return False
__bool__ = __nonzero__ # python3
@ -247,6 +250,14 @@ class LoggingContext(object):
"""
self.db_sched_duration_sec += sched_sec
def record_event_fetch(self, event_count):
"""Record a number of events being fetched from the db
Args:
event_count (int): number of events being fetched
"""
self.evt_db_fetch_count += event_count
class LoggingContextFilter(logging.Filter):
"""Logging filter that adds values from the current logging context to each