0b40e5de82
Add dependencies on `sqlite` to allow connecting to the fail2ban database. Add a new `db` module to handle all the database connections and data queries used to generate metrics. Export a new metric for the total number of bad IPs stored in the fail2ban database.
60 lines
1.3 KiB
Go
60 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
fail2banDb "fail2ban-prometheus-exporter/db"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
"log"
|
|
"net/http"
|
|
)
|
|
|
|
const namespace = "fail2ban"
|
|
|
|
var (
|
|
db = fail2banDb.MustConnectToDb("fail2ban.sqlite3")
|
|
metricUp = prometheus.NewDesc(
|
|
prometheus.BuildFQName(namespace, "", "up"),
|
|
"Was the last fail2ban query successful.",
|
|
nil, nil,
|
|
)
|
|
metricBadIpTotal = prometheus.NewDesc(
|
|
prometheus.BuildFQName(namespace, "", "badip_total"),
|
|
"Total number of bad IPs stored in the database.",
|
|
nil, nil,
|
|
)
|
|
)
|
|
|
|
type Exporter struct {
|
|
}
|
|
|
|
func (e *Exporter) Describe(ch chan<- *prometheus.Desc) {
|
|
ch <- metricUp
|
|
ch <- metricBadIpTotal
|
|
}
|
|
|
|
func (e *Exporter) Collect(ch chan<- prometheus.Metric) {
|
|
ch <- prometheus.MustNewConstMetric(
|
|
metricUp, prometheus.GaugeValue, 1,
|
|
)
|
|
ch <- *collectTotalBadIpMetric()
|
|
}
|
|
|
|
func collectTotalBadIpMetric() *prometheus.Metric {
|
|
count, _ := db.CountTotalBadIps()
|
|
metric := prometheus.MustNewConstMetric(
|
|
metricBadIpTotal, prometheus.GaugeValue, float64(count),
|
|
)
|
|
return &metric
|
|
}
|
|
|
|
func main() {
|
|
log.Print("starting fail2ban exporter")
|
|
|
|
exporter := &Exporter{}
|
|
prometheus.MustRegister(exporter)
|
|
|
|
http.Handle("/metrics", promhttp.Handler())
|
|
log.Fatal(http.ListenAndServe(":9101", nil))
|
|
}
|