Synchronize redis operations with task-aware latch.

celluloid · Aug 21, 2015 · 2b0508e · 2b0508e
1 parent bd7451d
commit 2b0508e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib/celluloid/redis.rb b/lib/celluloid/redis.rb
@@ -1,3 +1,19 @@
 require "redis"
 
 require "celluloid/redis/version"
+require "celluloid/io"
+
+module Celluloid
+  class Redis < ::Redis
+    VERSION = CelluloidRedis::VERSION
+
+    def synchronize
+      if @options[:driver] == :celluloid
+        @latch ||= Celluloid::IO::Stream::Latch.new
+        @latch.synchronize { yield @client }
+      else
+        super(&proc)
+      end
+    end
+  end
+end