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.RedisLog.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
}