Rework ZRangeWithScores.

This commit is contained in:
Vladimir Mihailenco 2014-07-05 13:46:27 +03:00
parent 00a131e3a9
commit dc9bffa57d
4 changed files with 124 additions and 139 deletions

View File

@ -21,7 +21,7 @@ var (
_ Cmder = (*StringSliceCmd)(nil) _ Cmder = (*StringSliceCmd)(nil)
_ Cmder = (*BoolSliceCmd)(nil) _ Cmder = (*BoolSliceCmd)(nil)
_ Cmder = (*StringStringMapCmd)(nil) _ Cmder = (*StringStringMapCmd)(nil)
_ Cmder = (*StringFloatMapCmd)(nil) _ Cmder = (*ZSliceCmd)(nil)
_ Cmder = (*ScanCmd)(nil) _ Cmder = (*ScanCmd)(nil)
) )
@ -493,37 +493,37 @@ func (cmd *StringStringMapCmd) parseReply(rd *bufio.Reader) error {
//------------------------------------------------------------------------------ //------------------------------------------------------------------------------
type StringFloatMapCmd struct { type ZSliceCmd struct {
*baseCmd *baseCmd
val map[string]float64 val []Z
} }
func NewStringFloatMapCmd(args ...string) *StringFloatMapCmd { func NewZSliceCmd(args ...string) *ZSliceCmd {
return &StringFloatMapCmd{ return &ZSliceCmd{
baseCmd: newBaseCmd(args...), baseCmd: newBaseCmd(args...),
} }
} }
func (cmd *StringFloatMapCmd) Val() map[string]float64 { func (cmd *ZSliceCmd) Val() []Z {
return cmd.val return cmd.val
} }
func (cmd *StringFloatMapCmd) Result() (map[string]float64, error) { func (cmd *ZSliceCmd) Result() ([]Z, error) {
return cmd.val, cmd.err return cmd.val, cmd.err
} }
func (cmd *StringFloatMapCmd) String() string { func (cmd *ZSliceCmd) String() string {
return cmdString(cmd, cmd.val) return cmdString(cmd, cmd.val)
} }
func (cmd *StringFloatMapCmd) parseReply(rd *bufio.Reader) error { func (cmd *ZSliceCmd) parseReply(rd *bufio.Reader) error {
v, err := parseReply(rd, parseStringFloatMap) v, err := parseReply(rd, parseZSlice)
if err != nil { if err != nil {
cmd.err = err cmd.err = err
return err return err
} }
cmd.val = v.(map[string]float64) cmd.val = v.([]Z)
return nil return nil
} }

View File

@ -854,11 +854,7 @@ func (c *Client) ZRange(key string, start, stop int64) *StringSliceCmd {
return c.zRange(key, start, stop, false) return c.zRange(key, start, stop, false)
} }
func (c *Client) ZRangeWithScores(key string, start, stop int64) *StringSliceCmd { func (c *Client) ZRangeWithScores(key string, start, stop int64) *ZSliceCmd {
return c.zRange(key, start, stop, true)
}
func (c *Client) ZRangeWithScoresMap(key string, start, stop int64) *StringFloatMapCmd {
args := []string{ args := []string{
"ZRANGE", "ZRANGE",
key, key,
@ -866,7 +862,7 @@ func (c *Client) ZRangeWithScoresMap(key string, start, stop int64) *StringFloat
strconv.FormatInt(stop, 10), strconv.FormatInt(stop, 10),
"WITHSCORES", "WITHSCORES",
} }
cmd := NewStringFloatMapCmd(args...) cmd := NewZSliceCmd(args...)
c.Process(cmd) c.Process(cmd)
return cmd return cmd
} }
@ -899,11 +895,7 @@ func (c *Client) ZRangeByScore(key string, opt ZRangeByScore) *StringSliceCmd {
return c.zRangeByScore(key, opt, false) return c.zRangeByScore(key, opt, false)
} }
func (c *Client) ZRangeByScoreWithScores(key string, opt ZRangeByScore) *StringSliceCmd { func (c *Client) ZRangeByScoreWithScores(key string, opt ZRangeByScore) *ZSliceCmd {
return c.zRangeByScore(key, opt, true)
}
func (c *Client) ZRangeByScoreWithScoresMap(key string, opt ZRangeByScore) *StringFloatMapCmd {
args := []string{"ZRANGEBYSCORE", key, opt.Min, opt.Max, "WITHSCORES"} args := []string{"ZRANGEBYSCORE", key, opt.Min, opt.Max, "WITHSCORES"}
if opt.Offset != 0 || opt.Count != 0 { if opt.Offset != 0 || opt.Count != 0 {
args = append( args = append(
@ -913,7 +905,7 @@ func (c *Client) ZRangeByScoreWithScoresMap(key string, opt ZRangeByScore) *Stri
strconv.FormatInt(opt.Count, 10), strconv.FormatInt(opt.Count, 10),
) )
} }
cmd := NewStringFloatMapCmd(args...) cmd := NewZSliceCmd(args...)
c.Process(cmd) c.Process(cmd)
return cmd return cmd
} }
@ -962,13 +954,9 @@ func (c *Client) ZRevRange(key, start, stop string) *StringSliceCmd {
return c.zRevRange(key, start, stop, false) return c.zRevRange(key, start, stop, false)
} }
func (c *Client) ZRevRangeWithScores(key, start, stop string) *StringSliceCmd { func (c *Client) ZRevRangeWithScores(key, start, stop string) *ZSliceCmd {
return c.zRevRange(key, start, stop, true)
}
func (c *Client) ZRevRangeWithScoresMap(key, start, stop string) *StringFloatMapCmd {
args := []string{"ZREVRANGE", key, start, stop, "WITHSCORES"} args := []string{"ZREVRANGE", key, start, stop, "WITHSCORES"}
cmd := NewStringFloatMapCmd(args...) cmd := NewZSliceCmd(args...)
c.Process(cmd) c.Process(cmd)
return cmd return cmd
} }
@ -995,11 +983,7 @@ func (c *Client) ZRevRangeByScore(key string, opt ZRangeByScore) *StringSliceCmd
return c.zRevRangeByScore(key, opt, false) return c.zRevRangeByScore(key, opt, false)
} }
func (c *Client) ZRevRangeByScoreWithScores(key string, opt ZRangeByScore) *StringSliceCmd { func (c *Client) ZRevRangeByScoreWithScores(key string, opt ZRangeByScore) *ZSliceCmd {
return c.zRevRangeByScore(key, opt, true)
}
func (c *Client) ZRevRangeByScoreWithScoresMap(key string, opt ZRangeByScore) *StringFloatMapCmd {
args := []string{"ZREVRANGEBYSCORE", key, opt.Max, opt.Min, "WITHSCORES"} args := []string{"ZREVRANGEBYSCORE", key, opt.Max, opt.Min, "WITHSCORES"}
if opt.Offset != 0 || opt.Count != 0 { if opt.Offset != 0 || opt.Count != 0 {
args = append( args = append(
@ -1009,7 +993,7 @@ func (c *Client) ZRevRangeByScoreWithScoresMap(key string, opt ZRangeByScore) *S
strconv.FormatInt(opt.Count, 10), strconv.FormatInt(opt.Count, 10),
) )
} }
cmd := NewStringFloatMapCmd(args...) cmd := NewZSliceCmd(args...)
c.Process(cmd) c.Process(cmd)
return cmd return cmd
} }

