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.
48 lines
956 B
48 lines
956 B
2 years ago
|
package dorm
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"errors"
|
||
|
"fmt"
|
||
|
"github.com/go-redis/redis/v9"
|
||
|
"time"
|
||
|
)
|
||
|
|
||
|
// RedisClient7
|
||
|
// https://redis.uptrace.dev/
|
||
|
type RedisClient7 struct {
|
||
|
Db *redis.Client // 驱动
|
||
|
config *ConfigRedisClient // 配置
|
||
|
}
|
||
|
|
||
|
// NewRedisClient7 Redis 7
|
||
|
func NewRedisClient7(config *ConfigRedisClient) (*RedisClient7, error) {
|
||
|
|
||
|
c := &RedisClient7{}
|
||
|
c.config = config
|
||
|
if c.config.PoolSize == 0 {
|
||
|
c.config.PoolSize = 100
|
||
|
}
|
||
|
|
||
|
c.Db = redis.NewClient(&redis.Options{
|
||
|
Addr: c.config.Addr, // 地址
|
||
|
Password: c.config.Password, // 密码
|
||
|
DB: c.config.DB, // 数据库
|
||
|
PoolSize: c.config.PoolSize, // 连接池大小
|
||
|
})
|
||
|
|
||
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||
|
defer cancel()
|
||
|
|
||
|
_, err := c.Db.Ping(ctx).Result()
|
||
|
if err != nil {
|
||
|
return nil, errors.New(fmt.Sprintf("连接失败:%v", err))
|
||
|
}
|
||
|
|
||
|
return c, nil
|
||
|
}
|
||
|
|
||
|
func (c *RedisClient7) GetDb() *redis.Client {
|
||
|
return c.Db
|
||
|
}
|