chore: (pre-release) add health check command

Author: alessio-form3

cmd/postgres_exporter/main.go

@@ -14,8 +14,10 @@
 package main
 
 import (
+	"errors"
 	"fmt"
 	"net/http"
+	"net"
 	"os"
 	"strings"
 
@@ -46,6 +48,7 @@ var (
 	autoDiscoverDatabases  = kingpin.Flag("auto-discover-databases", "Whether to discover the databases on a server dynamically. (DEPRECATED)").Default("false").Envar("PG_EXPORTER_AUTO_DISCOVER_DATABASES").Bool()
 	queriesPath            = kingpin.Flag("extend.query-path", "Path to custom queries to run. (DEPRECATED)").Default("").Envar("PG_EXPORTER_EXTEND_QUERY_PATH").String()
 	onlyDumpMaps           = kingpin.Flag("dumpmaps", "Do not run, simply dump the maps.").Bool()
+	onlyHealthCheck        = kingpin.Flag("healthcheck", "Do not run, just return if up and running.").Bool()
 	constantLabelsList     = kingpin.Flag("constantLabels", "A list of label=value separated by comma(,). (DEPRECATED)").Default("").Envar("PG_EXPORTER_CONSTANT_LABELS").String()
 	excludeDatabases       = kingpin.Flag("exclude-databases", "A list of databases to remove when autoDiscoverDatabases is enabled (DEPRECATED)").Default("").Envar("PG_EXPORTER_EXCLUDE_DATABASES").String()
 	includeDatabases       = kingpin.Flag("include-databases", "A list of databases to include when autoDiscoverDatabases is enabled (DEPRECATED)").Default("").Envar("PG_EXPORTER_INCLUDE_DATABASES").String()
@@ -81,6 +84,18 @@ func main() {
 		return
 	}
 
+	if *onlyHealthCheck {
+		healthy, err := runHealthCheck(webConfig)
+		if err != nil {
+			level.Error(logger).Log("msg", "error running health check", "err", err)
+		}
+		if healthy {
+			level.Info(logger).Log("msg", "health ok")
+			os.Exit(0)
+		}
+		os.Exit(1)
+	}
+
 	if err := c.ReloadConfig(*configFile, logger); err != nil {
 		// This is not fatal, but it means that auth must be provided for every dsn.
 		level.Warn(logger).Log("msg", "Error loading config", "err", err)
@@ -174,3 +189,30 @@ func main() {
 		os.Exit(1)
 	}
 }
+
+func runHealthCheck(webConfig *web.FlagConfig) (bool, error) {
+	if len(*webConfig.WebListenAddresses) == 0 {
+		return false, errors.New("no listen addresses to run the request to")
+	}
+	addr := (*webConfig.WebListenAddresses)[0]
+	host, port, err := net.SplitHostPort(addr)
+	if err != nil {
+		return false, err
+	}
+	if host == "" {
+		host = "localhost"
+	}
+	url := fmt.Sprintf("http://%s:%s/", host, port)
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return false, err
+	}
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return false, err
+	}
+
+	defer resp.Body.Close()
+	return resp.StatusCode == 200, nil
+}