add redis cluster host network port

apecloud · Dec 4, 2024 · caddc86 · caddc86
1 parent f02cafa
commit caddc86
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/engines/redis/manager.go b/engines/redis/manager.go
@@ -37,8 +37,9 @@ const (
 	EnvRootUser   = "REDIS_DEFAULT_USER"
 	EnvRootPasswd = "REDIS_DEFAULT_PASSWORD"
 
-	EnvRedisHostNetworkPort         = "REDIS_HOST_NETWORK_PORT"
-	EnvRedisSentinelHostNetworkPort = "REDIS_SENTINEL_HOST_NETWORK_PORT"
+	EnvRedisHostNetworkPort         = "DBCTL_REDIS_HOST_NETWORK_PORT"
+	EnvRedisSentinelHostNetworkPort = "DBCTL_REDIS_SENTINEL_HOST_NETWORK_PORT"
+	EnvRedisClusterHostNetworkPort  = "DBCTL_REDIS_CLUSTER_HOST_NETWORK_PORT"
 )
 
 var (
@@ -79,6 +80,8 @@ func NewManager(properties engines.Properties) (engines.DBManager, error) {
 
 	if viper.IsSet(EnvRedisHostNetworkPort) {
 		properties["redisHost"] = fmt.Sprintf("127.0.0.1:%s", viper.GetString(EnvRedisHostNetworkPort))
+	} else if viper.IsSet(EnvRedisClusterHostNetworkPort) {
+		properties["redisHost"] = fmt.Sprintf("127.0.0.1:%s", viper.GetString(EnvRedisClusterHostNetworkPort))
 	}
 
 	managerBase, err := engines.NewDBManagerBase(logger)