zishuo/tiny-rdm
1
0
Fork 0
mirror of https://github.com/tiny-craft/tiny-rdm.git synced 2025-10-07 08:01:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add app launch log page

Browse Source
This commit is contained in:
tiny-craft
2023-07-16 01:50:01 +08:00
parent 15c80bc9f7
commit 4879901a33
16 changed files with 355 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
backend/utils/redis/log_hook.go
View File

@@ -7,31 +7,41 @@ import (
"net"
)
type execCallback func(string)
type LogHook struct {
name string
name string
cmdExec execCallback
}
func NewHook(name string) LogHook {
return LogHook{
name: name,
func NewHook(name string, cmdExec execCallback) *LogHook {
return &LogHook{
name: name,
cmdExec: cmdExec,
}
}
func (LogHook) DialHook(next redis.DialHook) redis.DialHook {
func (l *LogHook) DialHook(next redis.DialHook) redis.DialHook {
return func(ctx context.Context, network, addr string) (net.Conn, error) {
return next(ctx, network, addr)
}
}
func (LogHook) ProcessHook(next redis.ProcessHook) redis.ProcessHook {
func (l *LogHook) ProcessHook(next redis.ProcessHook) redis.ProcessHook {
return func(ctx context.Context, cmd redis.Cmder) error {
log.Println(cmd.String())
log.Println(cmd)
if l.cmdExec != nil {
l.cmdExec(cmd.String())
}
return next(ctx, cmd)
}
}
func (LogHook) ProcessPipelineHook(next redis.ProcessPipelineHook) redis.ProcessPipelineHook {
func (l *LogHook) ProcessPipelineHook(next redis.ProcessPipelineHook) redis.ProcessPipelineHook {
return func(ctx context.Context, cmds []redis.Cmder) error {
for _, cmd := range cmds {
log.Println(cmd.String())
log.Println("pipeline: ", cmd)
if l.cmdExec != nil {
l.cmdExec(cmd.String())
}
}
return next(ctx, cmds)
}