优化 redis 配置方式

This commit is contained in:
dragon
2025-07-23 16:46:58 +08:00
parent 93a26f1007
commit 4ca455fdae
2 changed files with 16 additions and 7 deletions

View File

@@ -40,7 +40,7 @@ func NewAutoConfig(src string) (KVConfig, error) {
if pass == "" {
pass = query.Get("password")
}
db := query.Get("db")
db := strings.TrimPrefix(parse.Path, "/")
if db == "" {
db = "0"
}

View File

@@ -5,6 +5,8 @@ import (
"fmt"
"time"
"go.uber.org/zap"
"github.com/redis/go-redis/v9"
)
@@ -17,13 +19,20 @@ func NewConfigRedis(ctx context.Context, addr string, password string, db int) (
if addr == "" {
return nil, fmt.Errorf("addr is empty")
}
zap.L().Debug("connect redis", zap.String("addr", addr), zap.Int("db", db))
client := redis.NewClient(&redis.Options{
Addr: addr,
Password: password,
DB: db,
})
_, err := client.Ping(ctx).Result()
if err != nil {
_ = client.Close()
return nil, err
}
return &ConfigRedis{
ctx: ctx,
client: redis.NewClient(&redis.Options{
Addr: addr,
Password: password,
DB: db,
}),
ctx: ctx,
client: client,
}, nil
}