diff options
| -rwxr-xr-x | 3rd_party/ovs_pmd_stats/ovs_pmd_stats.py | 109 |
1 files changed, 109 insertions, 0 deletions
diff --git a/3rd_party/ovs_pmd_stats/ovs_pmd_stats.py b/3rd_party/ovs_pmd_stats/ovs_pmd_stats.py new file mode 100755 index 00000000..fc6045b9 --- /dev/null +++ b/3rd_party/ovs_pmd_stats/ovs_pmd_stats.py @@ -0,0 +1,109 @@ +#!/usr/bin/env python +# +# Copyright(c) 2017 Intel Corporation. All rights reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +# Authors: +# Roman Korynkevych <romanx.korynkevych@intel.com> + +import socket +import argparse +import json +import logging + +HOSTNAME = socket.gethostname() +PROG_NAME = 'ovs_pmd_stats' +TYPE = 'counter' + +MAIN_THREAD = 'main thread' +PMD_THREAD = 'pmd thread' + +REQUEST_MESSAGE = '{"id":0,"method":"dpif-netdev/pmd-stats-show","params":[]}' +RESPONSE_MESSAGE_TIMEOUT = 1.0 + +# Setup arguments +parser = argparse.ArgumentParser(prog=PROG_NAME) +parser.add_argument('--socket-pid-file', required=True, help='ovs-vswitchd.pid file location') +args = parser.parse_args() + +try: + fp = open(args.socket_pid_file, 'r') + pid = fp.readline() + fp.close() +except IOError as e: + logging.error('I/O error({}): {}'.format(e.errno, e.strerror)) + raise SystemExit() +except: + logging.error('Unexpected error:', sys.exc_info()[0]) + raise SystemExit() + +server_address = args.socket_pid_file.replace('.pid', '.{}.ctl'.format(pid.strip())) + +# open unix socket to ovs-vswitch +sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) +try: + sock.connect(server_address) +except socket.error as msg: + logging.error('Socket address: {} Error: {}'.format(server_address, msg)) + raise SystemExit() + +# set timeout +sock.settimeout(RESPONSE_MESSAGE_TIMEOUT) + +# send request +sock.sendall(REQUEST_MESSAGE) + +# listen for respnse message +rdata = '' +while True: + try: + rdata += sock.recv(4096) + + if rdata.count('{') == rdata.count('}'): + break + except socket.timeout: + logging.error('Response message has not been received in {} sec.'.format(RESPONSE_MESSAGE_TIMEOUT)) + raise SystemExit() + except socket.error as e: + logging.error('Error received while reading: {}'.format(e.strerror)) + raise SystemExit() + +# parse the message +try: + s = json.loads(rdata, strict=False) +except ValueError as e: + logging.error('Failed to parse JSON response: {}'.format(e.strerror)) + raise SystemExit() + +# check for key string presence in the string +if 'result' not in s or 'id' not in s or 'error' not in s: + logging.error("One of the keys: ['id'], ['result'], ['error'] is missed in the response") + logging.error('Msg: {}'.format(s)) + raise SystemExit() + +array = s['result'].replace('\t', '').splitlines() + +# submit metrics in collectd format +plugin_instance = '' +for el in array: + if MAIN_THREAD in el or PMD_THREAD in el: + plugin_instance = el[:-1].replace(' ', '_') + else: + type_instance = el.split(':')[0].replace(' ', "_") + value = el.split(':')[1].split(' ')[0] + print('PUTVAL %s/%s-%s/%s-%s N:%s' % (HOSTNAME, PROG_NAME, plugin_instance, TYPE, type_instance, value)) + +# close socket +sock.close() + |