Add additional tag support to CounterWrapper

Signed-off-by: James Mitchell <[email protected]>
patricknieves · Sep 30, 2017 · 6259ec8 · 6259ec8
1 parent 0e17c9b
commit 6259ec8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/validator/sawtooth_validator/metrics/wrappers.py b/validator/sawtooth_validator/metrics/wrappers.py
@@ -25,9 +25,11 @@ def gauge(self, name, default=0):
             ''.join([name, ',host=', platform.node()]),
             default=default)
 
-    def counter(self, name):
+    def counter(self, name, tags=None):
+        if not tags:
+            tags = []
         return self._registry.counter(
-            ''.join([name, ',host=', platform.node()]))
+            ','.join([name, 'host={}'.format(platform.node())] + tags))
 
     def timer(self, name, tags=None):
         if not tags: