mirror of
https://github.com/KevinMidboe/mktxp-no-cli.git
synced 2025-10-29 17:50:23 +00:00
81 lines
4.7 KiB
Python
81 lines
4.7 KiB
Python
# 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.utils.utils import parse_uptime
|
|
from mktxp.cli.config.config import MKTXPConfigKeys
|
|
from mktxp.collectors.base_collector import BaseCollector
|
|
from mktxp.router_metric import RouterMetric
|
|
|
|
|
|
class CapsmanCollector(BaseCollector):
|
|
''' CAPsMAN Metrics collector
|
|
'''
|
|
@staticmethod
|
|
def collect(router_metric):
|
|
remote_caps_labels = ['identity', 'version', 'base_mac', 'board', 'base_mac']
|
|
remote_caps_records = router_metric.capsman_remote_caps_records(remote_caps_labels)
|
|
if not remote_caps_records:
|
|
return range(0)
|
|
|
|
remote_caps_metrics = BaseCollector.info_collector('capsman_remote_caps', 'CAPsMAN remote caps', remote_caps_records, remote_caps_labels)
|
|
yield remote_caps_metrics
|
|
|
|
registration_labels = ['interface', 'ssid', 'mac_address', 'tx_rate', 'rx_rate', 'rx_signal', 'uptime', 'bytes']
|
|
registration_records = router_metric.capsman_registration_table_records(registration_labels)
|
|
if not registration_records:
|
|
return range(0)
|
|
|
|
# calculate number of registrations per interface
|
|
registration_per_interface = {}
|
|
for registration_record in registration_records:
|
|
registration_per_interface[registration_record['interface']] = registration_per_interface.get(registration_record['interface'], 0) + 1
|
|
# compile registrations-per-interface records
|
|
registration_per_interface_records = [{ MKTXPConfigKeys.ROUTERBOARD_NAME: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_NAME],
|
|
MKTXPConfigKeys.ROUTERBOARD_ADDRESS: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_ADDRESS],
|
|
'interface': key, 'count': value} for key, value in registration_per_interface.items()]
|
|
# yield registrations-per-interface metrics
|
|
registration_per_interface_metrics = BaseCollector.gauge_collector('capsman_registrations_count', 'Number of active registration per CAPsMAN interface', registration_per_interface_records, 'count', ['interface'])
|
|
yield registration_per_interface_metrics
|
|
|
|
# the client info metrics
|
|
if router_metric.router_entry.capsman_clients:
|
|
# translate / trim / augment registration records
|
|
dhcp_lease_labels = ['mac_address', 'host_name', 'comment']
|
|
dhcp_lease_records = router_metric.dhcp_lease_records(dhcp_lease_labels)
|
|
for registration_record in registration_records:
|
|
try:
|
|
dhcp_lease_record = next((dhcp_lease_record for dhcp_lease_record in dhcp_lease_records if dhcp_lease_record['mac_address']==registration_record['mac_address']))
|
|
registration_record['name'] = dhcp_lease_record.get('comment', dhcp_lease_record.get('host_name', dhcp_lease_record.get('mac_address')))
|
|
except StopIteration:
|
|
registration_record['name'] = f"{registration_record['mac_address']}: No DHCP registration"
|
|
|
|
# split out tx/rx bytes
|
|
registration_record['tx_bytes'] = registration_record['bytes'].split(',')[0]
|
|
registration_record['rx_bytes'] = registration_record['bytes'].split(',')[1]
|
|
del registration_record['bytes']
|
|
|
|
tx_byte_metrics = BaseCollector.counter_collector('capsman_clients_tx_bytes', 'Number of sent packet bytes', registration_records, 'tx_bytes', ['name'])
|
|
yield tx_byte_metrics
|
|
|
|
rx_byte_metrics = BaseCollector.counter_collector('capsman_clients_rx_bytes', 'Number of received packet bytes', registration_records, 'rx_bytes', ['name'])
|
|
yield rx_byte_metrics
|
|
|
|
signal_strength_metrics = BaseCollector.gauge_collector('capsman_clients_signal_strength', 'Client devices signal strength', registration_records, 'rx_signal', ['name'])
|
|
yield signal_strength_metrics
|
|
|
|
registration_metrics = BaseCollector.info_collector('capsman_clients_devices', 'Registered client devices info',
|
|
registration_records, ['name', 'rx_signal', 'ssid', 'tx_rate', 'rx_rate', 'interface', 'mac_address', 'uptime'])
|
|
yield registration_metrics
|
|
|
|
|