44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
package redis
|
|
|
|
import (
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
|
|
"github.com/redis/go-redis/extra/redisotel/v9"
|
|
"github.com/redis/go-redis/v9"
|
|
|
|
conf "github.com/tx7do/kratos-bootstrap/api/gen/go/conf/v1"
|
|
)
|
|
|
|
// NewClient create go-redis client
|
|
func NewClient(conf *conf.Data, logger *log.Helper) (rdb *redis.Client) {
|
|
if rdb = redis.NewClient(&redis.Options{
|
|
Addr: conf.GetRedis().GetAddr(),
|
|
Password: conf.GetRedis().GetPassword(),
|
|
DB: int(conf.GetRedis().GetDb()),
|
|
DialTimeout: conf.GetRedis().GetDialTimeout().AsDuration(),
|
|
WriteTimeout: conf.GetRedis().GetWriteTimeout().AsDuration(),
|
|
ReadTimeout: conf.GetRedis().GetReadTimeout().AsDuration(),
|
|
}); rdb == nil {
|
|
logger.Fatalf("failed opening connection to redis")
|
|
return nil
|
|
}
|
|
|
|
// open tracing instrumentation.
|
|
if conf.GetRedis().GetEnableTracing() {
|
|
if err := redisotel.InstrumentTracing(rdb); err != nil {
|
|
logger.Fatalf("failed open tracing: %s", err.Error())
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// open metrics instrumentation.
|
|
if conf.GetRedis().GetEnableMetrics() {
|
|
if err := redisotel.InstrumentMetrics(rdb); err != nil {
|
|
logger.Fatalf("failed open metrics: %s", err.Error())
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return rdb
|
|
}
|