DS refactor, fixes/optimizations

2025-10-29 17:50:23 +00:00 · 2021-02-06 15:23:55 +01:00
parent 02b418937a
commit 3689386af0
33 changed files with 59 additions and 59 deletions
--- a/mktxp/collector/monitor_collector.py
+++ b/mktxp/collector/monitor_collector.py
@@ -0,0 +1,77 @@
+# 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 mktxp.collector.base_collector import BaseCollector
+from mktxp.cli.output.base_out import BaseOutputProcessor
+from mktxp.datasource.interface_ds import InterfaceMonitorMetricsDataSource
+
+
+class MonitorCollector(BaseCollector):
+    ''' Ethernet Interface Monitor Metrics collector
+    '''    
+    @staticmethod
+    def collect(router_entry):
+        monitor_labels = ('status', 'rate', 'full_duplex', 'name')
+        monitor_records = InterfaceMonitorMetricsDataSource.metric_records(router_entry, metric_labels = monitor_labels, include_comments = True)   
+        if monitor_records:
+            # translate records to appropriate values
+            for monitor_record in monitor_records:
+                for monitor_label in monitor_labels:
+                    value = monitor_record.get(monitor_label, None)    
+                    if value:            
+                        monitor_record[monitor_label] = MonitorCollector._translated_values(monitor_label, value)
+
+            monitor_status_metrics = BaseCollector.gauge_collector('interface_status', 'Current interface link status', monitor_records, 'status', ['name'])
+            yield monitor_status_metrics
+
+            # limit records according to the relevant metrics
+            rate_records = [monitor_record for monitor_record in monitor_records if monitor_record.get('rate', None)]
+            monitor_rates_metrics = BaseCollector.gauge_collector('interface_rate', 'Actual interface connection data rate', rate_records, 'rate', ['name'])
+            yield monitor_rates_metrics
+
+            full_duplex_records = [monitor_record for monitor_record in monitor_records if monitor_record.get('full_duplex', None)]
+            monitor_rates_metrics = BaseCollector.gauge_collector('interface_full_duplex', 'Full duplex data transmission', full_duplex_records, 'full_duplex', ['name'])
+            yield monitor_rates_metrics
+
+    # Helpers
+    @staticmethod
+    def _translated_values(monitor_label, value):
+        return {
+                'status': lambda value: '1' if value=='link-ok' else '0',
+                'rate': lambda value: MonitorCollector._rates(value),
+                'full_duplex': lambda value: '1' if value=='true' else '0',
+                'name': lambda value: value
+                }[monitor_label](value)
+
+    @staticmethod
+    def _rates(rate_option):
+        # according mikrotik docs, an interface rate should be one of these
+        rate_value =  {
+                '10Mbps': '10',
+                '100Mbps': '100',
+                '1Gbps': '1000',
+                '2.5Gbps': '2500',
+                '5Gbps': '5000',
+                '10Gbps': '10000',
+                '40Gbps': '40000'
+                }.get(rate_option, None)
+        if rate_value:
+            return rate_value
+        
+        # ...or just calculate in case it's not
+        return BaseOutputProcessor.parse_interface_rate(rate_option)
+
+
+
+