-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
97 lines (78 loc) · 2.24 KB
/
main.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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
package main
import (
"flag"
"fmt"
"io/ioutil"
"net/http"
"os"
"path/filepath"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
log "github.com/sirupsen/logrus"
yaml "gopkg.in/yaml.v2"
)
const version string = "0.1"
type Config struct {
Nodes []string `yaml:"nodes,omitempty"`
}
var (
showVersion = flag.Bool("version", false, "Print version information.")
listenAddress = flag.String("listen-address", ":9330", "Address on which to expose metrics.")
metricsPath = flag.String("path", "/metrics", "Path under which to expose metrics.")
configFile = flag.String("config-file", "config.yml", "Path to config file")
config *Config
)
func init() {
flag.Usage = func() {
fmt.Println("Usage: leontp_exporter [ ... ]\n\nParameters:")
fmt.Println()
flag.PrintDefaults()
}
}
func main() {
flag.Parse()
if *showVersion {
printVersion()
os.Exit(0)
}
filename, _ := filepath.Abs(*configFile)
yamlFile, err := ioutil.ReadFile(filename)
if err != nil {
log.Fatal("Can't read config file")
}
err = yaml.Unmarshal(yamlFile, &config)
if err != nil {
log.Fatal("Can't read names file")
}
startServer()
}
func printVersion() {
fmt.Println("leontp_exporter")
fmt.Printf("Version: %s\n", version)
}
func startServer() {
log.Infof("Starting LeoNTP exporter (Version: %s)\n", version)
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(`<html>
<head><title>LeoNTP Exporter (Version ` + version + `)</title></head>
<body>
<h1>LeoNTP Exporter</h1>
<p><a href="` + *metricsPath + `">Metrics</a></p>
<h2>More information:</h2>
<p><a href="https://github.com/lwlcom/leontp_exporter">github.com/lwlcom/leontp_exporter</a></p>
</body>
</html>`))
})
http.HandleFunc(*metricsPath, handleMetricsRequest)
log.Infof("Listening for %s on %s\n", *metricsPath, *listenAddress)
log.Fatal(http.ListenAndServe(*listenAddress, nil))
}
func handleMetricsRequest(w http.ResponseWriter, r *http.Request) {
reg := prometheus.NewRegistry()
reg.MustRegister(&leontpCollector{})
l := log.New()
l.Level = log.ErrorLevel
promhttp.HandlerFor(reg, promhttp.HandlerOpts{
ErrorLog: l,
ErrorHandling: promhttp.ContinueOnError}).ServeHTTP(w, r)
}