DS refactor, fixes/optimizations

2025-12-08 20:38:48 +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/resource_collector.py
+++ b/mktxp/collector/resource_collector.py
@@ -0,0 +1,71 @@
+# 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.system_resource_ds import SystemResourceMetricsDataSource
+
+
+class SystemResourceCollector(BaseCollector):
+    ''' System Resource Metrics collector
+    '''        
+    @staticmethod
+    def collect(router_entry):
+        resource_labels = ['uptime', 'version', 'free_memory', 'total_memory', 
+                           'cpu', 'cpu_count', 'cpu_frequency', 'cpu_load', 
+                           'free_hdd_space', 'total_hdd_space', 
+                           'architecture_name', 'board_name']
+                                   
+        resource_records = SystemResourceMetricsDataSource.metric_records(router_entry, metric_labels = resource_labels)   
+        if resource_records:
+            # translate records to appropriate values
+            translated_fields = ['uptime']        
+            for resource_record in resource_records:
+                for translated_field in translated_fields:
+                    value = resource_record.get(translated_field, None)    
+                    if value:            
+                        resource_record[translated_field] = SystemResourceCollector._translated_values(translated_field, value)
+
+            uptime_metrics = BaseCollector.gauge_collector('system_uptime', 'Time interval since boot-up', resource_records, 'uptime', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield uptime_metrics
+
+            free_memory_metrics = BaseCollector.gauge_collector('system_free_memory', 'Unused amount of RAM', resource_records, 'free_memory', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield free_memory_metrics
+
+            total_memory_metrics = BaseCollector.gauge_collector('system_total_memory', 'Amount of installed RAM', resource_records, 'total_memory', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield total_memory_metrics
+
+            free_hdd_metrics = BaseCollector.gauge_collector('system_free_hdd_space', 'Free space on hard drive or NAND', resource_records, 'free_hdd_space', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield free_hdd_metrics
+
+            total_hdd_metrics = BaseCollector.gauge_collector('system_total_hdd_space', 'Size of the hard drive or NAND', resource_records, 'total_hdd_space', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield total_hdd_metrics
+
+            cpu_load_metrics = BaseCollector.gauge_collector('system_cpu_load', 'Percentage of used CPU resources', resource_records, 'cpu_load', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield cpu_load_metrics
+
+            cpu_count_metrics = BaseCollector.gauge_collector('system_cpu_count', 'Number of CPUs present on the system', resource_records, 'cpu_count', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield cpu_count_metrics
+
+            cpu_frequency_metrics = BaseCollector.gauge_collector('system_cpu_frequency', 'Current CPU frequency', resource_records, 'cpu_frequency', ['version', 'board_name', 'cpu', 'architecture_name'])
+            yield cpu_frequency_metrics
+
+
+    # Helpers
+    @staticmethod
+    def _translated_values(translated_field, value):
+        return {
+                'uptime': lambda value: BaseOutputProcessor.parse_timedelta_seconds(value)
+                }[translated_field](value)
+