View File

@ -11,8 +11,7 @@ import (
type multiBulkParser func(rd *bufio.Reader, n int64) (interface{}, error) type multiBulkParser func(rd *bufio.Reader, n int64) (interface{}, error)
var ( var (
errReaderTooSmall = errors.New("redis: reader is too small") errReaderTooSmall = errors.New("redis: reader is too small")
errInvalidReplyType = errors.New("redis: invalid reply type")
) )
//------------------------------------------------------------------------------ //------------------------------------------------------------------------------
@ -175,15 +174,15 @@ func parseSlice(rd *bufio.Reader, n int64) (interface{}, error) {
func parseStringSlice(rd *bufio.Reader, n int64) (interface{}, error) { func parseStringSlice(rd *bufio.Reader, n int64) (interface{}, error) {
vals := make([]string, 0, n) vals := make([]string, 0, n)
for i := int64(0); i < n; i++ { for i := int64(0); i < n; i++ {
vi, err := parseReply(rd, nil) viface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if v, ok := vi.(string); ok { v, ok := viface.(string)
vals = append(vals, v) if !ok {
} else { return nil, fmt.Errorf("got %T, expected string", viface)
return nil, errInvalidReplyType
} }
vals = append(vals, v)
} }
return vals, nil return vals, nil
} }
@ -191,15 +190,15 @@ func parseStringSlice(rd *bufio.Reader, n int64) (interface{}, error) {
func parseBoolSlice(rd *bufio.Reader, n int64) (interface{}, error) { func parseBoolSlice(rd *bufio.Reader, n int64) (interface{}, error) {
vals := make([]bool, 0, n) vals := make([]bool, 0, n)
for i := int64(0); i < n; i++ { for i := int64(0); i < n; i++ {
vi, err := parseReply(rd, nil) viface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
if v, ok := vi.(int64); ok { v, ok := viface.(int64)
vals = append(vals, v == 1) if !ok {
} else { return nil, fmt.Errorf("got %T, expected int64", viface)
return nil, errInvalidReplyType
} }
vals = append(vals, v == 1)
} }
return vals, nil return vals, nil
} }
@ -207,22 +206,22 @@ func parseBoolSlice(rd *bufio.Reader, n int64) (interface{}, error) {
func parseStringStringMap(rd *bufio.Reader, n int64) (interface{}, error) { func parseStringStringMap(rd *bufio.Reader, n int64) (interface{}, error) {
m := make(map[string]string, n/2) m := make(map[string]string, n/2)
for i := int64(0); i < n; i += 2 { for i := int64(0); i < n; i += 2 {
keyI, err := parseReply(rd, nil) keyiface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
key, ok := keyI.(string) key, ok := keyiface.(string)
if !ok { if !ok {
return nil, errInvalidReplyType return nil, fmt.Errorf("got %T, expected string", keyiface)
} }
valueI, err := parseReply(rd, nil) valueiface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
value, ok := valueI.(string) value, ok := valueiface.(string)
if !ok { if !ok {
return nil, errInvalidReplyType return nil, fmt.Errorf("got %T, expected string", valueiface)
} }
m[key] = value m[key] = value
@ -230,32 +229,34 @@ func parseStringStringMap(rd *bufio.Reader, n int64) (interface{}, error) {
return m, nil return m, nil
} }
func parseStringFloatMap(rd *bufio.Reader, n int64) (interface{}, error) { func parseZSlice(rd *bufio.Reader, n int64) (interface{}, error) {
m := make(map[string]float64, n/2) zz := make([]Z, n/2)
for i := int64(0); i < n; i += 2 { for i := int64(0); i < n; i += 2 {
keyI, err := parseReply(rd, nil) z := &zz[i/2]
if err != nil {
return nil, err
}
key, ok := keyI.(string)
if !ok {
return nil, errInvalidReplyType
}
valueI, err := parseReply(rd, nil) memberiface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
valueS, ok := valueI.(string) member, ok := memberiface.(string)
if !ok { if !ok {
return nil, errInvalidReplyType return nil, fmt.Errorf("got %T, expected string", memberiface)
} }
value, err := strconv.ParseFloat(valueS, 64) z.Member = member
scoreiface, err := parseReply(rd, nil)
if err != nil { if err != nil {
return nil, err return nil, err
} }
scorestr, ok := scoreiface.(string)
m[key] = value if !ok {
return nil, fmt.Errorf("got %T, expected string", scoreiface)
}
score, err := strconv.ParseFloat(scorestr, 64)
if err != nil {
return nil, err
}
z.Score = score
} }
return m, nil return zz, nil
} }

View File

@ -1984,9 +1984,9 @@ func (t *RedisTest) TestZAdd(c *C) {
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
c.Assert(zAdd.Val(), Equals, int64(0)) c.Assert(zAdd.Val(), Equals, int64(0))
zRange := t.client.ZRangeWithScores("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"one", "1", "uno", "1", "two", "3"}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}, {1, "uno"}, {3, "two"}})
} }
func (t *RedisTest) TestZCard(c *C) { func (t *RedisTest) TestZCard(c *C) {
@ -2027,9 +2027,9 @@ func (t *RedisTest) TestZIncrBy(c *C) {
c.Assert(zIncrBy.Err(), IsNil) c.Assert(zIncrBy.Err(), IsNil)
c.Assert(zIncrBy.Val(), Equals, float64(3)) c.Assert(zIncrBy.Val(), Equals, float64(3))
zRange := t.client.ZRangeWithScores("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"two", "2", "one", "3"}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}, {3, "one"}})
} }
func (t *RedisTest) TestZInterStore(c *C) { func (t *RedisTest) TestZInterStore(c *C) {
@ -2050,9 +2050,9 @@ func (t *RedisTest) TestZInterStore(c *C) {
c.Assert(zInterStore.Err(), IsNil) c.Assert(zInterStore.Err(), IsNil)
c.Assert(zInterStore.Val(), Equals, int64(2)) c.Assert(zInterStore.Val(), Equals, int64(2))
zRange := t.client.ZRangeWithScores("out", 0, -1) val, err := t.client.ZRangeWithScores("out", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"one", "5", "two", "10"}) c.Assert(val, DeepEquals, []redis.Z{{5, "one"}, {10, "two"}})
} }
func (t *RedisTest) TestZRange(c *C) { func (t *RedisTest) TestZRange(c *C) {
@ -2076,7 +2076,7 @@ func (t *RedisTest) TestZRange(c *C) {
c.Assert(zRange.Val(), DeepEquals, []string{"two", "three"}) c.Assert(zRange.Val(), DeepEquals, []string{"two", "three"})
} }
func (t *RedisTest) TestZRangeWithScoresMap(c *C) { func (t *RedisTest) TestZRangeWithScores(c *C) {
zAdd := t.client.ZAdd("zset", redis.Z{1, "one"}) zAdd := t.client.ZAdd("zset", redis.Z{1, "one"})
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
zAdd = t.client.ZAdd("zset", redis.Z{2, "two"}) zAdd = t.client.ZAdd("zset", redis.Z{2, "two"})
@ -2084,17 +2084,17 @@ func (t *RedisTest) TestZRangeWithScoresMap(c *C) {
zAdd = t.client.ZAdd("zset", redis.Z{3, "three"}) zAdd = t.client.ZAdd("zset", redis.Z{3, "three"})
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
zRange := t.client.ZRangeWithScoresMap("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, map[string]float64{"one": 1, "two": 2, "three": 3}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}, {2, "two"}, {3, "three"}})
zRange = t.client.ZRangeWithScoresMap("zset", 2, 3) val, err = t.client.ZRangeWithScores("zset", 2, 3).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, map[string]float64{"three": 3}) c.Assert(val, DeepEquals, []redis.Z{{3, "three"}})
zRange = t.client.ZRangeWithScoresMap("zset", -2, -1) val, err = t.client.ZRangeWithScores("zset", -2, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, map[string]float64{"two": 2, "three": 3}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}, {3, "three"}})
} }
func (t *RedisTest) TestZRangeByScore(c *C) { func (t *RedisTest) TestZRangeByScore(c *C) {
@ -2142,33 +2142,33 @@ func (t *RedisTest) TestZRangeByScoreWithScoresMap(c *C) {
zAdd = t.client.ZAdd("zset", redis.Z{3, "three"}) zAdd = t.client.ZAdd("zset", redis.Z{3, "three"})
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
zRangeByScore := t.client.ZRangeByScoreWithScoresMap("zset", redis.ZRangeByScore{ val, err := t.client.ZRangeByScoreWithScores("zset", redis.ZRangeByScore{
Min: "-inf", Min: "-inf",
Max: "+inf", Max: "+inf",
}) }).Result()
c.Assert(zRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRangeByScore.Val(), DeepEquals, map[string]float64{"one": 1, "two": 2, "three": 3}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}, {2, "two"}, {3, "three"}})
zRangeByScore = t.client.ZRangeByScoreWithScoresMap("zset", redis.ZRangeByScore{ val, err = t.client.ZRangeByScoreWithScores("zset", redis.ZRangeByScore{
Min: "1", Min: "1",
Max: "2", Max: "2",
}) }).Result()
c.Assert(zRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRangeByScore.Val(), DeepEquals, map[string]float64{"one": 1, "two": 2}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}, {2, "two"}})
zRangeByScore = t.client.ZRangeByScoreWithScoresMap("zset", redis.ZRangeByScore{ val, err = t.client.ZRangeByScoreWithScores("zset", redis.ZRangeByScore{
Min: "(1", Min: "(1",
Max: "2", Max: "2",
}) }).Result()
c.Assert(zRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRangeByScore.Val(), DeepEquals, map[string]float64{"two": 2}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}})
zRangeByScore = t.client.ZRangeByScoreWithScoresMap("zset", redis.ZRangeByScore{ val, err = t.client.ZRangeByScoreWithScores("zset", redis.ZRangeByScore{
Min: "(1", Min: "(1",
Max: "(2", Max: "(2",
}) }).Result()
c.Assert(zRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRangeByScore.Val(), DeepEquals, map[string]float64{}) c.Assert(val, DeepEquals, []redis.Z{})
} }
func (t *RedisTest) TestZRank(c *C) { func (t *RedisTest) TestZRank(c *C) {
@ -2200,9 +2200,9 @@ func (t *RedisTest) TestZRem(c *C) {
c.Assert(zRem.Err(), IsNil) c.Assert(zRem.Err(), IsNil)
c.Assert(zRem.Val(), Equals, int64(1)) c.Assert(zRem.Val(), Equals, int64(1))
zRange := t.client.ZRangeWithScores("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"one", "1", "three", "3"}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}, {3, "three"}})
} }
func (t *RedisTest) TestZRemRangeByRank(c *C) { func (t *RedisTest) TestZRemRangeByRank(c *C) {
@ -2217,9 +2217,9 @@ func (t *RedisTest) TestZRemRangeByRank(c *C) {
c.Assert(zRemRangeByRank.Err(), IsNil) c.Assert(zRemRangeByRank.Err(), IsNil)
c.Assert(zRemRangeByRank.Val(), Equals, int64(2)) c.Assert(zRemRangeByRank.Val(), Equals, int64(2))
zRange := t.client.ZRangeWithScores("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"three", "3"}) c.Assert(val, DeepEquals, []redis.Z{{3, "three"}})
} }
func (t *RedisTest) TestZRemRangeByScore(c *C) { func (t *RedisTest) TestZRemRangeByScore(c *C) {
@ -2234,9 +2234,9 @@ func (t *RedisTest) TestZRemRangeByScore(c *C) {
c.Assert(zRemRangeByScore.Err(), IsNil) c.Assert(zRemRangeByScore.Err(), IsNil)
c.Assert(zRemRangeByScore.Val(), Equals, int64(1)) c.Assert(zRemRangeByScore.Val(), Equals, int64(1))
zRange := t.client.ZRangeWithScores("zset", 0, -1) val, err := t.client.ZRangeWithScores("zset", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"two", "2", "three", "3"}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}, {3, "three"}})
} }
func (t *RedisTest) TestZRevRange(c *C) { func (t *RedisTest) TestZRevRange(c *C) {
@ -2268,17 +2268,17 @@ func (t *RedisTest) TestZRevRangeWithScoresMap(c *C) {
zAdd = t.client.ZAdd("zset", redis.Z{3, "three"}) zAdd = t.client.ZAdd("zset", redis.Z{3, "three"})
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
zRevRange := t.client.ZRevRangeWithScoresMap("zset", "0", "-1") val, err := t.client.ZRevRangeWithScores("zset", "0", "-1").Result()
c.Assert(zRevRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRange.Val(), DeepEquals, map[string]float64{"three": 3, "two": 2, "one": 1}) c.Assert(val, DeepEquals, []redis.Z{{3, "three"}, {2, "two"}, {1, "one"}})
zRevRange = t.client.ZRevRangeWithScoresMap("zset", "2", "3") val, err = t.client.ZRevRangeWithScores("zset", "2", "3").Result()
c.Assert(zRevRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRange.Val(), DeepEquals, map[string]float64{"one": 1}) c.Assert(val, DeepEquals, []redis.Z{{1, "one"}})
zRevRange = t.client.ZRevRangeWithScoresMap("zset", "-2", "-1") val, err = t.client.ZRevRangeWithScores("zset", "-2", "-1").Result()
c.Assert(zRevRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRange.Val(), DeepEquals, map[string]float64{"two": 2, "one": 1}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}, {1, "one"}})
} }
func (t *RedisTest) TestZRevRangeByScore(c *C) { func (t *RedisTest) TestZRevRangeByScore(c *C) {
@ -2316,7 +2316,7 @@ func (t *RedisTest) TestZRevRangeByScoreWithScores(c *C) {
vals, err := t.client.ZRevRangeByScoreWithScores( vals, err := t.client.ZRevRangeByScoreWithScores(
"zset", redis.ZRangeByScore{Max: "+inf", Min: "-inf"}).Result() "zset", redis.ZRangeByScore{Max: "+inf", Min: "-inf"}).Result()
c.Assert(err, IsNil) c.Assert(err, IsNil)
c.Assert(vals, DeepEquals, []string{"three", "3", "two", "2", "one", "1"}) c.Assert(vals, DeepEquals, []redis.Z{{3, "three"}, {2, "two"}, {1, "one"}})
} }
func (t *RedisTest) TestZRevRangeByScoreWithScoresMap(c *C) { func (t *RedisTest) TestZRevRangeByScoreWithScoresMap(c *C) {
@ -2327,20 +2327,20 @@ func (t *RedisTest) TestZRevRangeByScoreWithScoresMap(c *C) {
zAdd = t.client.ZAdd("zset", redis.Z{3, "three"}) zAdd = t.client.ZAdd("zset", redis.Z{3, "three"})
c.Assert(zAdd.Err(), IsNil) c.Assert(zAdd.Err(), IsNil)
zRevRangeByScore := t.client.ZRevRangeByScoreWithScoresMap( val, err := t.client.ZRevRangeByScoreWithScores(
"zset", redis.ZRangeByScore{Max: "+inf", Min: "-inf"}) "zset", redis.ZRangeByScore{Max: "+inf", Min: "-inf"}).Result()
c.Assert(zRevRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRangeByScore.Val(), DeepEquals, map[string]float64{"three": 3, "two": 2, "one": 1}) c.Assert(val, DeepEquals, []redis.Z{{3, "three"}, {2, "two"}, {1, "one"}})
zRevRangeByScore = t.client.ZRevRangeByScoreWithScoresMap( val, err = t.client.ZRevRangeByScoreWithScores(
"zset", redis.ZRangeByScore{Max: "2", Min: "(1"}) "zset", redis.ZRangeByScore{Max: "2", Min: "(1"}).Result()
c.Assert(zRevRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRangeByScore.Val(), DeepEquals, map[string]float64{"two": 2}) c.Assert(val, DeepEquals, []redis.Z{{2, "two"}})
zRevRangeByScore = t.client.ZRevRangeByScoreWithScoresMap( val, err = t.client.ZRevRangeByScoreWithScores(
"zset", redis.ZRangeByScore{Max: "(2", Min: "(1"}) "zset", redis.ZRangeByScore{Max: "(2", Min: "(1"}).Result()
c.Assert(zRevRangeByScore.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRevRangeByScore.Val(), DeepEquals, map[string]float64{}) c.Assert(val, DeepEquals, []redis.Z{})
} }
func (t *RedisTest) TestZRevRank(c *C) { func (t *RedisTest) TestZRevRank(c *C) {
@ -2387,9 +2387,9 @@ func (t *RedisTest) TestZUnionStore(c *C) {
c.Assert(zUnionStore.Err(), IsNil) c.Assert(zUnionStore.Err(), IsNil)
c.Assert(zUnionStore.Val(), Equals, int64(3)) c.Assert(zUnionStore.Val(), Equals, int64(3))
zRange := t.client.ZRangeWithScores("out", 0, -1) val, err := t.client.ZRangeWithScores("out", 0, -1).Result()
c.Assert(zRange.Err(), IsNil) c.Assert(err, IsNil)
c.Assert(zRange.Val(), DeepEquals, []string{"one", "5", "three", "9", "two", "10"}) c.Assert(val, DeepEquals, []redis.Z{{5, "one"}, {9, "three"}, {10, "two"}})
} }
//------------------------------------------------------------------------------ //------------------------------------------------------------------------------