From e35bfba308a1ed21226143c18feb4e11f5bec845 Mon Sep 17 00:00:00 2001
From: timifasubaa <30888507+timifasubaa@users.noreply.github.com>
Date: Mon, 10 Sep 2018 10:31:58 -0700
Subject: [PATCH] add timing to stats logger (#5844)

---
 superset/stats_logger.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

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)