36 lines
825 B
Go
36 lines
825 B
Go
package redis
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
red "github.com/go-redis/redis"
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
"github.com/tal-tech/go-zero/core/mapping"
|
|
"github.com/tal-tech/go-zero/core/timex"
|
|
)
|
|
|
|
func checkDuration(slowThreshold time.Duration) func(proc func(red.Cmder) error) func(red.Cmder) error {
|
|
return func(proc func(red.Cmder) error) func(red.Cmder) error {
|
|
return func(cmd red.Cmder) error {
|
|
start := timex.Now()
|
|
|
|
defer func() {
|
|
duration := timex.Since(start)
|
|
if duration > slowThreshold {
|
|
var buf strings.Builder
|
|
for i, arg := range cmd.Args() {
|
|
if i > 0 {
|
|
buf.WriteByte(' ')
|
|
}
|
|
buf.WriteString(mapping.Repr(arg))
|
|
}
|
|
logx.WithDuration(duration).Slowf("[REDIS] slowcall on executing: %s", buf.String())
|
|
}
|
|
}()
|
|
|
|
return proc(cmd)
|
|
}
|
|
}
|
|
}
|