73 lines
1.9 KiB
Go
73 lines
1.9 KiB
Go
package f2b
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"gitlab.com/hectorjsmith/fail2ban-prometheus-exporter/cfg"
|
|
"gitlab.com/hectorjsmith/fail2ban-prometheus-exporter/socket"
|
|
)
|
|
|
|
type Collector struct {
|
|
socketPath string
|
|
exporterVersion string
|
|
lastError error
|
|
socketConnectionErrorCount int
|
|
socketRequestErrorCount int
|
|
exitOnSocketConnError bool
|
|
}
|
|
|
|
func NewExporter(appSettings *cfg.AppSettings, exporterVersion string) *Collector {
|
|
log.Printf("reading fail2ban metrics socket file: %s", appSettings.Fail2BanSocketPath)
|
|
s, err := socket.ConnectToSocket(appSettings.Fail2BanSocketPath)
|
|
if err != nil {
|
|
log.Printf("error connecting to socket: %v", err)
|
|
} else {
|
|
version, err := s.GetServerVersion()
|
|
if err != nil {
|
|
log.Printf("error interacting with socket: %v", err)
|
|
} else {
|
|
log.Printf("successfully connected to fail2ban socket! fail2ban version: %s", version)
|
|
}
|
|
}
|
|
return &Collector{
|
|
socketPath: appSettings.Fail2BanSocketPath,
|
|
exporterVersion: exporterVersion,
|
|
lastError: nil,
|
|
socketConnectionErrorCount: 0,
|
|
socketRequestErrorCount: 0,
|
|
exitOnSocketConnError: appSettings.ExitOnSocketConnError,
|
|
}
|
|
}
|
|
|
|
func (c *Collector) Describe(ch chan<- *prometheus.Desc) {
|
|
ch <- metricServerUp
|
|
ch <- metricJailCount
|
|
ch <- metricJailFailedCurrent
|
|
ch <- metricJailFailedTotal
|
|
ch <- metricJailBannedCurrent
|
|
ch <- metricJailBannedTotal
|
|
ch <- metricErrorCount
|
|
}
|
|
|
|
func (c *Collector) Collect(ch chan<- prometheus.Metric) {
|
|
s, err := socket.ConnectToSocket(c.socketPath)
|
|
if err != nil {
|
|
log.Printf("error opening socket: %v", err)
|
|
c.socketConnectionErrorCount++
|
|
if c.exitOnSocketConnError {
|
|
os.Exit(1)
|
|
}
|
|
} else {
|
|
defer s.Close()
|
|
}
|
|
|
|
c.collectServerUpMetric(ch, s)
|
|
if err == nil && s != nil {
|
|
c.collectJailMetrics(ch, s)
|
|
c.collectVersionMetric(ch, s)
|
|
}
|
|
c.collectErrorCountMetric(ch)
|
|
}
|