add timing to stats logger (apache#5844)

lyft · Oct 12, 2018 · ee4ef74 · ee4ef74
1 parent c6d5191
commit ee4ef74
Showing 1 changed file with 11 additions and 0 deletions.
diff --git 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)