package main import ( "fmt" "os" "os/signal" "syscall" "time" "github.com/eko/pihole-exporter/config" "github.com/eko/pihole-exporter/internal/metrics" "github.com/eko/pihole-exporter/internal/pihole" "github.com/eko/pihole-exporter/internal/server" ) const ( name = "pihole-exporter" ) var ( s *server.Server ) func main() { conf := config.Load() metrics.Init() initPiHoleClient(conf.PIHoleProtocol, conf.PIHoleHostname, conf.PIHolePassword, conf.PIHoleApiToken, conf.Interval) initHttpServer(conf.Port) handleExitSignal() } func initPiHoleClient(protocol, hostname, password, apiToken string, interval time.Duration) { client := pihole.NewClient(protocol, hostname, password, apiToken, interval) go client.Scrape() } func initHttpServer(port string) { s = server.NewServer(port) go s.ListenAndServe() } func handleExitSignal() { stop := make(chan os.Signal, 1) signal.Notify(stop, os.Interrupt, syscall.SIGTERM) <-stop s.Stop() fmt.Println(fmt.Sprintf("\n%s HTTP server stopped", name)) }