Check store on startup.

jbub · Dec 30, 2019 · d4e8ae3 · d4e8ae3
1 parent 777e244
commit d4e8ae3
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/cmd/server.go b/cmd/server.go
@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"context"
 	"fmt"
 
 	"github.com/jbub/pgbouncer_exporter/internal/collector"
@@ -28,6 +29,13 @@ func runServer(ctx *cli.Context) error {
 	}
 	defer st.Close()
 
+	checkCtx, cancel := context.WithTimeout(context.Background(), cfg.StoreTimeout)
+	defer cancel()
+
+	if err := st.Check(checkCtx); err != nil {
+		return fmt.Errorf("could not check store: %v", err)
+	}
+
 	exp := collector.New(cfg, st)
 	srv := server.New(cfg, exp)