diff --git a/superset/stats_logger.py b/superset/stats_logger.py index aaab8a11f19d6..e16ecbb5878c4 100644 --- a/superset/stats_logger.py +++ b/superset/stats_logger.py @@ -29,6 +29,9 @@ def decr(self, key): """Decrement a counter""" raise NotImplementedError() + def timing(self, key, value): + raise NotImplementedError() + def gauge(self, key): """Setup a gauge""" raise NotImplementedError() @@ -44,6 +47,11 @@ def decr(self, key): Fore.CYAN + '[stats_logger] (decr) ' + key + Style.RESET_ALL)) + def timing(self, key, value): + logging.debug(( + Fore.CYAN + '[stats_logger] (timing) {key} | {value} ' + + Style.RESET_ALL).format(**locals())) + def gauge(self, key, value): logging.debug(( Fore.CYAN + '[stats_logger] (gauge) ' @@ -63,6 +71,9 @@ def incr(self, key): def decr(self, key): self.client.decr(key) + def timing(self, key, value): + self.client.timing(key, value) + def gauge(self, key): # pylint: disable=no-value-for-parameter self.client.gauge(key)