Adds defaults for status query. Refs #3

ariel17 · Mar 7, 2023 · 7937b8b · 7937b8b
1 parent d7de0be
commit 7937b8b
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pkg/configs/db.go b/pkg/configs/db.go
@@ -5,8 +5,9 @@ import (
 )
 
 const (
-	dsnKey         = "DATABASE_DSN"
-	statusQueryKey = "DATABASE_STATUS_QUERY"
+	dsnKey             = "DATABASE_DSN"
+	statusQueryKey     = "DATABASE_STATUS_QUERY"
+	defaultStatusQuery = "SELECT 1"
 )
 
 var (
@@ -27,5 +28,9 @@ func GetStatusQuery() string {
 
 func init() {
 	dsn = os.Getenv(dsnKey)
+
 	statusQuery = os.Getenv(statusQueryKey)
+	if statusQuery == "" {
+		statusQuery = defaultStatusQuery
+	}
 }