集中式pg

Author: lupx

cmd/postgres_exporter/main.go

@@ -14,8 +14,13 @@
 package main
 
 import (
+	//"context"
+	"fmt"
 	"net/http"
 	"os"
+	//"strconv"
+	//"strings"
+	//"time"
 
 	"github.com/go-kit/log"
 	"github.com/go-kit/log/level"
@@ -28,6 +33,7 @@ import (
 	"github.com/prometheus/exporter-toolkit/web"
 	webflag "github.com/prometheus/exporter-toolkit/web/kingpinflag"
 	"gopkg.in/alecthomas/kingpin.v2"
+	"net/url"
 )
 
 var (
@@ -61,6 +67,63 @@ const (
 	serverLabelName = "server"
 )
 
+
+func newHandler() http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+
+		v := r.URL.Query()
+		//params := v["collect[]"]
+		insname := v.Get("target")
+		dbname := v.Get("dbname")
+		user :=v.Get("user")
+		password :=v.Get("password")
+		ui := url.UserPassword(user, password).String()
+		dsn := fmt.Sprintf("postgresql://" + ui + "@" + insname+"/"+dbname+"?sslmode=disable")
+		// Use request context for cancellation when connection gets closed.
+		level.Info(logger).Log(dsn)
+
+		if len(dsn) == 0 {
+			level.Error(logger).Log("msg", "Couldn't find environment variables describing the datasource to use")
+			//dsn, err := getDataSources()
+			//if err != nil {
+			//	level.Error(logger).Log("msg", "Failed reading data sources", "err", err.Error())
+			//	os.Exit(1)
+			//}
+		}
+
+		opts := []ExporterOpt{
+			DisableDefaultMetrics(*disableDefaultMetrics),
+			DisableSettingsMetrics(*disableSettingsMetrics),
+			AutoDiscoverDatabases(*autoDiscoverDatabases),
+			WithUserQueriesPath(*queriesPath),
+			WithConstantLabels(*constantLabelsList),
+			ExcludeDatabases(*excludeDatabases),
+			IncludeDatabases(*includeDatabases),
+		}
+
+		exporter := NewExporter([]string{dsn}, opts...)
+		defer func() {
+			exporter.servers.Close()
+		}()
+
+		prometheus.MustRegister(version.NewCollector(exporterName))
+
+		prometheus.MustRegister(exporter)
+
+		pe, err := collector.NewPostgresCollector(logger, []string{dsn})
+		if err != nil {
+			level.Error(logger).Log("msg", "Failed to create PostgresCollector", "err", err.Error())
+			os.Exit(1)
+		}
+		prometheus.MustRegister(pe)
+
+		// Delegate http serving to Prometheus client library, which will call collector.Collect.
+		h := promhttp.HandlerFor(prometheus.DefaultGatherer, promhttp.HandlerOpts{})
+		h.ServeHTTP(w, r)
+	}
+}
+
+
 func main() {
 	kingpin.Version(version.Print(exporterName))
 	promlogConfig := &promlog.Config{}
@@ -85,50 +148,16 @@ func main() {
 		return
 	}
 
-	dsn, err := getDataSources()
-	if err != nil {
-		level.Error(logger).Log("msg", "Failed reading data sources", "err", err.Error())
-		os.Exit(1)
-	}
-
-	if len(dsn) == 0 {
-		level.Error(logger).Log("msg", "Couldn't find environment variables describing the datasource to use")
-		os.Exit(1)
-	}
-
-	opts := []ExporterOpt{
-		DisableDefaultMetrics(*disableDefaultMetrics),
-		DisableSettingsMetrics(*disableSettingsMetrics),
-		AutoDiscoverDatabases(*autoDiscoverDatabases),
-		WithUserQueriesPath(*queriesPath),
-		WithConstantLabels(*constantLabelsList),
-		ExcludeDatabases(*excludeDatabases),
-		IncludeDatabases(*includeDatabases),
-	}
-
-	exporter := NewExporter(dsn, opts...)
-	defer func() {
-		exporter.servers.Close()
-	}()
-
-	prometheus.MustRegister(version.NewCollector(exporterName))
-
-	prometheus.MustRegister(exporter)
-
-	pe, err := collector.NewPostgresCollector(logger, dsn)
-	if err != nil {
-		level.Error(logger).Log("msg", "Failed to create PostgresCollector", "err", err.Error())
-		os.Exit(1)
-	}
-	prometheus.MustRegister(pe)
 
-	http.Handle(*metricPath, promhttp.Handler())
+	//http.Handle(*metricPath, promhttp.Handler())
+	handlerFunc := newHandler()
+	http.Handle(*metricPath, promhttp.InstrumentMetricHandler(prometheus.DefaultRegisterer, handlerFunc))
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Set("Content-Type", "text/html; charset=UTF-8") // nolint: errcheck
 		w.Write(landingPage)                                       // nolint: errcheck
 	})
 
-	level.Info(logger).Log("msg", "Listening on address", "address", *listenAddress)
+	level.Info(logger).Log("msg", "@lupuxiao Listening on address", "address", *listenAddress)
 	srv := &http.Server{Addr: *listenAddress}
 	if err := web.ListenAndServe(srv, *webConfig, logger); err != nil {
 		level.Error(logger).Log("msg", "Error running HTTP server", "err", err)

cmd/postgres_exporter/postgres_exporter.exe

@@ -12,5 +12,6 @@ require (
 	github.com/prometheus/exporter-toolkit v0.7.1
 	gopkg.in/alecthomas/kingpin.v2 v2.2.6
 	gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c
+	gopkg.in/ini.v1 v1.66.4 // indirect
 	gopkg.in/yaml.v2 v2.4.0
 )

go.mod

@@ -469,6 +469,8 @@ gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
 gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
+gopkg.in/ini.v1 v1.66.4 h1:SsAcf+mM7mRZo2nJNGt8mZCjG8ZRaNGMURJw7BsIST4=
+gopkg.in/ini.v1 v1.66.4/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=
 gopkg.in/yaml.v2 v2.2.1/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=