Add Support for BF.INSERT

Author: dengliming

client.go

@@ -155,6 +155,30 @@ func (client *Client) BfLoadChunk(key string, iter int64, data []byte) (string,
 	return redis.String(conn.Do("BF.LOADCHUNK", key, iter, data))
 }
 
+// This command will add one or more items to the bloom filter, by default creating it if it does not yet exist.
+func (client *Client) BfInsert(key string, cap int64, errorRatio float64, expansion int64, noCreate bool, nonScaling bool, items []string) ([]int64, error) {
+	conn := client.Pool.Get()
+	defer conn.Close()
+	args := redis.Args{key}
+	if cap > 0 {
+		args = args.Add("CAPACITY", cap)
+	}
+	if errorRatio > 0 {
+		args = args.Add("ERROR", errorRatio)
+	}
+	if expansion > 0 {
+		args = args.Add("EXPANSION", expansion)
+	}
+	if noCreate {
+		args = args.Add("NOCREATE")
+	}
+	if nonScaling {
+		args = args.Add("NONSCALING")
+	}
+	args = args.Add("ITEMS").AddFlat(items)
+	return redis.Int64s(conn.Do("BF.INSERT", args...))
+}
+
 // Initializes a TopK with specified parameters.
 func (client *Client) TopkReserve(key string, topk int64, width int64, depth int64, decay float64) (string, error) {
 	conn := client.Pool.Get()

client_test.go

@@ -130,6 +130,19 @@ func TestClient_BfExistsMulti(t *testing.T) {
 	assert.Equal(t, int64(0), existsResult[2])
 }
 
+func TestClient_BfInsert(t *testing.T) {
+	client.FlushAll()
+	key := "test_bf_insert"
+	ret, err := client.BfInsert(key, 1000, 0.1, -1, false, false, []string{"a"})
+	assert.Nil(t, err)
+	assert.Equal(t, 1, len(ret))
+	assert.True(t, ret[0] > 0)
+	existsResult, err := client.BfExistsMulti(key, []string{"a"})
+	assert.Nil(t, err)
+	assert.Equal(t, 1, len(existsResult))
+	assert.Equal(t, int64(1), existsResult[0])
+}
+
 func TestClient_TopkReserve(t *testing.T) {
 	client.FlushAll()
 	ret, err := client.TopkReserve("test_topk_reserve", 10, 2000, 7, 0.925)