mirror of
https://github.com/nalgeon/redka.git
synced 2025-10-15 20:40:39 +08:00
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package zset
|
|
|
|
import (
|
|
"github.com/nalgeon/redka/internal/parser"
|
|
"github.com/nalgeon/redka/internal/redis"
|
|
)
|
|
|
|
// Iterates over members and scores of a sorted set.
|
|
// ZSCAN key cursor [MATCH pattern] [COUNT count]
|
|
// https://redis.io/commands/zscan
|
|
type ZScan struct {
|
|
redis.BaseCmd
|
|
Key string
|
|
Cursor int
|
|
Match string
|
|
Count int
|
|
}
|
|
|
|
func ParseZScan(b redis.BaseCmd) (*ZScan, error) {
|
|
cmd := &ZScan{BaseCmd: b}
|
|
|
|
err := parser.New(
|
|
parser.String(&cmd.Key),
|
|
parser.Int(&cmd.Cursor),
|
|
parser.Named("match", parser.String(&cmd.Match)),
|
|
parser.Named("count", parser.Int(&cmd.Count)),
|
|
).Required(2).Run(cmd.Args())
|
|
if err != nil {
|
|
return cmd, err
|
|
}
|
|
|
|
// all elements by default
|
|
if cmd.Match == "" {
|
|
cmd.Match = "*"
|
|
}
|
|
|
|
return cmd, nil
|
|
}
|
|
|
|
func (cmd *ZScan) Run(w redis.Writer, red redis.Redka) (any, error) {
|
|
res, err := red.ZSet().Scan(cmd.Key, cmd.Cursor, cmd.Match, cmd.Count)
|
|
if err != nil {
|
|
w.WriteError(cmd.Error(err))
|
|
return nil, err
|
|
}
|
|
|
|
w.WriteArray(2)
|
|
w.WriteInt(res.Cursor)
|
|
w.WriteArray(len(res.Items) * 2)
|
|
for _, it := range res.Items {
|
|
w.WriteBulk(it.Elem)
|
|
redis.WriteFloat(w, it.Score)
|
|
}
|
|
return res, nil
|
|
}
|