-
Notifications
You must be signed in to change notification settings - Fork 16
/
main.go
executable file
·104 lines (90 loc) · 2.88 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
98
99
100
101
102
103
104
package main
import (
"net/http"
"github.com/jenningsloy318/panos_exporter/collector"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
"github.com/prometheus/common/log"
kingpin "gopkg.in/alecthomas/kingpin.v2"
)
var (
configFile = kingpin.Flag(
"config.file",
"Path to configuration file.",
).String()
listenAddress = kingpin.Flag(
"web.listen-address",
"Address to listen on for web interface and telemetry.",
).Default(":9654").String()
sc = &SafeConfig{
C: &Config{},
}
reloadCh chan chan error
)
// define new http handleer
func metricsHandler() http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
registry := prometheus.NewRegistry()
ctx := r.Context()
target := r.URL.Query().Get("target")
if target == "" {
http.Error(w, "'target' parameter must be specified", 400)
return
}
log.Infof("Scraping target %s", target)
var deviceConfig *DeviceConfig
var err error
if deviceConfig, err = sc.DeviceConfigForTarget(target); err != nil {
log.Errorf("Error getting credentialfor target %s, error: %s", target, err)
return
}
collector := collector.NewPanosCollector(ctx, target, deviceConfig.Username, deviceConfig.Password)
registry.MustRegister(collector)
gatherers := prometheus.Gatherers{
prometheus.DefaultGatherer,
registry,
}
// Delegate http serving to Prometheus client library, which will call collector.Collect.
h := promhttp.HandlerFor(gatherers, promhttp.HandlerOpts{})
h.ServeHTTP(w, r)
}
}
var Version string
var BuildRevision string
var BuildBranch string
var BuildTime string
var BuildHost string
func init() {
log.Infof("panos_exporter version %s, build reversion %s, build branch %s, build at %s on host %s", Version, BuildRevision, BuildBranch, BuildTime, BuildHost)
}
func main() {
log.AddFlags(kingpin.CommandLine)
kingpin.HelpFlag.Short('h')
kingpin.Parse()
log.Infoln("Starting panos_exporter")
if err := sc.ReloadConfig(*configFile); err != nil {
log.Fatalf("Error parsing config file: %s", err)
}
http.Handle("/panos", metricsHandler()) // Regular metrics endpoint for panos metrics.
http.Handle("/metrics", promhttp.Handler()) // for metrics endpoint for itself metrics
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(`<html>
<head>
<title>Panos Exporter</title>
</head>
<body>
<h1>Panos Exporter</h1>
<form action="/panos">
<label>Target:</label> <input type="text" name="target" placeholder="X.X.X.X" value="1.2.3.4"><br>
<input type="submit" value="Submit">
</form>
<p><a href="/metrics">Local metrics</a></p>
</body>
</html>`))
})
log.Infof("Listening on %s", *listenAddress)
err := http.ListenAndServe(*listenAddress, nil)
if err != nil {
log.Fatal(err)
}
}