You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
1.7 KiB
80 lines
1.7 KiB
package dorm
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"github.com/redis/go-redis/v9"
|
|
"time"
|
|
)
|
|
|
|
// RedisClientFun *RedisClient 驱动
|
|
type RedisClientFun func() *RedisClient
|
|
|
|
type RedisClientConfig struct {
|
|
Addr string // 地址
|
|
Password string // 密码
|
|
DB int // 数据库
|
|
PoolSize int // 连接池大小
|
|
ReadTimeout time.Duration // 读取超时
|
|
}
|
|
|
|
// RedisClient
|
|
// https://redis.uptrace.dev/
|
|
type RedisClient struct {
|
|
db *redis.Client // 驱动
|
|
}
|
|
|
|
// NewRedisClient 创建实例
|
|
func NewRedisClient(config *RedisClientConfig) (*RedisClient, error) {
|
|
|
|
c := &RedisClient{}
|
|
if config.PoolSize == 0 {
|
|
config.PoolSize = 100
|
|
}
|
|
|
|
c.db = redis.NewClient(&redis.Options{
|
|
Addr: config.Addr, // 地址
|
|
Password: config.Password, // 密码
|
|
DB: config.DB, // 数据库
|
|
PoolSize: config.PoolSize, // 连接池大小
|
|
ReadTimeout: config.ReadTimeout, // 读取超时
|
|
})
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// 检测 Redis 连接是否正常连接
|
|
_, err := c.db.Ping(ctx).Result()
|
|
if err != nil {
|
|
return nil, errors.New(fmt.Sprintf("连接失败：%v", err))
|
|
}
|
|
|
|
return c, nil
|
|
}
|
|
|
|
// NewRedisClientURL 创建实例
|
|
func NewRedisClientURL(redisURL string) (*RedisClient, error) {
|
|
|
|
c := &RedisClient{}
|
|
|
|
opt, err := redis.ParseURL(redisURL)
|
|
if err != nil {
|
|
return c, nil
|
|
}
|
|
|
|
// 创建 Redis 客户端
|
|
c.db = redis.NewClient(opt)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
// 检测 Redis 连接是否正常连接
|
|
_, err = c.db.Ping(ctx).Result()
|
|
if err != nil {
|
|
return nil, errors.New(fmt.Sprintf("连接失败：%v", err))
|
|
}
|
|
|
|
return c, nil
|
|
}
|