KevinMidboe/mktxp-no-cli
1
0
Fork 0
mirror of https://github.com/KevinMidboe/mktxp-no-cli.git synced 2025-10-29 17:50:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

simple queue, parallel router fetch as_completed , config autoupdate fixes

Browse Source
This commit is contained in:
Arseniy Kuznetsov
2022-12-30 19:11:07 +01:00
parent 27c71ee843
commit 660dd33369
6 changed files with 80 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
mktxp/datasource/queue_ds.py
View File

@@ -15,31 +15,37 @@
from mktxp.datasource.base_ds import BaseDSProcessor
class QueueTreeMetricsDataSource:
''' Queue Tree Metrics data provider
class QueueMetricsDataSource:
''' Queue Metrics data provider
'''
@staticmethod
def metric_records(router_entry, *, metric_labels = None):
def metric_records(router_entry, *, metric_labels = None, kind = 'tree'):
if metric_labels is None:
metric_labels = []
try:
queue_tree_records = router_entry.api_connection.router_api().get_resource('/queue/tree/').get()
return BaseDSProcessor.trimmed_records(router_entry, router_records = queue_tree_records, metric_labels = metric_labels)
queue_records = router_entry.api_connection.router_api().get_resource(f'/queue/{kind}/').get()
queue_records = BaseDSProcessor.trimmed_records(router_entry, router_records = queue_records, metric_labels = metric_labels)
except Exception as exc:
print(f'Error getting system resource info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
return None
if kind == 'tree':
return queue_records
# simple queue records need splitting upload/download values
splitted_queue_records = []
for queue_record in queue_records:
splitted_queue_record = {}
for key, value in queue_record.items():
split_values = value.split('/')
if split_values and len(split_values) > 1:
splitted_queue_record[f'{key}_up'] = split_values[0]
splitted_queue_record[f'{key}_down'] = split_values[1]
else:
splitted_queue_record[key] = value
splitted_queue_records.append(splitted_queue_record)
return splitted_queue_records
class SimpleQueueMetricsDataSource:
''' Simple Queue Metrics data provider
'''
@staticmethod
def metric_records(router_entry, *, metric_labels = None):
if metric_labels is None:
metric_labels = []
try:
simple_queue_records = router_entry.api_connection.router_api().get_resource('/queue/simple/').get()
return BaseDSProcessor.trimmed_records(router_entry, router_records = simple_queue_records, metric_labels = metric_labels)
except Exception as exc:
print(f'Error getting system resource info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
return None