mirror of
https://github.com/go-eagle/eagle.git
synced 2025-10-05 16:46:59 +08:00
72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package redis
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/alicebob/miniredis"
|
|
"github.com/go-redis/redis/v8"
|
|
|
|
"github.com/1024casts/snake/pkg/log"
|
|
)
|
|
|
|
// RedisClient redis 客户端
|
|
var RedisClient *redis.Client
|
|
|
|
// ErrRedisNotFound not exist in redis
|
|
const ErrRedisNotFound = redis.Nil
|
|
|
|
// Config redis config
|
|
type Config struct {
|
|
Addr string
|
|
Password string
|
|
DB int
|
|
MinIdleConn int
|
|
DialTimeout time.Duration
|
|
ReadTimeout time.Duration
|
|
WriteTimeout time.Duration
|
|
PoolSize int
|
|
PoolTimeout time.Duration
|
|
}
|
|
|
|
// Init 实例化一个redis client
|
|
func Init(c *Config) *redis.Client {
|
|
RedisClient = redis.NewClient(&redis.Options{
|
|
Addr: c.Addr,
|
|
Password: c.Password,
|
|
DB: c.DB,
|
|
MinIdleConns: c.MinIdleConn,
|
|
DialTimeout: c.DialTimeout,
|
|
ReadTimeout: c.ReadTimeout,
|
|
WriteTimeout: c.WriteTimeout,
|
|
PoolSize: c.PoolSize,
|
|
PoolTimeout: c.PoolTimeout,
|
|
})
|
|
|
|
_, err := RedisClient.Ping(context.Background()).Result()
|
|
if err != nil {
|
|
log.Panicf("[redis] redis ping err: %+v", err)
|
|
}
|
|
|
|
// hook tracing (with open telemetry)
|
|
RedisClient.AddHook(NewTracingHook())
|
|
|
|
return RedisClient
|
|
}
|
|
|
|
// InitTestRedis 实例化一个可以用于单元测试的redis
|
|
func InitTestRedis() {
|
|
mr, err := miniredis.Run()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
// 打开下面命令可以测试链接关闭的情况
|
|
// defer mr.Close()
|
|
|
|
RedisClient = redis.NewClient(&redis.Options{
|
|
Addr: mr.Addr(),
|
|
})
|
|
fmt.Println("mini redis addr:", mr.Addr())
|
|
}
|