mirror of
https://github.com/KevinMidboe/mktxp-no-cli.git
synced 2025-10-29 17:50:23 +00:00
DS refactor, fixes/optimizations
This commit is contained in:
@@ -11,21 +11,24 @@
|
||||
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
## GNU General Public License for more details.
|
||||
|
||||
|
||||
from mktxp.cli.config.config import MKTXPConfigKeys
|
||||
from mktxp.collectors.base_collector import BaseCollector
|
||||
from mktxp.datasources.route_ds import RouteMetricsDataSource
|
||||
|
||||
|
||||
class RouteCollector(BaseCollector):
|
||||
''' IP Route Metrics collector
|
||||
'''
|
||||
@staticmethod
|
||||
def collect(router_metric):
|
||||
def collect(router_entry):
|
||||
route_labels = ['connect', 'dynamic', 'static', 'bgp', 'ospf']
|
||||
route_records = router_metric.route_records(route_labels)
|
||||
route_records = RouteMetricsDataSource.metric_records(router_entry, metric_labels = route_labels)
|
||||
if route_records:
|
||||
# compile total routes records
|
||||
total_routes = len(route_records)
|
||||
total_routes_records = [{ MKTXPConfigKeys.ROUTERBOARD_NAME: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_NAME],
|
||||
MKTXPConfigKeys.ROUTERBOARD_ADDRESS: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_ADDRESS],
|
||||
total_routes_records = [{ MKTXPConfigKeys.ROUTERBOARD_NAME: router_entry.router_id[MKTXPConfigKeys.ROUTERBOARD_NAME],
|
||||
MKTXPConfigKeys.ROUTERBOARD_ADDRESS: router_entry.router_id[MKTXPConfigKeys.ROUTERBOARD_ADDRESS],
|
||||
'count': total_routes
|
||||
}]
|
||||
total_routes_metrics = BaseCollector.gauge_collector('routes_total_routes', 'Overall number of routes in RIB', total_routes_records, 'count')
|
||||
@@ -40,8 +43,8 @@ class RouteCollector(BaseCollector):
|
||||
routes_per_protocol[route_label] += 1
|
||||
|
||||
# compile route-per-protocol records
|
||||
route_per_protocol_records = [{ MKTXPConfigKeys.ROUTERBOARD_NAME: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_NAME],
|
||||
MKTXPConfigKeys.ROUTERBOARD_ADDRESS: router_metric.router_id[MKTXPConfigKeys.ROUTERBOARD_ADDRESS],
|
||||
route_per_protocol_records = [{ MKTXPConfigKeys.ROUTERBOARD_NAME: router_entry.router_id[MKTXPConfigKeys.ROUTERBOARD_NAME],
|
||||
MKTXPConfigKeys.ROUTERBOARD_ADDRESS: router_entry.router_id[MKTXPConfigKeys.ROUTERBOARD_ADDRESS],
|
||||
'protocol': key, 'count': value} for key, value in routes_per_protocol.items()]
|
||||
|
||||
# yield route-per-protocol metrics
|
||||
|
||||
Reference in New Issue
Block a user