2020-11-01 23:09:36 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2020-11-02 19:19:37 +01:00
|
|
|
"fmt"
|
|
|
|
"os"
|
|
|
|
"os/signal"
|
|
|
|
"syscall"
|
|
|
|
"time"
|
2020-11-02 18:18:08 +01:00
|
|
|
|
2020-11-02 19:19:37 +01:00
|
|
|
"github.com/ebrianne/adguard-exporter/config"
|
|
|
|
"github.com/ebrianne/adguard-exporter/internal/adguard"
|
|
|
|
"github.com/ebrianne/adguard-exporter/internal/metrics"
|
|
|
|
"github.com/ebrianne/adguard-exporter/internal/server"
|
2020-11-01 23:09:36 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2020-11-02 19:19:37 +01:00
|
|
|
name = "adguard-exporter"
|
2020-11-01 23:09:36 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2020-11-02 19:19:37 +01:00
|
|
|
s *server.Server
|
2020-11-01 23:09:36 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2020-11-02 19:19:37 +01:00
|
|
|
conf := config.Load()
|
2020-11-02 18:18:08 +01:00
|
|
|
|
2020-11-02 19:19:37 +01:00
|
|
|
metrics.Init()
|
2020-11-01 23:09:36 +01:00
|
|
|
|
2020-11-04 17:10:50 +01:00
|
|
|
initAdguardClient(conf.AdguardProtocol, conf.AdguardHostname, conf.AdguardUsername, conf.AdguardPassword, conf.Interval, conf.LogLimit)
|
|
|
|
initHttpServer(conf.ServerPort)
|
2020-11-01 23:09:36 +01:00
|
|
|
|
2020-11-02 19:19:37 +01:00
|
|
|
handleExitSignal()
|
2020-11-02 18:18:08 +01:00
|
|
|
}
|
2020-11-01 23:09:36 +01:00
|
|
|
|
2020-11-04 17:10:50 +01:00
|
|
|
func initAdguardClient(protocol, hostname string, username, password string, interval time.Duration, logLimit string) {
|
|
|
|
client := adguard.NewClient(protocol, hostname, username, password, interval, logLimit)
|
2020-11-02 19:19:37 +01:00
|
|
|
go client.Scrape()
|
2020-11-01 23:09:36 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func initHttpServer(port string) {
|
2020-11-02 19:19:37 +01:00
|
|
|
s = server.NewServer(port)
|
|
|
|
go s.ListenAndServe()
|
2020-11-01 23:09:36 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func handleExitSignal() {
|
2020-11-02 19:19:37 +01:00
|
|
|
stop := make(chan os.Signal, 1)
|
|
|
|
signal.Notify(stop, os.Interrupt, syscall.SIGTERM)
|
2020-11-01 23:09:36 +01:00
|
|
|
|
2020-11-02 19:19:37 +01:00
|
|
|
<-stop
|
2020-11-01 23:09:36 +01:00
|
|
|
|
2020-11-02 19:19:37 +01:00
|
|
|
s.Stop()
|
|
|
|
fmt.Println(fmt.Sprintf("\n%s HTTP server stopped", name))
|
2020-11-01 23:09:36 +01:00
|
|
|
}
|