mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-01 23:10:26 +08:00
4e30bdb7cc
* fix(connection):fix the bug for nil error in connection when connection reset * fix(sqleditor): fix the spell error in sql editor
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package rdm
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"mayfly-go/pkg/errorx"
|
||
"mayfly-go/pkg/logx"
|
||
|
||
"github.com/redis/go-redis/v9"
|
||
)
|
||
|
||
// redis连接信息
|
||
type RedisConn struct {
|
||
Id string
|
||
Info *RedisInfo
|
||
|
||
Cli *redis.Client
|
||
ClusterCli *redis.ClusterClient
|
||
}
|
||
|
||
/******************* pool.Conn impl *******************/
|
||
|
||
func (r *RedisConn) Close() error {
|
||
mode := r.Info.Mode
|
||
if mode == StandaloneMode || mode == SentinelMode {
|
||
if err := r.Cli.Close(); err != nil {
|
||
logx.Errorf("close redis standalone instance [%s] connection failed: %s", r.Id, err.Error())
|
||
return err
|
||
}
|
||
r.Cli = nil
|
||
return nil
|
||
}
|
||
|
||
if mode == ClusterMode {
|
||
if err := r.ClusterCli.Close(); err != nil {
|
||
logx.Errorf("close redis cluster instance [%s] connection failed: %s", r.Id, err.Error())
|
||
return err
|
||
}
|
||
r.ClusterCli = nil
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *RedisConn) Ping() error {
|
||
// 首先检查r是否为nil
|
||
if r == nil {
|
||
return fmt.Errorf("redis connection is nil")
|
||
}
|
||
// 然后检查r.Cli是否为nil，这是避免空指针异常的关键
|
||
if r.Cli == nil {
|
||
return fmt.Errorf("redis client is nil")
|
||
}
|
||
cmd := r.Cli.Ping(context.Background())
|
||
if cmd == nil {
|
||
return fmt.Errorf("the ping cmd is nil")
|
||
}
|
||
_, err := cmd.Result()
|
||
return err
|
||
}
|
||
|
||
// 获取命令执行接口的具体实现
|
||
func (r *RedisConn) GetCmdable() redis.Cmdable {
|
||
redisMode := r.Info.Mode
|
||
if redisMode == "" || redisMode == StandaloneMode || r.Info.Mode == SentinelMode {
|
||
return r.Cli
|
||
}
|
||
if redisMode == ClusterMode {
|
||
return r.ClusterCli
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r *RedisConn) Scan(cursor uint64, match string, count int64) ([]string, uint64, error) {
|
||
return r.GetCmdable().Scan(context.Background(), cursor, match, count).Result()
|
||
}
|
||
|
||
// 执行redis命令
|
||
// 如: SET str value命令则args为['SET', 'str', 'val']
|
||
func (r *RedisConn) RunCmd(ctx context.Context, args ...any) (any, error) {
|
||
redisMode := r.Info.Mode
|
||
if redisMode == "" || redisMode == StandaloneMode || r.Info.Mode == SentinelMode {
|
||
return r.Cli.Do(ctx, args...).Result()
|
||
}
|
||
if redisMode == ClusterMode {
|
||
return r.ClusterCli.Do(ctx, args...).Result()
|
||
}
|
||
return nil, errorx.NewBiz("redis mode error")
|
||
}
|