Merge pull request #5 from fly-apps/add-pg-exporter

Add PG exporter to dockerfile

Author: davissp14

Dockerfile

@@ -10,6 +10,7 @@ RUN CGO_ENABLED=0 GOOS=linux go build -v -o /fly/bin/flyadmin ./cmd/flyadmin
 RUN CGO_ENABLED=0 GOOS=linux go build -v -o /fly/bin/start ./cmd/start
 COPY ./bin/* /fly/bin/
 
+FROM wrouesnel/postgres_exporter:latest AS postgres_exporter
 FROM postgres:${PG_VERSION}
 ENV PGDATA=/data/pg_data
 ARG VERSION 
@@ -22,8 +23,11 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
     ca-certificates iproute2 consul curl bash dnsutils vim procps jq \
     && apt autoremove -y
 
+COPY --from=postgres_exporter /postgres_exporter /usr/local/bin/
 COPY --from=0 /fly/bin/* /usr/local/bin
 
+ADD /config/* /fly/
+
 EXPOSE 5432
 
 

cmd/start/main.go

@@ -56,9 +56,26 @@ func main() {
 		}
 	}()
 
+	node, err := flypg.NewNode()
+	if err != nil {
+		PanicHandler(err)
+	}
+
 	svisor := supervisor.New("flypg", 5*time.Minute)
 	svisor.AddProcess("flypg", "gosu postgres postgres -D /data/postgres/")
 
+	exporterEnv := map[string]string{
+		"DATA_SOURCE_URI":                      fmt.Sprintf("[%s]:%d/postgres?sslmode=disable", node.PrivateIP, node.PGPort),
+		"DATA_SOURCE_USER":                     node.SUCredentials.Username,
+		"DATA_SOURCE_PASS":                     node.SUCredentials.Password,
+		"PG_EXPORTER_EXCLUDE_DATABASE":         "template0,template1",
+		"PG_EXPORTER_DISABLE_SETTINGS_METRICS": "true",
+		"PG_EXPORTER_AUTO_DISCOVER_DATABASES":  "true",
+		"PG_EXPORTER_EXTEND_QUERY_PATH":        "/fly/queries.yaml",
+	}
+
+	svisor.AddProcess("exporter", "postgres_exporter", supervisor.WithEnv(exporterEnv), supervisor.WithRestart(0, 1*time.Second))
+
 	svisor.StopOnSignal(syscall.SIGINT, syscall.SIGTERM)
 	svisor.StartHttpListener()
 

config/queries.yaml

@@ -0,0 +1,11 @@
+pg_database:
+  query: "SELECT pg_database.datname, pg_database_size(pg_database.datname) as size_bytes FROM pg_database"
+  master: true
+  cache_seconds: 30
+  metrics:
+    - datname:
+        usage: "LABEL"
+        description: "Name of the database"
+    - size_bytes:
+        usage: "GAUGE"
+        description: "Disk space used by the database"

fly.toml

@@ -31,3 +31,7 @@ processes = []
 [[mounts]]
   destination = "/data"
   source = "pg_data"
+
+[metrics]
+  path = "/metrics"
+  port = 9187