DS refactor, fixes/optimizations

mktxp/collector/base_collector.py

@@ -0,0 +1,59 @@
+# coding=utf8
+## Copyright (c) 2020 Arseniy Kuznetsov
+##
+## This program is free software; you can redistribute it and/or
+## modify it under the terms of the GNU General Public License
+## as published by the Free Software Foundation; either version 2
+## of the License, or (at your option) any later version.
+##
+## This program is distributed in the hope that it will be useful,
+## but WITHOUT ANY WARRANTY; without even the implied warranty of
+## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+## GNU General Public License for more details.
+
+
+from prometheus_client.core import GaugeMetricFamily, CounterMetricFamily, InfoMetricFamily
+from mktxp.cli.config.config import MKTXPConfigKeys
+     
+     
+class BaseCollector:
+    ''' Base Collector methods
+        For use by custom collector
+    '''
+    @staticmethod
+    def info_collector(name, decription, router_records, metric_labels=[]):
+        BaseCollector._add_id_labels(metric_labels)
+        collector = InfoMetricFamily(f'mktxp_{name}', decription)
+
+        for router_record in router_records:
+            label_values = {label: router_record.get(label) if router_record.get(label) else '' for label in metric_labels}
+            collector.add_metric(metric_labels, label_values)
+        return collector
+
+    @staticmethod
+    def counter_collector(name, decription, router_records, metric_key, metric_labels=[]):
+        BaseCollector._add_id_labels(metric_labels)
+        collector = CounterMetricFamily(f'mktxp_{name}', decription, labels=metric_labels)
+
+        for router_record in router_records:           
+            label_values = [router_record.get(label) for label in metric_labels]
+            collector.add_metric(label_values, router_record.get(metric_key, 0))
+        return collector
+
+    @staticmethod
+    def gauge_collector(name, decription, router_records, metric_key, metric_labels=[], add_id_labels = True):
+        if add_id_labels:
+            BaseCollector._add_id_labels(metric_labels)
+        collector = GaugeMetricFamily(f'mktxp_{name}', decription, labels=metric_labels)
+
+        for router_record in router_records:       
+            label_values = [router_record.get(label) for label in metric_labels]
+            collector.add_metric(label_values, router_record.get(metric_key, 0))
+        return collector
+
+
+    # Helpers
+    @staticmethod
+    def _add_id_labels(metric_labels):
+        metric_labels.append(MKTXPConfigKeys.ROUTERBOARD_NAME)
+        metric_labels.append(MKTXPConfigKeys.ROUTERBOARD_ADDRESS)