redis hook

resource/redisCli/redis.go

@@ -26,6 +26,8 @@ func init() {
 	} else {
 		mylogrus.MyLog.Info("redis db0 connection success - ", pong)
 	}
+	// log hook
+	RedisClient.AddHook(redisHook{})
 }
 
 func GetRedis() *redis.Client {

resource/redisCli/redis_hook.go

+package redisCli
+
+import (
+	"context"
+	"git.hilo.cn/hilo-common/mylogrus"
+	"github.com/go-redis/redis/v8"
+	"time"
+)
+
+type redisCost struct{}
+
+var redisCostKey = redisCost{}
+
+type redisHook struct{}
+
+func (redisHook) BeforeProcess(ctx context.Context, cmd redis.Cmder) (context.Context, error) {
+	ctx = context.WithValue(ctx, redisCostKey, time.Now())
+	return ctx, nil
+}
+
+func (redisHook) AfterProcess(ctx context.Context, cmd redis.Cmder) error {
+	traceId, userId := ctx.Value("traceId"), ctx.Value("userId")
+	start := ctx.Value(redisCostKey)
+	var cost int64
+	if s, ok := start.(time.Time); ok {
+		cost = time.Now().Sub(s).Milliseconds()
+	}
+	mylogrus.MyLog.Infof("redis cmd: <%s>,err:%v traceId:%v,userId:%v,cost:%v ms", cmd.String(), cmd.Err(), traceId, userId, cost)
+	return nil
+}
+
+func (redisHook) BeforeProcessPipeline(ctx context.Context, cmds []redis.Cmder) (context.Context, error) {
+	return ctx, nil
+}
+
+func (redisHook) AfterProcessPipeline(ctx context.Context, cmds []redis.Cmder) error {
+	return nil
+}