apps/redis-go
0
0
Fork 0
mirror of https://github.com/HDT3213/godis.git synced 2025-10-07 17:51:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

added strlen and append commands

Browse Source
This commit is contained in:
Oleg
2021-06-25 11:48:52 +03:00
committed by finley
parent ae25076689
commit af0ffe6e83
2 changed files with 87 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
string.go
View File

@@ -473,6 +473,34 @@ func execDecrBy(db *DB, args [][]byte) redis.Reply {
return reply.MakeIntReply(-delta)
}
// execStrLen returns len of string value bound to the given key
func execStrLen(db *DB, args [][]byte) redis.Reply {
key := string(args[0])
bytes, err := db.getAsString(key)
if err != nil {
return err
}
if bytes == nil {
return reply.MakeIntReply(0)
}
return reply.MakeIntReply(int64(len(bytes)))
}
// execAppend sets string value and time to live to the given key
func execAppend(db *DB, args [][]byte) redis.Reply {
key := string(args[0])
bytes, err := db.getAsString(key)
if err != nil {
return err
}
bytes = append(bytes, args[1]...)
db.PutEntity(key, &DataEntity{
Data: bytes,
})
db.AddAof(makeAofCmd("append", args))
return reply.MakeIntReply(int64(len(bytes)))
}
func init() {
RegisterCommand("Set", execSet, writeFirstKey, rollbackFirstKey, -3)
RegisterCommand("SetNx", execSetNX, writeFirstKey, rollbackFirstKey, 3)
@@ -488,4 +516,6 @@ func init() {
RegisterCommand("IncrByFloat", execIncrByFloat, writeFirstKey, rollbackFirstKey, 3)
RegisterCommand("Decr", execDecr, writeFirstKey, rollbackFirstKey, 2)
RegisterCommand("DecrBy", execDecrBy, writeFirstKey, rollbackFirstKey, 3)
RegisterCommand("StrLen", execStrLen, readFirstKey, nil, 2)
RegisterCommand("Append", execAppend, writeFirstKey, rollbackFirstKey, 3)
}

57
string_test.go
View File

@@ -315,3 +315,60 @@ func TestMSetNX(t *testing.T) {
result = testDB.Exec(nil, utils.ToCmdLine2("MSETNX", args[0:4]...))
asserts.AssertIntReply(t, result, 0)
}
func TestStrLen(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
testDB.Exec(nil, utils.ToCmdLine2("SET", key, key))
actual := testDB.Exec(nil, utils.ToCmdLine("StrLen", key))
len, ok := actual.(*reply.IntReply)
if !ok {
t.Errorf("expect int bulk reply, get: %s", string(actual.ToBytes()))
return
}
asserts.AssertIntReply(t, len, 10)
}
func TestStrLen_KeyNotExist(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
actual := testDB.Exec(nil, utils.ToCmdLine("StrLen", key))
result, ok := actual.(*reply.IntReply)
if !ok {
t.Errorf("expect null bulk reply, get: %s", string(actual.ToBytes()))
return
}
asserts.AssertIntReply(t, result, 0)
}
func TestAppend_KeyExist(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
key2 := utils.RandString(10)
testDB.Exec(nil, utils.ToCmdLine2("SET", key, key))
actual := testDB.Exec(nil, utils.ToCmdLine("Append", key, key2))
val, ok := actual.(*reply.IntReply)
if !ok {
t.Errorf("expect nil bulk reply, get: %s", string(actual.ToBytes()))
return
}
asserts.AssertIntReply(t, val, len(key)*2)
}
func TestAppend_KeyNotExist(t *testing.T) {
testDB.Flush()
key := utils.RandString(10)
actual := testDB.Exec(nil, utils.ToCmdLine("Append", key, key))
val, ok := actual.(*reply.IntReply)
if !ok {
t.Errorf("expect nil bulk reply, get: %s", string(actual.ToBytes()))
return
}
asserts.AssertIntReply(t, val, len(key))
}