Add Support for TOPK.INCRBY

Author: dengliming

README.md

@@ -102,7 +102,7 @@ func main() {
 | :---          |  ----: |
 | [TOPK.RESERVE](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkreserve) |  [TopkReserve](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkReserve)  |
 | [TOPK.ADD](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkadd) |   [TopkAdd](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkAdd)  |
-| [TOPK.INCRBY](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkincrby) |  N/A  |
+| [TOPK.INCRBY](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkincrby) |  [TopkIncrby](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkIncrby)  |
 | [TOPK.QUERY](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkquery) |   [TopkQuery](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkQuery)  |
 | [TOPK.COUNT](https://oss.redislabs.com/redisbloom/TopK_Commands/#topkcount) |   [TopkCount](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkCount)  |
 | [TOPK.LIST](https://oss.redislabs.com/redisbloom/TopK_Commands/#topklist) |   [TopkList](https://godoc.org/github.com/RedisBloom/redisbloom-go#Client.TopkList)  |

client.go

@@ -204,3 +204,15 @@ func (client *Client) TopkInfo(key string) (map[string]string, error) {
 	}
 	return m, err
 }
+
+// Increase the score of an item in the data structure by increment.
+func (client *Client) TopkIncrby(key string, itemIncrements map[string]int64) ([]string, error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	args := redis.Args{key}
+	for k, v := range itemIncrements {
+		args = args.Add(k, v)
+	}
+	reply, err := conn.Do("TOPK.INCRBY", args...)
+	return redis.Strings(reply, err)
+}

client_test.go

@@ -190,3 +190,20 @@ func TestClient_TopkInfo(t *testing.T) {
 	info, err = client.TopkInfo("notexists")
 	assert.NotNil(t, err)
 }
+
+func TestClient_TopkIncrby(t *testing.T) {
+	client.FlushAll()
+	key := "test_topk_incrby"
+	ret, err := client.TopkReserve(key, 50, 2000, 7, 0.925)
+	assert.Nil(t, err)
+	assert.Equal(t, "OK", ret)
+
+	rets, err := client.TopkAdd(key, []string{"foo", "bar", "42"})
+	assert.Nil(t, err)
+	assert.NotNil(t, rets)
+
+	rets, err = client.TopkIncrby(key, map[string]int64{"foo": 3, "bar": 2, "42": 30})
+	assert.Nil(t, err)
+	assert.Equal(t, 3, len(rets))
+	assert.Equal(t, "", rets[2])
+}