96 lines
1.7 KiB
Go
96 lines
1.7 KiB
Go
package miniredis
|
|
|
|
import "github.com/alicebob/miniredis/v2/server"
|
|
|
|
// commandsHll handles all hll related operations.
|
|
func commandsHll(m *Miniredis) {
|
|
m.srv.Register("PFADD", m.cmdPfadd)
|
|
m.srv.Register("PFCOUNT", m.cmdPfcount)
|
|
m.srv.Register("PFMERGE", m.cmdPfmerge)
|
|
}
|
|
|
|
// PFADD
|
|
func (m *Miniredis) cmdPfadd(c *server.Peer, cmd string, args []string) {
|
|
if len(args) < 2 {
|
|
setDirty(c)
|
|
c.WriteError(errWrongNumber(cmd))
|
|
return
|
|
}
|
|
if !m.handleAuth(c) {
|
|
return
|
|
}
|
|
if m.checkPubsub(c, cmd) {
|
|
return
|
|
}
|
|
|
|
key, items := args[0], args[1:]
|
|
|
|
withTx(m, c, func(c *server.Peer, ctx *connCtx) {
|
|
db := m.db(ctx.selectedDB)
|
|
|
|
if db.exists(key) && db.t(key) != "hll" {
|
|
c.WriteError(ErrNotValidHllValue.Error())
|
|
return
|
|
}
|
|
|
|
altered := db.hllAdd(key, items...)
|
|
c.WriteInt(altered)
|
|
})
|
|
}
|
|
|
|
// PFCOUNT
|
|
func (m *Miniredis) cmdPfcount(c *server.Peer, cmd string, args []string) {
|
|
if len(args) < 1 {
|
|
setDirty(c)
|
|
c.WriteError(errWrongNumber(cmd))
|
|
return
|
|
}
|
|
if !m.handleAuth(c) {
|
|
return
|
|
}
|
|
if m.checkPubsub(c, cmd) {
|
|
return
|
|
}
|
|
|
|
keys := args
|
|
|
|
withTx(m, c, func(c *server.Peer, ctx *connCtx) {
|
|
db := m.db(ctx.selectedDB)
|
|
|
|
count, err := db.hllCount(keys)
|
|
if err != nil {
|
|
c.WriteError(err.Error())
|
|
return
|
|
}
|
|
|
|
c.WriteInt(count)
|
|
})
|
|
}
|
|
|
|
// PFMERGE
|
|
func (m *Miniredis) cmdPfmerge(c *server.Peer, cmd string, args []string) {
|
|
if len(args) < 1 {
|
|
setDirty(c)
|
|
c.WriteError(errWrongNumber(cmd))
|
|
return
|
|
}
|
|
if !m.handleAuth(c) {
|
|
return
|
|
}
|
|
if m.checkPubsub(c, cmd) {
|
|
return
|
|
}
|
|
|
|
keys := args
|
|
|
|
withTx(m, c, func(c *server.Peer, ctx *connCtx) {
|
|
db := m.db(ctx.selectedDB)
|
|
|
|
if err := db.hllMerge(keys); err != nil {
|
|
c.WriteError(err.Error())
|
|
return
|
|
}
|
|
c.WriteOK()
|
|
})
|
|
}
|