mirror
/
redis
mirror of https://github.com/go-redis/redis.git
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

Support multiple keys for the PFCOUNT command

This commit is contained in:
Rich Hong 2015-12-23 17:24:42 -05:00
parent f6d6826d82
commit a6da93713a
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
commands.go
View File

@ -1340,8 +1340,13 @@ func (c *commandable) PFAdd(key string, fields ...string) *IntCmd {
return cmd return cmd
} }
func (c *commandable) PFCount(key string) *IntCmd { func (c *commandable) PFCount(keys ...string) *IntCmd {
cmd := NewIntCmd("PFCOUNT", key) args := make([]interface{}, 1+len(keys))
args[0] = "PFCOUNT"
for i, key := range keys {
args[1+i] = key
}
cmd := NewIntCmd(args...)
c.Process(cmd) c.Process(cmd)
return cmd return cmd
} }

4
commands_test.go
View File

@ -1203,6 +1203,10 @@ var _ = Describe("Commands", func() {
pfCount = client.PFCount("hllMerged") pfCount = client.PFCount("hllMerged")
Expect(pfCount.Err()).NotTo(HaveOccurred()) Expect(pfCount.Err()).NotTo(HaveOccurred())
Expect(pfCount.Val()).To(Equal(int64(10))) Expect(pfCount.Val()).To(Equal(int64(10)))
pfCount = client.PFCount("hll1", "hll2")
Expect(pfCount.Err()).NotTo(HaveOccurred())
Expect(pfCount.Val()).To(Equal(int64(10)))
}) })
}) })