Add pipeline support (#9)

Author: ashtul

redisbloom/client.py

@@ -1,6 +1,7 @@
 import six
 import redis
 from redis import Redis, RedisError 
+from redis.client import Pipeline
 from redis.client import bool_ok
 from redis._compat import (long, nativestr)
 from redis.exceptions import DataError
@@ -192,7 +193,6 @@ def appendBucketSize(params, bucket_size):
             params.extend(['BUCKETSIZE', bucket_size])
 
 ################## Bloom Filter Functions ######################
-
     def bfCreate(self, key, errorRate, capacity, expansion=None, noScale=None):
         """
         Creates a new Bloom Filter ``key`` with desired probability of false 
@@ -502,3 +502,21 @@ def topkInfo(self, key):
 
         return self.execute_command(self.TOPK_INFO, key)
 
+    def pipeline(self, transaction=True, shard_hint=None):
+        """
+        Return a new pipeline object that can queue multiple commands for
+        later execution. ``transaction`` indicates whether all commands
+        should be executed atomically. Apart from making a group of operations
+        atomic, pipelines are useful for reducing the back-and-forth overhead
+        between the client and server.
+        Overridden in order to provide the right client through the pipeline.
+        """
+        p = Pipeline(
+            connection_pool=self.connection_pool,
+            response_callbacks=self.response_callbacks,
+            transaction=transaction,
+            shard_hint=shard_hint)
+        return p
+
+class Pipeline(Pipeline, Client):
+    "Pipeline for RedisBloom Client"

test_commands.py

@@ -180,5 +180,22 @@ def testTopK(self):
         self.assertEqual(3, info.depth)  
         self.assertAlmostEqual(0.9, float(info.decay))  
 
+    def test_pipeline(self):
+        pipeline = rb.pipeline()
+        
+        self.assertFalse(rb.execute_command('get pipeline'))
+
+        self.assertTrue(rb.bfCreate('pipeline', 0.01, 1000))
+        for i in range(100):
+            pipeline.bfAdd('pipeline', i)
+        for i in range(100):
+            self.assertFalse(rb.bfExists('pipeline', i))
+
+        pipeline.execute()
+
+        for i in range(100):
+            self.assertTrue(rb.bfExists('pipeline', i))
+        
+
 if __name__ == '__main__':
     unittest.main()