Add wlan station and interface metrics collection. (#14)

* Fix newlines.
* Added wlan metrics.
* Fix WlanSTA collector - return on errors.

collector/wlansta_collector.go

@@ -0,0 +1,111 @@
+package collector
+
+import (
+	"strconv"
+	"strings"
+
+	"github.com/prometheus/client_golang/prometheus"
+	log "github.com/sirupsen/logrus"
+	"gopkg.in/routeros.v2/proto"
+)
+
+type wlanSTACollector struct {
+	props        []string
+	descriptions map[string]*prometheus.Desc
+}
+
+func newWlanSTACollector() routerOSCollector {
+	c := &wlanSTACollector{}
+	c.init()
+	return c
+}
+
+func (c *wlanSTACollector) init() {
+	c.props = []string{"interface", "mac-address", "signal-to-noise", "signal-strength-ch0", "packets", "bytes", "frames"}
+	labelNames := []string{"name", "address", "interface", "mac_address"}
+	c.descriptions = make(map[string]*prometheus.Desc)
+	for _, p := range c.props[:len(c.props)-3] {
+		c.descriptions[p] = descriptionForPropertyName("wlan_station", p, labelNames)
+	}
+	for _, p := range c.props[len(c.props)-3:] {
+		c.descriptions["tx_"+p] = descriptionForPropertyName("wlan_station", "tx_"+p, labelNames)
+		c.descriptions["rx_"+p] = descriptionForPropertyName("wlan_station", "rx_"+p, labelNames)
+	}
+}
+
+func (c *wlanSTACollector) describe(ch chan<- *prometheus.Desc) {
+	for _, d := range c.descriptions {
+		ch <- d
+	}
+}
+
+func (c *wlanSTACollector) collect(ctx *collectorContext) error {
+	stats, err := c.fetch(ctx)
+	if err != nil {
+		return err
+	}
+
+	for _, re := range stats {
+		c.collectForStat(re, ctx)
+	}
+
+	return nil
+}
+
+func (c *wlanSTACollector) fetch(ctx *collectorContext) ([]*proto.Sentence, error) {
+	reply, err := ctx.client.Run("/interface/wireless/registration-table/print", "=.proplist="+strings.Join(c.props, ","))
+	if err != nil {
+		log.WithFields(log.Fields{
+			"device": ctx.device.Name,
+			"error":  err,
+		}).Error("error fetching wlan station metrics")
+		return nil, err
+	}
+
+	return reply.Re, nil
+}
+
+func (c *wlanSTACollector) collectForStat(re *proto.Sentence, ctx *collectorContext) {
+	iface := re.Map["interface"]
+	mac := re.Map["mac-address"]
+
+	for _, p := range c.props[2 : len(c.props)-3] {
+		c.collectMetricForProperty(p, iface, mac, re, ctx)
+	}
+	for _, p := range c.props[len(c.props)-3:] {
+		c.collectMetricForTXRXCounters(p, iface, mac, re, ctx)
+	}
+}
+
+func (c *wlanSTACollector) collectMetricForProperty(property, iface, mac string, re *proto.Sentence, ctx *collectorContext) {
+	v, err := strconv.ParseFloat(re.Map[property], 64)
+	if err != nil {
+		log.WithFields(log.Fields{
+			"device":   ctx.device.Name,
+			"property": property,
+			"value":    re.Map[property],
+			"error":    err,
+		}).Error("error parsing wlan station metric value")
+		return
+	}
+
+	desc := c.descriptions[property]
+	ctx.ch <- prometheus.MustNewConstMetric(desc, prometheus.GaugeValue, v, ctx.device.Name, ctx.device.Address, iface, mac)
+}
+
+func (c *wlanSTACollector) collectMetricForTXRXCounters(property, iface, mac string, re *proto.Sentence, ctx *collectorContext) {
+	tx, rx, err := splitStringToFloats(re.Map[property])
+	if err != nil {
+		log.WithFields(log.Fields{
+			"device":   ctx.device.Name,
+			"property": property,
+			"value":    re.Map[property],
+			"error":    err,
+		}).Error("error parsing wlan station metric value")
+		return
+	}
+	desc_tx := c.descriptions["tx_"+property]
+	desc_rx := c.descriptions["rx_"+property]
+	ctx.ch <- prometheus.MustNewConstMetric(desc_tx, prometheus.CounterValue, tx, ctx.device.Name, ctx.device.Address, iface, mac)
+	ctx.ch <- prometheus.MustNewConstMetric(desc_rx, prometheus.CounterValue, rx, ctx.device.Name, ctx.device.Address, iface, mac)
+}