Added basic exporter features

Author: DerKnerd

.gitignore

@@ -10,3 +10,65 @@
 
 # Output of the go coverage tool, specifically when used with LiteIDE
 *.out
+### JetBrains template
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and WebStorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+

app.go

@@ -0,0 +1,35 @@
+package main
+
+import (
+	"./exporter"
+	"./utils"
+	"flag"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"log"
+	"net/http"
+)
+
+func main() {
+	flag.Parse()
+	utils.ParseConfig()
+
+	exp := exporter.New()
+
+	prometheus.MustRegister(exp)
+
+	listenAddress := utils.Config().Listen.Address
+
+	http.Handle(utils.Config().Listen.MetricsPath, promhttp.Handler())
+	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		http.Redirect(w, r, utils.Config().Listen.MetricsPath, http.StatusMovedPermanently)
+	})
+
+	log.Printf("Starting Raspberry PI exporter on %q", listenAddress)
+
+	if err := http.ListenAndServe(listenAddress, nil); err != nil {
+		log.Fatalf("Cannot start Raspberry PI exporter: %s", err)
+	} else {
+		log.Printf("Started Raspberry PI exporter on %q", listenAddress)
+	}
+}

collector/clock.go

@@ -0,0 +1,36 @@
+package collector
+
+import (
+	"../utils"
+	"github.com/prometheus/client_golang/prometheus"
+	"regexp"
+	"strconv"
+)
+
+const (
+	CoreClock string = "core"
+	EmmcClock string = "emmc"
+	ArmClock  string = "arm"
+)
+
+func getClock(desc *prometheus.Desc, device string) prometheus.Metric {
+	clock, err := utils.ExecuteVcGen("measure_clock", device)
+
+	if err != nil {
+		return prometheus.NewInvalidMetric(desc, err)
+	}
+
+	clock = regexp.MustCompile(`frequency\(\d*\)=|(\n)|(\r)`).ReplaceAllString(clock, "")
+
+	clockFloat, err := strconv.ParseFloat(clock, 64)
+
+	if err != nil {
+		return prometheus.NewInvalidMetric(desc, err)
+	}
+
+	return prometheus.MustNewConstMetric(
+		desc,
+		prometheus.GaugeValue,
+		clockFloat,
+	)
+}

collector/coretemp.go

@@ -0,0 +1,44 @@
+package collector
+
+import (
+	"../utils"
+	"github.com/prometheus/client_golang/prometheus"
+	"io/ioutil"
+	"regexp"
+	"strconv"
+	"strings"
+)
+
+func getCoreTemp(collector VcGenCmdCollector) prometheus.Metric {
+	coreTempFromSys, err := ioutil.ReadFile("/sys/class/thermal/thermal_zone0/temp")
+	var coreTemp string
+	method := "file"
+
+	if err != nil {
+		if coreTemp, err = utils.ExecuteVcGen("measure_temp"); err != nil {
+			return prometheus.NewInvalidMetric(collector.CoreTemp, err)
+		} else {
+			coreTemp = regexp.MustCompile(`(temp=)|('C)|(\n)|(\r)`).ReplaceAllString(coreTemp, "")
+			method = "vcgen"
+		}
+	} else {
+		coreTemp = string(coreTempFromSys)
+	}
+
+	coreTemp = strings.TrimSuffix(coreTemp, "\n")
+	coreTempFloat, err := strconv.ParseFloat(coreTemp, 64)
+
+	if err != nil {
+		return prometheus.NewInvalidMetric(collector.CoreTemp, err)
+	}
+
+	if method == "file" {
+		coreTempFloat /= 1000
+	}
+
+	return prometheus.MustNewConstMetric(
+		collector.CoreTemp,
+		prometheus.GaugeValue,
+		coreTempFloat,
+	)
+}

collector/vcgencmcollector.go

