// ZSCORE func (m *Miniredis) cmdZscore(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] member := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteNil() return } if db.t(key) != "zset" { out.WriteErrorString(ErrWrongType.Error()) return } if !db.ssetExists(key, member) { out.WriteNil() return } out.WriteString(formatFloat(db.ssetScore(key, member))) }) }
// RPOPLPUSH func (m *Miniredis) cmdRpoplpush(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } src := r.Args[0] dst := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(src) { out.WriteNil() return } if db.t(src) != "list" || (db.exists(dst) && db.t(dst) != "list") { out.WriteErrorString(msgWrongType) return } elem := db.listPop(src) db.listLpush(dst, elem) out.WriteString(elem) }) }
// RPUSHX func (m *Miniredis) cmdRpushx(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] value := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "list" { out.WriteErrorString(msgWrongType) return } newLen := db.listPush(key, value) out.WriteInt(newLen) }) }
// HMGET func (m *Miniredis) cmdHmget(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hmget' command") return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if t, ok := db.keys[key]; ok && t != "hash" { out.WriteErrorString(msgWrongType) return } f, ok := db.hashKeys[key] if !ok { f = map[string]string{} } out.WriteBulkLen(len(r.Args) - 1) for _, k := range r.Args[1:] { v, ok := f[k] if !ok { out.WriteNil() continue } out.WriteString(v) } }) }
// LLEN func (m *Miniredis) cmdLlen(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) t, ok := db.keys[key] if !ok { // No such key. That's zero length. out.WriteZero() return } if t != "list" { out.WriteErrorString(msgWrongType) return } out.WriteInt(len(db.listKeys[key])) }) }
// HEXISTS func (m *Miniredis) cmdHexists(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hexists' command") return nil } key := r.Args[0] field := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) t, ok := db.keys[key] if !ok { out.WriteInt(0) return } if t != "hash" { out.WriteErrorString(msgWrongType) return } if _, ok := db.hashKeys[key][field]; !ok { out.WriteInt(0) return } out.WriteInt(1) }) }
// HVALS func (m *Miniredis) cmdHvals(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hvals' command") return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) t, ok := db.keys[key] if !ok { out.WriteBulkLen(0) return } if t != "hash" { out.WriteErrorString(msgWrongType) return } out.WriteBulkLen(len(db.hashKeys[key])) for _, v := range db.hashKeys[key] { out.WriteString(v) } }) }
// LPUSHX func (m *Miniredis) cmdLpushx(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'lpushx' command") return nil } key := r.Args[0] value := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "list" { out.WriteErrorString(msgWrongType) return } newLen := db.listLpush(key, value) out.WriteInt(newLen) }) }
// PSETEX func (m *Miniredis) cmdPsetex(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 3 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] ttl, err := strconv.Atoi(r.Args[1]) if err != nil { setDirty(r.Client()) out.WriteErrorString(msgInvalidInt) return nil } value := r.Args[2] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) db.del(key, true) // Clear any existing keys. db.stringSet(key, value) db.expire[key] = ttl // We put millisecond keys in with the second keys. out.WriteOK() }) }
// SDIFF func (m *Miniredis) cmdSdiff(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 1 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } keys := r.Args return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) set, err := db.setDiff(keys) if err != nil { out.WriteErrorString(err.Error()) return } out.WriteBulkLen(len(set)) for k := range set { out.WriteString(k) } }) }
// LPOP func (m *Miniredis) cmdLpop(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'lpop' command") return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { // Non-existing key is fine. out.WriteNil() return } if db.t(key) != "list" { out.WriteErrorString(msgWrongType) return } elem := db.listLpop(key) out.WriteString(elem) }) }
// SCARD func (m *Miniredis) cmdScard(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "set" { out.WriteErrorString(ErrWrongType.Error()) return } members := db.setMembers(key) out.WriteInt(len(members)) }) }
// SUNIONSTORE func (m *Miniredis) cmdSunionstore(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } dest := r.Args[0] keys := r.Args[1:] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) set, err := db.setUnion(keys) if err != nil { out.WriteErrorString(err.Error()) return } db.del(dest, true) db.setSet(dest, set) out.WriteInt(len(set)) }) }
// SREM func (m *Miniredis) cmdSrem(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] fields := r.Args[1:] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteInt(0) return } if db.t(key) != "set" { out.WriteErrorString(ErrWrongType.Error()) return } out.WriteInt(db.setRem(key, fields...)) }) }
// SREM func (m *Miniredis) cmdSrem(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'srem' command") return nil } key := r.Args[0] fields := r.Args[1:] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteInt(0) return } if db.t(key) != "set" { out.WriteErrorString(ErrWrongType.Error()) return } out.WriteInt(db.setRem(key, fields...)) }) }
// MSET func (m *Miniredis) cmdMset(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } if len(r.Args)%2 != 0 { setDirty(r.Client()) // non-default error message out.WriteErrorString("ERR wrong number of arguments for MSET") return nil } return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) for len(r.Args) > 0 { key := r.Args[0] value := r.Args[1] r.Args = r.Args[2:] db.del(key, true) // clear TTL db.stringSet(key, value) } out.WriteOK() }) }
// SCARD func (m *Miniredis) cmdScard(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'scard' command") return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "set" { out.WriteErrorString(ErrWrongType.Error()) return } members := db.setMembers(key) out.WriteInt(len(members)) }) }
// GETSET func (m *Miniredis) cmdGetset(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] value := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if t, ok := db.keys[key]; ok && t != "string" { out.WriteErrorString(msgWrongType) return } old, ok := db.stringKeys[key] db.stringSet(key, value) // a GETSET clears the expire delete(db.expire, key) if !ok { out.WriteNil() return } out.WriteString(old) return }) }
// HKEYS func (m *Miniredis) cmdHkeys(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hkeys' command") return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteBulkLen(0) return } if db.t(key) != "hash" { out.WriteErrorString(msgWrongType) return } fields := db.hashFields(key) out.WriteBulkLen(len(fields)) for _, f := range fields { out.WriteString(f) } }) }
// DECR func (m *Miniredis) cmdDecr(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) key := r.Args[0] if t, ok := db.keys[key]; ok && t != "string" { out.WriteErrorString(msgWrongType) return } v, err := db.stringIncr(key, -1) if err != nil { out.WriteErrorString(err.Error()) return } // Don't touch TTL out.WriteInt(v) }) }
// HSET func (m *Miniredis) cmdHset(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 3 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hset' command") return nil } key := r.Args[0] field := r.Args[1] value := r.Args[2] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if t, ok := db.keys[key]; ok && t != "hash" { out.WriteErrorString(msgWrongType) return } if db.hashSet(key, field, value) { out.WriteZero() } else { out.WriteOne() } }) }
// APPEND func (m *Miniredis) cmdAppend(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] value := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if t, ok := db.keys[key]; ok && t != "string" { out.WriteErrorString(msgWrongType) return } newValue := db.stringKeys[key] + value db.stringSet(key, newValue) out.WriteInt(len(newValue)) }) }
// HSETNX func (m *Miniredis) cmdHsetnx(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 3 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'hsetnx' command") return nil } key := r.Args[0] field := r.Args[1] value := r.Args[2] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if t, ok := db.keys[key]; ok && t != "hash" { out.WriteErrorString(msgWrongType) return } if _, ok := db.hashKeys[key]; !ok { db.hashKeys[key] = map[string]string{} db.keys[key] = "hash" } _, ok := db.hashKeys[key][field] if ok { out.WriteZero() return } db.hashKeys[key][field] = value db.keyVersion[key]++ out.WriteOne() }) }
// MOVE func (m *Miniredis) cmdMove(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] targetDB, err := strconv.Atoi(r.Args[1]) if err != nil { targetDB = 0 } return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { if ctx.selectedDB == targetDB { out.WriteErrorString("ERR source and destination objects are the same") return } db := m.db(ctx.selectedDB) targetDB := m.db(targetDB) if !db.move(key, targetDB) { out.WriteZero() return } out.WriteOne() }) }
// RPOP func (m *Miniredis) cmdRpop(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 1 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteNil() return } if db.t(key) != "list" { out.WriteErrorString(msgWrongType) return } elem := db.listPop(key) out.WriteString(elem) }) }
// RENAMENX func (m *Miniredis) cmdRenamenx(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } from := r.Args[0] to := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(from) { out.WriteErrorString(msgKeyNotFound) return } if db.exists(to) { out.WriteZero() return } db.rename(from, to) out.WriteOne() }) }
// RPUSH func (m *Miniredis) cmdRpush(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] args := r.Args[1:] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if db.exists(key) && db.t(key) != "list" { out.WriteErrorString(msgWrongType) return } var newLen int for _, value := range args { newLen = db.listPush(key, value) } out.WriteInt(newLen) }) }
// SISMEMBER func (m *Miniredis) cmdSismember(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 2 { setDirty(r.Client()) out.WriteErrorString("ERR wrong number of arguments for 'sismember' command") return nil } key := r.Args[0] value := r.Args[1] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "set" { out.WriteErrorString(ErrWrongType.Error()) return } if db.setIsMember(key, value) { out.WriteOne() return } out.WriteZero() }) }
// LINSERT func (m *Miniredis) cmdLinsert(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) != 4 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] where := 0 switch strings.ToLower(r.Args[1]) { case "before": where = -1 case "after": where = +1 default: setDirty(r.Client()) out.WriteErrorString(msgSyntaxError) return nil } pivot := r.Args[2] value := r.Args[3] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) t, ok := db.keys[key] if !ok { // No such key out.WriteZero() return } if t != "list" { out.WriteErrorString(msgWrongType) return } l := db.listKeys[key] for i, el := range l { if el != pivot { continue } if where < 0 { l = append(l[:i], append(listKey{value}, l[i:]...)...) } else { if i == len(l)-1 { l = append(l, value) } else { l = append(l[:i+1], append(listKey{value}, l[i+1:]...)...) } } db.listKeys[key] = l db.keyVersion[key]++ out.WriteInt(len(l)) return } out.WriteInt(-1) }) }
// ZREM func (m *Miniredis) cmdZrem(out *redeo.Responder, r *redeo.Request) error { if len(r.Args) < 2 { setDirty(r.Client()) return r.WrongNumberOfArgs() } if !m.handleAuth(r.Client(), out) { return nil } key := r.Args[0] members := r.Args[1:] return withTx(m, out, r, func(out *redeo.Responder, ctx *connCtx) { db := m.db(ctx.selectedDB) if !db.exists(key) { out.WriteZero() return } if db.t(key) != "zset" { out.WriteErrorString(ErrWrongType.Error()) return } deleted := 0 for _, member := range members { if db.ssetRem(key, member) { deleted++ } } out.WriteInt(deleted) }) }