/
litespeed_exporter.go
60 lines (51 loc) · 1.71 KB
/
litespeed_exporter.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
package main
import (
"net/http"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/prometheus/common/log"
"github.com/prometheus/common/version"
"gopkg.in/alecthomas/kingpin.v2"
"github.com/myokoo/litespeed_exporter/collector"
"github.com/myokoo/litespeed_exporter/pkg/rtreport"
)
var (
listenAddress = kingpin.Flag(
"web.listen-address",
"Listen address for web interface and telemetry.",
).Default(":9104").String()
metricPath = kingpin.Flag(
"web.telemetry-path",
"URL path under which to expose metrics.",
).Default("/metrics").String()
reportPath = kingpin.Flag(
"lsws.report-path",
"Filesystem path under which exist lsws real-time statistics reports.",
).Default(rtreport.DefaultReportPath).String()
)
func main() {
// Parse flags.
log.AddFlags(kingpin.CommandLine)
kingpin.Version(version.Print("litespeed_exporter"))
kingpin.HelpFlag.Short('h')
kingpin.Parse()
// landingPage contains the HTML served at '/'.
// TODO: Make this nicer and more informative.
var landingPage = []byte(`<html>
<head><title>LiteSpeed exporter</title></head>
<body>
<h1>LiteSpeed exporter</h1>
<p><a href='` + *metricPath + `'>Metrics</a></p>
</body>
</html>
`)
log.Infoln("Starting litespeed_exporter", version.Info())
log.Infoln("Build context", version.BuildContext())
log.Infoln("Listening on", *listenAddress)
exporter := collector.New(reportPath)
prometheus.MustRegister(exporter)
prometheus.MustRegister(version.NewCollector("litespeed_exporter"))
http.Handle(*metricPath, promhttp.Handler())
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { w.Write(landingPage) })
log.Fatal(http.ListenAndServe(*listenAddress, nil))
}