go/go-mixed
1
0
Fork 0
mirror of https://github.com/charlienet/go-mixed.git synced 2025-07-18 08:32:40 +08:00
Code Issues Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
Charlie
2023-10-13 14:38:42 +08:00
parent d1c269ed90
commit 6647f96978
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
redis/rename_hook.go
View File

@ -8,10 +8,6 @@ import (
"github.com/redis/go-redis/v9" "github.com/redis/go-redis/v9"
) )
var (
// sequentials = sets.NewHashSet("RENAME", "RENAMENX", "MGET", "BLPOP", "BRPOP", "RPOPLPUSH", "SDIFFSTORE", "SINTER")
)
type renameKey struct { type renameKey struct {
prefix string prefix string
separator string separator string
@ -57,10 +53,19 @@ func (r renameKey) renameKey(cmd redis.Cmder) {
switch strings.ToUpper(cmd.Name()) { switch strings.ToUpper(cmd.Name()) {
case "SELECT": case "SELECT":
// 无KEY指令 // 无KEY指令
case "RENAME", "RENAMENX", "MGET", "BLPOP", "BRPOP", "RPOPLPUSH", "SDIFFSTORE", "SINTER", "SINTERSTORE", "SUNIONSTORE": case
"RENAME", "RENAMENX",
"MGET",
"RPOPLPUSH",
"SDIFF", "SDIFFSTORE", "SINTER", "SINTERSTORE",
"SUNION", "SUNIONSTORE",
"WATCH":
// 连续KEY // 连续KEY
r.rename(args, createSepuence(1, len(args), 1)...) r.rename(args, createSepuence(1, len(args), 1)...)
case "sssss": case
"BLPOP", "BRPOP",
"BRPOPLPUSH ",
"SMOVE":
// 除最后一个外连续键 // 除最后一个外连续键
r.rename(args, createSepuence(1, len(args)-1, 1)...) r.rename(args, createSepuence(1, len(args)-1, 1)...)
case "MSET", "MSETNX": case "MSET", "MSETNX":
@ -87,7 +92,7 @@ func (r renameKey) rename(args []any, indexes ...int) {
func createSepuence(start, end, step int) []int { func createSepuence(start, end, step int) []int {
ret := make([]int, 0, (end-start)/step+1) ret := make([]int, 0, (end-start)/step+1)
for i := start; i <= end; i += step { for i := start; i < end; i += step {
ret = append(ret, i) ret = append(ret, i)
} }
return ret return ret