main.go

package main

import (
	"flag"
	"net/http"
	"os"
	"strconv"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/sirupsen/logrus"
	"github.com/skoef/gop1"
)

var listenAddr string
var verbose bool
var metricNamePrefix = "p1_"

var (
	registry                   = prometheus.NewRegistry()
	electricityUsageHighMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "usage_electricity_high",
		Help: "Electricity usage high tariff",
	})
	electricityUsageLowMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "usage_electricity_low",
		Help: "Electricity usage low tariff",
	})
	electricityReturnedHighMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "returned_electricity_high",
		Help: "Electricity returned high tariff",
	})
	electricityReturnedLowMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "returned_electricity_low",
		Help: "Electricity returned low tariff",
	})
	actualElectricityDeliveredMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "actual_electricity_delivered",
		Help: "Actual electricity power delivered to client",
	})
	actualElectricityGeneratedMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "actual_electricity_generated",
		Help: "Actual electricity power generated by client",
	})
	activeTarrifMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "active_tariff",
		Help: "Active tariff",
	})
	powerFailuresLongMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "power_failures_long",
		Help: "Power failures long",
	})
	powerFailuresShortMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "power_failures_short",
		Help: "Power failures short",
	})
	gasUsageMetric = prometheus.NewGauge(prometheus.GaugeOpts{
		Name: metricNamePrefix + "usage_gas",
		Help: "Gas usage",
	})
)

func init() {
	registry.MustRegister(electricityUsageHighMetric)
	registry.MustRegister(electricityUsageLowMetric)
	registry.MustRegister(electricityReturnedHighMetric)
	registry.MustRegister(electricityReturnedLowMetric)
	registry.MustRegister(actualElectricityDeliveredMetric)
	registry.MustRegister(actualElectricityGeneratedMetric)
	registry.MustRegister(activeTarrifMetric)
	registry.MustRegister(powerFailuresLongMetric)
	registry.MustRegister(powerFailuresShortMetric)
	registry.MustRegister(gasUsageMetric)
}

func floatValue(input string) (fval float64) {
	fval, _ = strconv.ParseFloat(input, 64)
	return
}

func main() {
	flag.StringVar(&listenAddr, "listen", "127.0.0.1:8888", "Listen address for HTTP metrics")
	flag.BoolVar(&verbose, "verbose", false, "Verbose output logging")
	flag.Parse()

	p1, err := gop1.New(gop1.P1Config{
		USBDevice: "/dev/ttyUSB0",
	})
	if err != nil {
		logrus.Errorln("Quitting because of error opening p1", err)
		os.Exit(1)
	}

	// Start
	p1.Start()

	logrus.SetFormatter(&logrus.TextFormatter{FullTimestamp: true})
	if verbose {
		logrus.SetLevel(logrus.DebugLevel)
	}

	go func() {
		for tgram := range p1.Incoming {
			for _, obj := range tgram.Objects {
				switch obj.Type {
				case gop1.OBISTypeElectricityDeliveredTariff1:
					electricityUsageLowMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityDeliveredTariff2:
					electricityUsageHighMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityGeneratedTariff1:
					electricityReturnedLowMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityGeneratedTariff2:
					electricityReturnedHighMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityDelivered:
					actualElectricityDeliveredMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityGenerated:
					actualElectricityGeneratedMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeNumberOfPowerFailures:
					powerFailuresShortMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeNumberOfLongPowerFailures:
					powerFailuresLongMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeElectricityTariffIndicator:
					activeTarrifMetric.Set(floatValue(obj.Values[0].Value))
				case gop1.OBISTypeGasDelivered:
					gasUsageMetric.Set(floatValue(obj.Values[0].Value))
				}
			}
		}
	}()

	logrus.Infoln("Start listening at", listenAddr)
	http.Handle("/metrics", promhttp.HandlerFor(registry, promhttp.HandlerOpts{}))
	logrus.Fatalln(http.ListenAndServe(listenAddr, nil))
}