Add withcount option

Author: ashtul

client.go

@@ -276,6 +276,12 @@ func (client *Client) TopkQuery(key string, items []string) ([]int64, error) {
 }
 
 // Return full list of items in Top K list.
+func (client *Client) TopkListWithCount(key string) (map[string]int64, error) {
+conn := client.Pool.Get()
+defer conn.Close()
+return ParseInfoReply(redis.Values(conn.Do("TOPK.LIST", key, "WITHCOUNT")))
+}
+
 func (client *Client) TopkList(key string) ([]string, error) {
 conn := client.Pool.Get()
 defer conn.Close()

client_test.go

@@ -228,6 +228,12 @@ func TestClient_TopkQuery(t *testing.T) {
 assert.Nil(t, err)
 assert.Equal(t, 3, len(keys))
 assert.Equal(t, []string{"A", "B", "E"}, keys)
+
+// WithCount option
+keysWithCount, err := client.TopkListWithCount(key1)
+assert.Nil(t, err)
+assert.Equal(t, 3, len(keysWithCount))
+assert.Equal(t, map[string]int64{"A": 4, "B": 3, "E": 3}, keysWithCount)
 }
 
 func TestClient_TopkInfo(t *testing.T) {