chore: update redis envs

apecloud · Aug 19, 2024 · f1d37e7 · f1d37e7
1 parent 287b13b
commit f1d37e7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/engines/redis/manager.go b/engines/redis/manager.go
@@ -32,6 +32,11 @@ import (
 	"github.com/apecloud/dbctl/engines"
 )
 
+const (
+	EnvRootUser   = "REDIS_DEFAULT_USER"
+	EnvRootPasswd = "REDIS_DEFAULT_PASSWORD"
+)
+
 var (
 	redisUser   = "default"
 	redisPasswd = ""
@@ -58,10 +63,14 @@ func NewManager(properties engines.Properties) (engines.DBManager, error) {
 
 	if viper.IsSet(constant.KBEnvServiceUser) {
 		redisUser = viper.GetString(constant.KBEnvServiceUser)
+	} else if viper.IsSet(EnvRootUser) {
+		redisUser = viper.GetString(EnvRootUser
 	}
 
 	if viper.IsSet(constant.KBEnvServicePassword) {
 		redisPasswd = viper.GetString(constant.KBEnvServicePassword)
+	} else if viper.IsSet(EnvRootPasswd) {
+		redisPasswd = viper.GetString(EnvRootPasswd)
 	}
 
 	managerBase, err := engines.NewDBManagerBase(logger)