48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
|
|
"kra/internal/conf"
|
|
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
// NewRedisClient 创建Redis客户端
|
|
func NewRedisClient(c *conf.Data, l log.Logger) (redis.UniversalClient, func(), error) {
|
|
helper := log.NewHelper(l)
|
|
|
|
var client redis.UniversalClient
|
|
|
|
if c.Redis.UseCluster {
|
|
client = redis.NewClusterClient(&redis.ClusterOptions{
|
|
Addrs: c.Redis.ClusterAddrs,
|
|
Password: c.Redis.Password,
|
|
})
|
|
} else {
|
|
client = redis.NewClient(&redis.Options{
|
|
Addr: c.Redis.Addr,
|
|
Password: c.Redis.Password,
|
|
DB: int(c.Redis.Db),
|
|
})
|
|
}
|
|
|
|
// 测试连接
|
|
ctx := context.Background()
|
|
if _, err := client.Ping(ctx).Result(); err != nil {
|
|
helper.Errorf("failed to connect redis: %v", err)
|
|
return nil, nil, err
|
|
}
|
|
|
|
cleanup := func() {
|
|
helper.Info("closing redis connection")
|
|
if err := client.Close(); err != nil {
|
|
helper.Errorf("failed to close redis: %v", err)
|
|
}
|
|
}
|
|
|
|
helper.Info("redis connected successfully")
|
|
return client, cleanup, nil
|
|
}
|