Add validation for duplicated database entries

ContaAzul · Nov 23, 2019 · 6965864 · 6965864
1 parent f9cfc56
commit 6965864
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -25,5 +25,16 @@ func Parse(path string) Config {
 	if err := yaml.Unmarshal(bts, &cfg); err != nil {
 		log.WithError(err).Fatalf("failed to unmarshall config file: %s", path)
 	}
-	return cfg
+	return validate(cfg)
+}
+
+func validate(config Config) Config {
+	names := make(map[string]bool)
+	for _, conf := range config.Databases {
+		if names[conf.Name] {
+			log.Fatalf("failed to validate configuration. A database named '%s' has already been declared'", conf.Name)
+		}
+		names[conf.Name] = true
+	}
+	return config
 }
diff --git a/main.go b/main.go
@@ -84,7 +84,7 @@ func watch(db *sql.DB, reg prometheus.Registerer, name string) {
 	reg.MustRegister(gauges.ScheduledCheckpoints())
 	reg.MustRegister(gauges.BuffersMaxWrittenClean())
 	reg.MustRegister(gauges.BuffersWritten())
-	//reg.MustRegister(gauges.DeadTuples())
+	reg.MustRegister(gauges.DeadTuples())
 	reg.MustRegister(gauges.HeapBlocksHit())
 	reg.MustRegister(gauges.HeapBlocksRead())
 	reg.MustRegister(gauges.IndexScans())