pkg/SugarDB
1
0
Fork 0
mirror of https://github.com/EchoVault/SugarDB.git synced 2025-10-14 04:03:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implemented SMEMBERS command

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2023-12-23 22:58:17 +03:00
parent 2690aa46d0
commit 4e57e427ce
1 changed files with 33 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/modules/set/commands.go
View File

@@ -449,6 +449,7 @@ func handleSISMEMBER(ctx context.Context, cmd []string, server utils.Server) ([]
if err != nil {
return nil, err
}
defer server.KeyRUnlock(key)
set, ok := server.GetValue(key).(*Set)
if !ok {
@@ -463,7 +464,38 @@ func handleSISMEMBER(ctx context.Context, cmd []string, server utils.Server) ([]
}
func handleSMEMBERS(ctx context.Context, cmd []string, server utils.Server) ([]byte, error) {
return nil, errors.New("SMEMBERS not implemented")
if len(cmd) != 2 {
return nil, errors.New(utils.WRONG_ARGS_RESPONSE)
}
key := cmd[1]
if !server.KeyExists(key) {
return []byte("*0\r\n\r\n"), nil
}
_, err := server.KeyRLock(ctx, key)
if err != nil {
return nil, err
}
defer server.KeyRUnlock(key)
set, ok := server.GetValue(key).(*Set)
if !ok {
return nil, fmt.Errorf("value at key %s is not a set", key)
}
elems := set.GetAll()
res := fmt.Sprintf("*%d", len(elems))
for i, e := range elems {
res = fmt.Sprintf("%s\r\n$%d\r\n%s", res, len(e), e)
if i == len(elems)-1 {
res += "\r\n\r\n"
}
}
return []byte(res), nil
}
func handleSMISMEMBER(ctx context.Context, cmd []string, server utils.Server) ([]byte, error) {