133 lines
3.8 KiB
Go
133 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"github.com/prometheus/common/log"
|
|
|
|
"mystrom-exporter/pkg/mystrom"
|
|
"mystrom-exporter/pkg/version"
|
|
)
|
|
|
|
type switchReport struct {
|
|
Power float64 `json:"power"`
|
|
WattPerSec float64 `json:"Ws"`
|
|
Relay bool `json:"relay"`
|
|
Temperature float64 `json:"temperature"`
|
|
}
|
|
|
|
const namespace = "mystrom"
|
|
|
|
var (
|
|
listenAddress = flag.String("web.listen-address", ":9452",
|
|
"Address to listen on")
|
|
metricsPath = flag.String("web.metrics-path", "/metrics",
|
|
"Path under which to expose exporters own metrics")
|
|
devicePath = flag.String("web.device-path", "/device",
|
|
"Path under which the metrics of the devices are fetched")
|
|
showVersion = flag.Bool("version", false,
|
|
"Show version information.")
|
|
)
|
|
var (
|
|
mystromDurationCounterVec *prometheus.CounterVec
|
|
mystromRequestsCounterVec *prometheus.CounterVec
|
|
)
|
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
// -- show version information
|
|
if *showVersion {
|
|
v, err := version.Print("mystrom_exporter")
|
|
if err != nil {
|
|
log.Fatalf("Failed to print version information: %#v", err)
|
|
}
|
|
|
|
fmt.Fprintln(os.Stdout, v)
|
|
os.Exit(0)
|
|
}
|
|
|
|
// -- create a new registry for the exporter telemetry
|
|
telemetryRegistry := prometheus.NewRegistry()
|
|
telemetryRegistry.MustRegister(prometheus.NewGoCollector())
|
|
telemetryRegistry.MustRegister(prometheus.NewProcessCollector(prometheus.ProcessCollectorOpts{}))
|
|
|
|
mystromDurationCounterVec = prometheus.NewCounterVec(prometheus.CounterOpts{
|
|
Name: "mystrom_request_duration_seconds_total",
|
|
Help: "Total duration of mystrom successful requests by target in seconds",
|
|
}, []string{"target"})
|
|
telemetryRegistry.MustRegister(mystromDurationCounterVec)
|
|
|
|
// -- make the build information is available through a metric
|
|
buildInfo := prometheus.NewGaugeVec(
|
|
prometheus.GaugeOpts{
|
|
Namespace: "scripts",
|
|
Name: "build_info",
|
|
Help: "A metric with a constant '1' value labeled by build information.",
|
|
},
|
|
[]string{"version", "revision", "branch", "goversion", "builddate", "builduser"},
|
|
)
|
|
buildInfo.WithLabelValues(version.Version, version.Revision, version.Branch, version.GoVersion, version.BuildDate, version.BuildUser).Set(1)
|
|
telemetryRegistry.MustRegister(buildInfo)
|
|
|
|
exporter := mystrom.NewExporter()
|
|
// prometheus.MustRegister(exporter)
|
|
|
|
router := http.NewServeMux()
|
|
router.Handle(*metricsPath, promhttp.HandlerFor(telemetryRegistry, promhttp.HandlerOpts{}))
|
|
router.Handle(*devicePath,
|
|
http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
scrapeHandler(exporter, w, r)
|
|
}),
|
|
)
|
|
router.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte(`<html>
|
|
<head><title>myStrom switch report Exporter</title></head>
|
|
<body>
|
|
<h1>myStrom Exporter</h1>
|
|
<p><a href='` + *metricsPath + `'>Metrics</a></p>
|
|
</body>
|
|
</html>`))
|
|
})
|
|
log.Infoln("Listening on address " + *listenAddress)
|
|
log.Fatal(http.ListenAndServe(*listenAddress, router))
|
|
}
|
|
|
|
func scrapeHandler(e *mystrom.Exporter, w http.ResponseWriter, r *http.Request) {
|
|
target := r.URL.Query().Get("target")
|
|
if target == "" {
|
|
http.Error(w, "'target' parameter must be specified", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
log.Infof("got scrape request for target '%v'", target)
|
|
|
|
start := time.Now()
|
|
gatherer, err := e.Scrape(target)
|
|
duration := time.Since(start).Seconds()
|
|
if err != nil {
|
|
if strings.Contains(fmt.Sprintf("%v", err), "unable to connect with target") {
|
|
} else if strings.Contains(fmt.Sprintf("%v", err), "i/o timeout") {
|
|
}
|
|
http.Error(
|
|
w,
|
|
fmt.Sprintf("failed to scrape target '%v': %v", target, err),
|
|
http.StatusInternalServerError,
|
|
)
|
|
log.Error(err)
|
|
return
|
|
}
|
|
mystromDurationCounterVec.WithLabelValues(target).Add(duration)
|
|
|
|
promhttp.HandlerFor(gatherer, promhttp.HandlerOpts{}).ServeHTTP(w, r)
|
|
|
|
}
|