chore: support reading db config from env (#105)

bnb-chain · Feb 26, 2024 · c6ca66e · c6ca66e
1 parent 6c96cc3
commit c6ca66e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/app/app.go b/app/app.go
@@ -39,6 +39,9 @@ func NewApp(cfg *config.Config) *App {
 	if password == "" {
 		password = getDBPass(&cfg.DBConfig)
 	}
+	if username == "" || password == "" { // read password from ENV
+		username, password = config.GetDBUsernamePasswordFromEnv()
+	}
 
 	dbPath := fmt.Sprintf("%s:%s@%s", username, password, cfg.DBConfig.DBPath)
 

diff --git a/config/utils.go b/config/utils.go
@@ -3,6 +3,7 @@ package config
 import (
 	"encoding/base64"
 	"fmt"
+	"os"
 
 	"github.com/aws/aws-sdk-go/aws"
 	"github.com/aws/aws-sdk-go/aws/session"
@@ -44,3 +45,9 @@ func GetSecret(secretName, region string) (string, error) {
 		return decodedBinarySecret, nil
 	}
 }
+
+func GetDBUsernamePasswordFromEnv() (string, string) {
+	username := os.Getenv("DB_USERNAME")
+	password := os.Getenv("DB_PASSWORD")
+	return username, password
+}