add zset

Author: vgmdj

redis/global.go

@@ -83,3 +83,13 @@ func (c *Client) Rename(key string, newKeyName string) error {
 	return nil
 
 }
+
+//Rename rename the key
+func (c *Client) FlushAll() error {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	_, err := conn.Do("FLUSHALL")
+	return err
+
+}

redis/zset.go

@@ -0,0 +1,57 @@
+package redis
+
+import (
+	"github.com/gomodule/redigo/redis"
+)
+
+// ZADD add score and member to sorted set
+// if already exist, return count = 0 and update member's score
+// else return count = 1
+func (c *Client) ZAdd(key string, score int, member string) (count int64, err error) {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	return redis.Int64(conn.Do("ZADD", key, score, member))
+
+}
+
+// ZAddWithIncr incr score to member
+// return count = origin score + incr score
+// eg. already exist member = a, score = 10
+//     zdd [key] incr 10 a
+// return count = 20
+// now member = a , score  = 20
+func (c *Client) ZAddWithIncr(key string, score int, member string) (resultScore int64, err error) {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	return redis.Int64(conn.Do("ZADD", key, "INCR", score, member))
+
+}
+
+// ZRange list members from start to end
+func (c *Client) ZRange(key string, start, end int) ([]string, error) {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	return redis.Strings(conn.Do("ZRANGE", key, start, end))
+
+}
+
+// ZRange list members from start to end
+func (c *Client) ZRangeWithScore(key string, start, end int) ([]string, error) {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	return redis.Strings(conn.Do("ZRANGE", key, start, end, "WITHSCORES"))
+
+}
+
+// ZRank return the index of the member
+func (c *Client) ZRank(key string, member string) (int64, error) {
+	conn := c.pool.Get()
+	defer conn.Close()
+
+	return redis.Int64(conn.Do("ZRANK", key, member))
+
+}

redis/zset_test.go

@@ -0,0 +1,54 @@
+package redis
+
+import (
+	"testing"
+
+	"github.com/vgmdj/utils/logger"
+)
+
+func TestZSet(t *testing.T) {
+	c := testRedisClient()
+	c.FlushAll()
+
+	key := "TestSet"
+	_, err := c.ZAdd(key, 16, "a")
+	if err != nil {
+		t.Error(err.Error())
+		return
+	}
+
+	score, err := c.ZAddWithIncr(key, 4, "a")
+	if err != nil {
+		t.Error(err.Error())
+		return
+	}
+
+	if score != 20 {
+		t.Errorf("score should be 20, but get %d\n", score)
+		return
+	}
+
+	for i := 0; i < 10; i++ {
+		c.ZAdd(key, i+5, string('b'+i))
+	}
+
+	res, err := c.ZRangeWithScore(key, 0, 10)
+	if err != nil {
+		t.Error(err.Error())
+		return
+	}
+
+	t.Log(res)
+
+	index,err := c.ZRank(key,"f")
+	if err != nil {
+		logger.Error(err.Error())
+		return
+	}
+
+	if index != 4{
+		t.Errorf("index should be 4 , but get %d\n",index)
+		return
+	}
+
+}