pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-05 16:06:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated sorted_set commands package to use new keyspace methods and its tests to use resp connection instead of calling handler directly

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2024-05-26 14:48:13 +08:00
parent 15191dccee
commit edf82886bd
4 changed files with 1996 additions and 1570 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/modules/sorted_set/key_funcs.go
View File

@@ -135,13 +135,11 @@ func zinterstoreKeyFunc(cmd []string) (internal.KeyExtractionFuncResult, error)
return internal.KeyExtractionFuncResult{}, errors.New(constants.WrongArgsResponse)
}
endIdx := slices.IndexFunc(cmd[1:], func(s string) bool {
if strings.EqualFold(s, "WEIGHTS") ||
return strings.EqualFold(s, "WEIGHTS") ||
strings.EqualFold(s, "AGGREGATE") ||
strings.EqualFold(s, "WITHSCORES") {
return true
}
return false
strings.EqualFold(s, "WITHSCORES")
})
if endIdx == -1 {
return internal.KeyExtractionFuncResult{
Channels: make([]string, 0),
@@ -149,13 +147,15 @@ func zinterstoreKeyFunc(cmd []string) (internal.KeyExtractionFuncResult, error)
WriteKeys: cmd[1:2],
}, nil
}
if endIdx >= 3 {
return internal.KeyExtractionFuncResult{
Channels: make([]string, 0),
ReadKeys: cmd[2:endIdx],
ReadKeys: cmd[2 : endIdx+1],
WriteKeys: cmd[1:2],
}, nil
}
return internal.KeyExtractionFuncResult{}, errors.New(constants.WrongArgsResponse)
}
@@ -377,7 +377,7 @@ func zunionstoreKeyFunc(cmd []string) (internal.KeyExtractionFuncResult, error)
if endIdx >= 1 {
return internal.KeyExtractionFuncResult{
Channels: make([]string, 0),
ReadKeys: cmd[2:endIdx],
ReadKeys: cmd[2 : endIdx+1],
WriteKeys: cmd[1:2],
}, nil
}