NetDev

Complexity

Total Complexity

8

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	poll()	0	6	1
B	check()	0	29	6
A	__init__()	0	15	1

# -*- coding: utf-8 -*-

__author__ = 'Kenny Freeman'
__email__ = 'kenny.freeman@gmail.com'
__license__ = "ISCL"
__docformat__ = 'reStructuredText'

import time
import traceback

import plumd
import plumd.plugins
from plumd.calc import Differential
from plumd.util import get_file_map, Filter


class NetDev(plumd.plugins.Reader):
    """Plugin to measure various kernel metrics from /proc."""
    defaults = {
        'poll.interval': 10,
        'proc_path': '/proc',
        'net_dev_re': "(virbr\d+)|(vnet\d+)", # not used atm
        'net_dev_devs': ["eth0", "eth1", "bond0", "bond1", "lo"],
        'net_dev_cols': ["rx_bytes", "rx_pkt", "rx_errs", "rx_drop",
                         "rx_fifo_errs", "rx_frame_errs", "rx_compressed",
                         "rx_mcast", "tx_bytes", "tx_pkt", "tx_errs", "tx_drop",
                         "tx_fifo_errs", "collissions", "carrier",
                         "tx_compressed"]
    }

    def __init__(self, log, config):
        """Plugin to measure network interface metrics from proc file net/dev.

        :param log: A logger
        :type log: logging.RootLogger
        :param config: a plumd.config.Conf configuration helper instance.
        :type config: plumd.config.Conf
        """
        super(NetDev, self).__init__(log, config)
        config.defaults(NetDev.defaults)
        self.proc_file = "{0}/net/dev".format(config.get('proc_path'))
        self.net_dev_cols = config.get('net_dev_cols')
        self.net_dev_devs = config.get('net_dev_devs')
        self.filter = Filter()
        self.calc = Differential()


    def poll(self):
        """Return network interface metrics from proc file net/dev.

        :rtype: ResultSet
        """
        return plumd.ResultSet(self.check())


    def check(self):
        """Return network interface metrics from proc file net/dev.

        :rtype: list
        """
        result = plumd.Result("net")
        dat = {}
        cols = self.net_dev_cols
        devs = self.net_dev_devs
        # read and process /proc/stat
        try:
            dat = get_file_map(self.proc_file, 0, 0)
        except Exception as e:
            tb = traceback.format_exc()
            self.log.error("proc_net_dev: exception: {0}: {1}".format(e, tb))
            return [result]
        ts = time.time()
        for key, val in dat.items():
            key = self.filter.process(key)
            if key not in devs:
                continue
            for mname in cols:
                if len(val) < 1:
                    break
                mval = int(val.popleft())
                mstr = "{0}.{1}".format(key, mname)
                dval = self.calc.per_second(mstr, mval, ts)
                result.add(plumd.Int(mstr, dval))
        return [result]