87 lines
3.7 KiB
Python
87 lines
3.7 KiB
Python
#!/usr/bin/python3
|
|
# -*- mode: python-mode; py-indent-tabs-mode: nil -*-
|
|
"""
|
|
/*
|
|
* Copyright (C) 2017 sysmocom s.f.m.c. GmbH
|
|
*
|
|
* All Rights Reserved
|
|
*
|
|
* 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 3 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.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
"""
|
|
|
|
from osmopy.osmo_ipa import Ctrl
|
|
import socket, argparse, sys, logging, csv
|
|
|
|
__version__ = "0.0.1" # bump this on every non-trivial change
|
|
|
|
def connect(host, port):
|
|
sck = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sck.setblocking(1)
|
|
sck.connect((host, port))
|
|
return sck
|
|
|
|
def get_var(sck, var):
|
|
(_, c) = Ctrl().cmd(var, None)
|
|
sck.send(c)
|
|
return Ctrl().parse_kv(sck.recv(4096))
|
|
|
|
def get_interval(group_name, group_counters, interval):
|
|
log.debug('Getting %s counter values: %s...' % (group_name, interval))
|
|
(_, c) = get_var(sock, 'rate_ctr.%s.%s' % (interval, group_name))
|
|
for ctr in c.split(';'):
|
|
if len(ctr):
|
|
(k, v) = ctr.split()
|
|
group_counters[k] = group_counters.get(k, (group_name,)) + (v,)
|
|
return len(group_counters)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
p = argparse.ArgumentParser(description='Dump rate counters into csv via Osmocom CTRL protocol.')
|
|
p.add_argument('-v', '--version', action='version', version=("%(prog)s v" + __version__))
|
|
p.add_argument('-p', '--port', type=int, default=4249, help="Port to use for CTRL interface, defaults to 4249")
|
|
p.add_argument('-c', '--ctrl', default='localhost', help="Adress to use for CTRL interface, defaults to localhost")
|
|
p.add_argument('-d', '--debug', action='store_true', help="Enable debug log")
|
|
p.add_argument('--header', action='store_true', help="Prepend column header to output")
|
|
p.add_argument('-o', '--output', nargs='?', type=argparse.FileType('w'), default=sys.stdout, help="Output file, defaults to stdout")
|
|
args = p.parse_args()
|
|
|
|
log = logging.getLogger('rate_ctr2csv')
|
|
log.setLevel(logging.DEBUG if args.debug else logging.INFO)
|
|
log.addHandler(logging.StreamHandler(sys.stderr))
|
|
|
|
log.info('Connecting to %s:%d...' % (args.ctrl, args.port))
|
|
sock = connect(args.ctrl, args.port)
|
|
|
|
log.info('Getting rate counter groups info...')
|
|
(_, g) = get_var(sock, 'rate_ctr.*')
|
|
|
|
w = csv.writer(args.output, dialect='unix')
|
|
total_groups = 0
|
|
total_rows = 0
|
|
|
|
if args.header:
|
|
w.writerow(['group', 'counter', 'absolute', 'second', 'minute', 'hour', 'day'])
|
|
|
|
for group in g.split(';'):
|
|
if len(group):
|
|
g_counters = {}
|
|
total_groups += 1
|
|
total_rows += list(map(lambda x: get_interval(group, g_counters, x), ('abs', 'per_sec', 'per_min', 'per_hour', 'per_day')))[0]
|
|
for (k, (gr, absolute, per_sec, per_min, per_hour, per_day)) in g_counters.items():
|
|
w.writerow([gr, k, absolute, per_sec, per_min, per_hour, per_day])
|
|
|
|
log.info('Completed: %d counters from %d groups received.' % (total_rows, total_groups))
|