Merge pull request #13 from dengliming/f_bf_command

Add Support for BF.MEXISTS BF.MADD

Author: ashtul

client.go

@@ -107,3 +107,27 @@ func (client *Client) Info(key string) (info map[string]int64, err error) {
 	}
 	return info, nil
 }
+
+// BfAddMulti - Adds one or more items to the Bloom Filter, creating the filter if it does not yet exist.
+// args:
+// key - the name of the filter
+// item - One or more items to add
+func (client *Client) BfAddMulti(key string, items []string) ([]int64, error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	args := redis.Args{key}.AddFlat(items)
+	result, err := conn.Do("BF.MADD", args...)
+	return redis.Int64s(result, err)
+}
+
+// BfExistsMulti - Determines if one or more items may exist in the filter or not.
+// args:
+// key - the name of the filter
+// item - one or more items to check
+func (client *Client) BfExistsMulti(key string, items []string) ([]int64, error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	args := redis.Args{key}.AddFlat(items)
+	result, err := conn.Do("BF.MEXISTS", args...)
+	return redis.Int64s(result, err)
+}

client_test.go

@@ -107,3 +107,25 @@ func TestExists(t *testing.T) {
 	assert.Nil(t, err)
 	assert.False(t, exists)
 }
+
+func TestClient_BfAddMulti(t *testing.T) {
+	client.FlushAll()
+	ret, err := client.BfAddMulti("test_add_multi", []string{"a", "b", "c"})
+	assert.Nil(t, err)
+	assert.NotNil(t, ret)
+}
+
+func TestClient_BfExistsMulti(t *testing.T) {
+	client.FlushAll()
+	key := "test_exists_multi"
+	ret, err := client.BfAddMulti(key, []string{"a", "b", "c"})
+	assert.Nil(t, err)
+	assert.NotNil(t, ret)
+
+	existsResult, err := client.BfExistsMulti(key, []string{"a", "b", "notexists"})
+	assert.Nil(t, err)
+	assert.Equal(t, 3, len(existsResult))
+	assert.Equal(t, int64(1), existsResult[0])
+	assert.Equal(t, int64(1), existsResult[1])
+	assert.Equal(t, int64(0), existsResult[2])
+}