chore: update mongodb envs

apecloud · Aug 15, 2024 · 6ba4d15 · 6ba4d15
1 parent 254df15
commit 6ba4d15
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/engines/mongodb/config.go b/engines/mongodb/config.go
@@ -41,8 +41,10 @@ const (
 	params           = "params"
 	adminDatabase    = "admin"
 
-	defaultTimeout = 5 * time.Second
-	defaultDBPort  = 27017
+	defaultTimeout  = 5 * time.Second
+	defaultDBPort   = 27017
+	RootUserEnv     = "MONGODB_USER"
+	RootPasswordEnv = "MONGODB_PASSWORD"
 )
 
 type Config struct {
@@ -87,10 +89,15 @@ func NewConfig(properties map[string]string) (*Config, error) {
 
 	if viper.IsSet(constant.KBEnvServiceUser) {
 		config.Username = viper.GetString(constant.KBEnvServiceUser)
+	} else if viper.IsSet(RootUserEnv) {
+		config.Username = viper.GetString(RootUserEnv)
+
 	}
 
 	if viper.IsSet(constant.KBEnvServicePassword) {
 		config.Password = viper.GetString(constant.KBEnvServicePassword)
+	} else if viper.IsSet(RootPasswordEnv) {
+		config.Password = viper.GetString(RootPasswordEnv)
 	}
 
 	if viper.IsSet(constant.KBEnvClusterCompName) {