mirror of
https://github.com/KevinMidboe/mktxp-no-cli.git
synced 2025-10-29 17:50:23 +00:00
80 lines
3.6 KiB
Python
80 lines
3.6 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.collector.base_collector import BaseCollector
|
|
from mktxp.flow.processor.output import BaseOutputProcessor
|
|
from mktxp.datasource.bgp_ds import BGPMetricsDataSource
|
|
|
|
|
|
class BGPCollector(BaseCollector):
|
|
'''BGP collector'''
|
|
@staticmethod
|
|
def collect(router_entry):
|
|
if not router_entry.config_entry.bgp:
|
|
return
|
|
|
|
bgp_labels = ['name', 'remote.address', 'remote.as', 'local.as', 'remote.afi', 'local.afi', 'remote.messages', 'remote.bytes', 'local.messages', 'local.bytes', 'prefix_count', 'established', 'uptime']
|
|
bgp_records = BGPMetricsDataSource.metric_records(router_entry, metric_labels=bgp_labels)
|
|
|
|
|
|
if bgp_records:
|
|
# translate records to appropriate values
|
|
translated_fields = ['established', 'uptime']
|
|
for bgp_record in bgp_records:
|
|
for translated_field in translated_fields:
|
|
value = bgp_record.get(translated_field, None)
|
|
if value:
|
|
bgp_record[translated_field] = BGPCollector._translated_values(translated_field, value)
|
|
|
|
session_id_labes = ['name', 'remote.address', 'remote.as', 'local.as', 'remote.afi', 'local.afi']
|
|
bgp_sessions_metrics = BaseCollector.info_collector('bgp_sessions_info', 'BGP sessions info', bgp_records, session_id_labes)
|
|
yield bgp_sessions_metrics
|
|
|
|
remote_messages_metrics = BaseCollector.counter_collector('bgp_remote_messages', 'Number of remote messages', bgp_records, 'remote.messages', session_id_labes)
|
|
yield remote_messages_metrics
|
|
|
|
|
|
local_messages_metrics = BaseCollector.counter_collector('bgp_local_messages', 'Number of local messages', bgp_records, 'local.messages', session_id_labes)
|
|
yield local_messages_metrics
|
|
|
|
|
|
remote_bytes_metrics = BaseCollector.counter_collector('bgp_remote_bytes', 'Number of remote bytes', bgp_records, 'remote.bytes', session_id_labes)
|
|
yield remote_bytes_metrics
|
|
|
|
|
|
local_bytes_metrics = BaseCollector.counter_collector('bgp_local_bytes', 'Number of local bytes', bgp_records, 'local.bytes', session_id_labes)
|
|
yield local_bytes_metrics
|
|
|
|
|
|
prefix_count_metrics = BaseCollector.gauge_collector('bgp_prefix_count', 'BGP prefix count', bgp_records, 'prefix_count', session_id_labes)
|
|
yield prefix_count_metrics
|
|
|
|
|
|
established_metrics = BaseCollector.gauge_collector('bgp_established', 'BGP established', bgp_records, 'established', session_id_labes)
|
|
yield established_metrics
|
|
|
|
|
|
uptime_metrics = BaseCollector.gauge_collector('bgp_uptime', 'BGP uptime in milliseconds', bgp_records, 'uptime', session_id_labes)
|
|
yield uptime_metrics
|
|
|
|
|
|
# Helpers
|
|
@staticmethod
|
|
def _translated_values(translated_field, value):
|
|
return {
|
|
'established': lambda value: '1' if value=='true' else '0',
|
|
'uptime': lambda value: BaseOutputProcessor.parse_timedelta_milliseconds(value)
|
|
}[translated_field](value)
|
|
|