MySql.poll()

Complexity

Conditions

11

Size

Total Lines

50

Duplication

Lines	0
Ratio	0 %

Importance

Changes	3
Bugs	0	Features	0

# -*- coding: utf-8 -*-
    def poll(self):
        """Poll for MySQL metrics.

        :rtype: ResultSet
        """
        result = plumd.Result("mysql")
        # add db name as prefix to all metrics
        db = self.config.get('db')
        # currently records selected values from status
        # future versions to support calculated values + others
        try:
            self.dbcur.execute("SHOW GLOBAL STATUS")
        except Exception as e:
            self.log.error("MySql: exception getting status: {0}".format(e))
            if not self.connect():
                return plumd.ResultSet([result])

        try:
            status = dict(self.dbcur.fetchall())
        except Exception as e:
            self.log.error("MySql: exception fetching status: {0}".format(e))
            self.connect()
            return plumd.ResultSet([result])

        # all values recorded as is - todo: gauges and rates
        del_metrics = {}
        for prefix, metrics in self.record.items():
            mfmt_str = "{0}.{1}.{2}"
            for metric in metrics:
                try:
                    val = status[metric]
                    mname = mfmt_str.format(db, prefix, metric)
                    result.add(plumd.Float(mname, val))
                except (KeyError, ValueError) as e:
                    msg = "MySql: disabling unknown/invalid metric: {0}: {1}"
                    self.log.warn(msg.format(metric, e))
                    if prefix not in del_metrics:
                        del_metrics[prefix] = deque()
                    del_metrics[prefix].append(metric)

        # remove missing/unknown metrics - do outside of above for loop 
        # as we can't change self.record in the loop
        for pfx, metrics in del_metrics.items():
            for metric in metrics:
                self.record[pfx].remove(metric)

        if self.record_table_status:
            self.table_status(result)

        return plumd.ResultSet([result])