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

Added tests for SET, GET, and MGET commands

Browse Source
This commit is contained in:
Kelvin Clement Mwinuka
2023-07-16 01:28:26 +08:00
parent b5cadf43a7
commit 0549bcc745
3 changed files with 77 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
server/plugins/commands/setget/setget.go
View File

@@ -3,7 +3,6 @@ package main
import (
"bufio"
"fmt"
"strconv"
"strings"
"github.com/kelvinmwinuka/memstore/utils"
@@ -67,15 +66,22 @@ func handleGet(cmd []string, s Server, conn *bufio.Writer) {
case string:
conn.Write([]byte(fmt.Sprintf("+%s\r\n\n", value)))
case float64:
conn.Write([]byte(fmt.Sprintf("+%f\r\n\n", value)))
s := strings.TrimRight(fmt.Sprintf("%f", value), "0")
conn.Write([]byte(fmt.Sprintf("+%s\r\n\n", s)))
case int:
conn.Write([]byte(fmt.Sprintf("+%d\r\n\n", value)))
conn.Write([]byte(fmt.Sprintf(":%d\r\n\n", value)))
}
conn.Flush()
}
func handleMGet(cmd []string, s Server, conn *bufio.Writer) {
if len(cmd) < 2 {
conn.Write([]byte("-Error wrong number of args for MGET command\r\n\n"))
conn.Flush()
return
}
vals := []string{}
s.Lock()
@@ -87,7 +93,7 @@ func handleMGet(cmd []string, s Server, conn *bufio.Writer) {
case string:
vals = append(vals, fmt.Sprintf("%s", s.GetData(key)))
case float64:
vals = append(vals, fmt.Sprintf("%f", s.GetData(key)))
vals = append(vals, strings.TrimRight(fmt.Sprintf("%f", s.GetData(key)), "0"))
case int:
vals = append(vals, fmt.Sprintf("%d", s.GetData(key)))
}
@@ -110,26 +116,10 @@ func handleSet(cmd []string, s Server, conn *bufio.Writer) {
default:
conn.Write([]byte("-Error wrong number of args for SET command\r\n\n"))
conn.Flush()
case x > 3:
s.Lock()
s.SetData(cmd[1], strings.Join(cmd[2:], " "))
s.Unlock()
conn.Write([]byte("+OK\r\n"))
case x == 3:
val, err := strconv.ParseFloat(cmd[2], 32)
s.Lock()
if err != nil {
s.SetData(cmd[1], cmd[2])
} else if !utils.IsInteger(val) {
s.SetData(cmd[1], val)
} else {
s.SetData(cmd[1], int(val))
}
s.SetData(cmd[1], utils.AdaptType(cmd[2]))
s.Unlock()
conn.Write([]byte("+OK\r\n\n"))
conn.Flush()
}