From ad8b5608f8f48d11740099c021116338fcedae83 Mon Sep 17 00:00:00 2001
From: Ildar Valiullin <preved.911@gmail.com>
Date: Thu, 1 Sep 2022 12:30:06 +0300
Subject: [PATCH] extended /probe path metrics

Signed-off-by: Ildar Valiullin <preved.911@gmail.com>
---
 cmd/postgres_exporter/probe.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/cmd/postgres_exporter/probe.go b/cmd/postgres_exporter/probe.go
index 7b215431..6c9a46bf 100644
--- a/cmd/postgres_exporter/probe.go
+++ b/cmd/postgres_exporter/probe.go
@@ -78,6 +78,23 @@ func handleProbe(logger log.Logger) http.HandlerFunc {
 		registry.MustRegister(probeSuccessGauge)
 		registry.MustRegister(probeDurationGauge)
 
+		opts := []ExporterOpt{
+			DisableDefaultMetrics(*disableDefaultMetrics),
+			DisableSettingsMetrics(*disableSettingsMetrics),
+			AutoDiscoverDatabases(*autoDiscoverDatabases),
+			WithUserQueriesPath(*queriesPath),
+			WithConstantLabels(*constantLabelsList),
+			ExcludeDatabases(*excludeDatabases),
+			IncludeDatabases(*includeDatabases),
+		}
+
+		dsns := []string{dsn.GetConnectionString()}
+		exporter := NewExporter(dsns, opts...)
+		defer func() {
+			exporter.servers.Close()
+		}()
+		registry.MustRegister(exporter)
+
 		// Run the probe
 		pc, err := collector.NewProbeCollector(tl, registry, dsn)
 		if err != nil {