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

zpopmin

Browse Source
This commit is contained in:
finley
2022-08-21 05:30:00 +08:00
parent 9da335811f
commit dddf9a8f6b
4 changed files with 71 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
database/sortedset.go
View File

@@ -514,6 +514,37 @@ func execZRemRangeByRank(db *DB, args [][]byte) redis.Reply {
return protocol.MakeIntReply(removed)
}
func execZPopMin(db *DB, args [][]byte) redis.Reply {
key := string(args[0])
count := 1
if len(args) > 1 {
var err error
count, err = strconv.Atoi(string(args[1]))
if err != nil {
return protocol.MakeErrReply("ERR value is not an integer or out of range")
}
}
sortedSet, errReply := db.getAsSortedSet(key)
if errReply != nil {
return errReply
}
if sortedSet == nil {
return protocol.MakeEmptyMultiBulkReply()
}
removed := sortedSet.PopMin(count)
if len(removed) > 0 {
db.addAof(utils.ToCmdLine3("zpopmin", args...))
}
result := make([][]byte, 0, len(removed)*2)
for _, element := range removed {
scoreStr := strconv.FormatFloat(element.Score, 'f', -1, 64)
result = append(result, []byte(element.Member), []byte(scoreStr))
}
return protocol.MakeMultiBulkReply(result)
}
// execZRem removes given members
func execZRem(db *DB, args [][]byte) redis.Reply {
// parse args
@@ -602,6 +633,7 @@ func init() {
RegisterCommand("ZRangeByScore", execZRangeByScore, readFirstKey, nil, -4, flagReadOnly)
RegisterCommand("ZRevRange", execZRevRange, readFirstKey, nil, -4, flagReadOnly)
RegisterCommand("ZRevRangeByScore", execZRevRangeByScore, readFirstKey, nil, -4, flagReadOnly)
RegisterCommand("ZPopMin", execZPopMin, writeFirstKey, rollbackFirstKey, -2, flagWrite)
RegisterCommand("ZRem", execZRem, writeFirstKey, undoZRem, -3, flagWrite)
RegisterCommand("ZRemRangeByScore", execZRemRangeByScore, writeFirstKey, rollbackFirstKey, 4, flagWrite)
RegisterCommand("ZRemRangeByRank", execZRemRangeByRank, writeFirstKey, rollbackFirstKey, 4, flagWrite)