KevinMidboe/mktxp-no-cli
1
0
Fork 0
mirror of https://github.com/KevinMidboe/mktxp-no-cli.git synced 2025-12-08 20:38:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

connections stats collector / cmd output, remote dhcp resolver, fixes / optimizations

Browse Source
This commit is contained in:
Arseniy Kuznetsov
2023-02-04 20:48:51 +01:00
parent 18ddbe3311
commit 9a381d028c
21 changed files with 351 additions and 161 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
mktxp/datasource/connection_ds.py
View File

@@ -12,6 +12,7 @@
## GNU General Public License for more details.
from collections import namedtuple
from mktxp.datasource.base_ds import BaseDSProcessor
@@ -33,3 +34,44 @@ class IPConnectionDatasource:
except Exception as exc:
print(f'Error getting IP connection info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
return None
class IPConnectionStatsDatasource:
''' IP connections stats data provider
'''
@staticmethod
def metric_records(router_entry, *, metric_labels = None, add_router_id = True):
if metric_labels is None:
metric_labels = []
try:
connection_records = router_entry.api_connection.router_api().get_resource('/ip/firewall/connection/').call('print', \
{'proplist':'src-address,dst-address,protocol'})
# calculate number of connections per src-address
connections_per_src_address = {}
for connection_record in connection_records:
#address, port = (connection_record['src-address'].split(':') + [None])[:2]
address = connection_record['src-address'].split(':')[0]
destination = f"{connection_record.get('dst-address')}({connection_record.get('protocol')})"
count, destinations = 0, set()
if connections_per_src_address.get(address):
count, destinations = connections_per_src_address[address]
count += 1
destinations.add(destination)
connections_per_src_address[address] = ConnStatsEntry(count, destinations)
# compile connections-per-interface records
records = []
for key, entry in connections_per_src_address.items():
record = {'src_address': key, 'connection_count': entry.count, 'dst_addresses': ', '.join(entry.destinations)}
if add_router_id:
for router_key, router_value in router_entry.router_id.items():
record[router_key] = router_value
records.append(record)
return records
except Exception as exc:
print(f'Error getting IP connection stats info from router{router_entry.router_name}@{router_entry.config_entry.hostname}: {exc}')
return None
ConnStatsEntry = namedtuple('ConnStatsEntry', ['count', 'destinations'])