@@ -0,0 +1,102 @@
+package collector
+
+import (
+	"github.com/prometheus/client_golang/prometheus"
+)
+
+type VcGenCmdCollector struct {
+	CoreTemp                *prometheus.Desc
+	CoreVoltage             *prometheus.Desc
+	SdramControllerVoltage  *prometheus.Desc
+	SdramInputOutputVoltage *prometheus.Desc
+	SdramPhysicalVoltage    *prometheus.Desc
+	CoreClock               *prometheus.Desc
+	ArmClock                *prometheus.Desc
+	EmmcClock               *prometheus.Desc
+}
+
+func (VcGenCmdCollector) Describe(channel chan<- *prometheus.Desc) {
+	var collector = NewVcGenCmdCollector()
+
+	channel <- collector.CoreTemp
+	channel <- collector.CoreVoltage
+	channel <- collector.SdramControllerVoltage
+	channel <- collector.SdramInputOutputVoltage
+	channel <- collector.SdramPhysicalVoltage
+	channel <- collector.CoreClock
+	channel <- collector.ArmClock
+	channel <- collector.EmmcClock
+}
+
+func (VcGenCmdCollector) Collect(channel chan<- prometheus.Metric) {
+	var collector = NewVcGenCmdCollector()
+
+	channel <- getCoreTemp(*collector)
+	channel <- getVoltage(collector.CoreVoltage, CoreVoltage)
+	channel <- getVoltage(collector.SdramPhysicalVoltage, SdramPhysicalVoltage)
+	channel <- getVoltage(collector.SdramInputOutputVoltage, SdramInputOutputVoltage)
+	channel <- getVoltage(collector.SdramControllerVoltage, SdramControllerVoltage)
+	channel <- getClock(collector.CoreClock, CoreClock)
+	channel <- getClock(collector.ArmClock, ArmClock)
+	channel <- getClock(collector.EmmcClock, EmmcClock)
+}
+
+var _ prometheus.Collector = &VcGenCmdCollector{}
+
+func NewVcGenCmdCollector() *VcGenCmdCollector {
+	const (
+		subsystem = "vcgencmd"
+		namespace = "pi"
+	)
+
+	return &VcGenCmdCollector{
+		CoreTemp: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "core_temp"),
+			"Temperature of the SoC",
+			nil,
+			nil,
+		),
+		CoreVoltage: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "core_voltage"),
+			"Voltage of the CPU",
+			nil,
+			nil,
+		),
+		SdramControllerVoltage: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "sdram_controller_voltage"),
+			"Voltage of the SDRAM controller",
+			nil,
+			nil,
+		),
+		SdramInputOutputVoltage: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "sdram_input_output_voltage"),
+			"Voltage of the SDRAM IO",
+			nil,
+			nil,
+		),
+		SdramPhysicalVoltage: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "sdram_physical_voltage"),
+			"Voltage of the physical SDRAM",
+			nil,
+			nil,
+		),
+		CoreClock: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "core_clock"),
+			"Clock of the core in Hz",
+			nil,
+			nil,
+		),
+		ArmClock: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "arm_clock"),
+			"Clock of the ARM CPU in Hz",
+			nil,
+			nil,
+		),
+		EmmcClock: prometheus.NewDesc(
+			prometheus.BuildFQName(namespace, subsystem, "emmc_clock"),
+			"Clock of the external MMC in Hz",
+			nil,
+			nil,
+		),
+	}
+}

collector/voltage.go

@@ -0,0 +1,37 @@
+package collector
+
+import (
+	"../utils"
+	"github.com/prometheus/client_golang/prometheus"
+	"regexp"
+	"strconv"
+)
+
+const (
+	CoreVoltage             string = "core"
+	SdramControllerVoltage  string = "sdram_c"
+	SdramInputOutputVoltage string = "sdram_i"
+	SdramPhysicalVoltage    string = "sdram_p"
+)
+
+func getVoltage(desc *prometheus.Desc, device string) prometheus.Metric {
+	voltage, err := utils.ExecuteVcGen("measure_volts", device)
+
+	if err != nil {
+		return prometheus.NewInvalidMetric(desc, err)
+	}
+
+	voltage = regexp.MustCompile(`(volt=)|(V)|(\n)|(\r)`).ReplaceAllString(voltage, "")
+
+	voltageFloat, err := strconv.ParseFloat(voltage, 64)
+
+	if err != nil {
+		return prometheus.NewInvalidMetric(desc, err)
+	}
+
+	return prometheus.MustNewConstMetric(
+		desc,
+		prometheus.GaugeValue,
+		voltageFloat,
+	)
+}

exporter/raspberryexporter.go

@@ -0,0 +1,35 @@
+package exporter
+
+import (
+	"../collector"
+	"github.com/prometheus/client_golang/prometheus"
+	"sync"
+)
+
+type Exporter struct {
+	mutex      sync.Mutex
+	collectors []prometheus.Collector
+}
+
+func New() Exporter {
+	var exporter = Exporter{}
+	exporter.collectors = []prometheus.Collector{
+		collector.NewVcGenCmdCollector(),
+	}
+
+	return exporter
+}
+
+func (exporter Exporter) Describe(channel chan<- *prometheus.Desc) {
+	for _, cc := range exporter.collectors {
+		cc.Describe(channel)
+	}
+}
+
+func (exporter Exporter) Collect(channel chan<- prometheus.Metric) {
+	for _, c := range exporter.collectors {
+		c.Collect(channel)
+	}
+}
+
+var _ prometheus.Collector = &Exporter{}