Files
mktxp-no-cli/mktxp/cli/output/conn_stats_out.py

50 lines
2.0 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.flow.processor.output import BaseOutputProcessor
from mktxp.datasource.connection_ds import IPConnectionStatsDatasource
class ConnectionsStatsOutput:
''' Connections Stats Output
'''
@staticmethod
def clients_summary(router_entry):
connection_records = IPConnectionStatsDatasource.metric_records(router_entry, add_router_id = False)
if not connection_records:
print('No connection stats records')
return
conn_cnt = 0
output_records = []
for registration_record in sorted(connection_records, key = lambda rt_record: rt_record['connection_count'], reverse=True):
BaseOutputProcessor.resolve_dhcp(router_entry, registration_record, id_key = 'src_address', resolve_address = False)
output_records.append(registration_record)
conn_cnt += registration_record['connection_count']
output_records_cnt = 0
output_entry = BaseOutputProcessor.OutputConnStatsEntry
output_table = BaseOutputProcessor.output_table(output_entry)
for record in output_records:
output_table.add_row(output_entry(**record))
output_table.add_row(output_entry())
output_records_cnt += 1
print (output_table.draw())
print(f'Distinct source addresses: {output_records_cnt}')
print(f'Total open connections: {conn_cnt}', '\n')