Yield a connection to pipeline blocks.

Closes redis#205.
jodosha · Mar 30, 2012 · 1c68738 · 1c68738
1 parent 619256e
commit 1c68738
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/lib/redis.rb b/lib/redis.rb
@@ -1770,7 +1770,7 @@ def pipelined
     synchronize do |client|
       begin
         original, @client = @client, Pipeline.new
-        yield
+        yield(self)
         original.call_pipeline(@client)
       ensure
         @client = original

diff --git a/test/pipelining_commands_test.rb b/test/pipelining_commands_test.rb
@@ -184,3 +184,11 @@
 
   assert ["key"] == result.first
 end
+
+test "Pipeline yields a connection" do |r|
+  r.pipelined do |p|
+    p.set("foo", "bar")
+  end
+
+  assert_equal "bar", r.get("foo")
+end