Add a connection lock

Fixes #80
NoBrainerORM · Aug 15, 2014 · 5281d52 · 5281d52
1 parent 8bfb44f
commit 5281d52
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lib/no_brainer/query_runner.rb b/lib/no_brainer/query_runner.rb
@@ -11,7 +11,8 @@ def initialize(runner)
   end
 
   autoload :Driver, :DatabaseOnDemand, :TableOnDemand, :WriteError,
-           :Reconnect, :Selection, :RunOptions, :Logger, :MissingIndex
+           :Reconnect, :Selection, :RunOptions, :Logger, :MissingIndex,
+           :ConnectionLock
 
   class << self
     attr_accessor :stack
@@ -32,6 +33,7 @@ def run(*args, &block)
     use DatabaseOnDemand
     use TableOnDemand
     use Logger
+    use ConnectionLock
     use Reconnect
     use Driver
   end

diff --git a/lib/no_brainer/query_runner/connection_lock.rb b/lib/no_brainer/query_runner/connection_lock.rb
@@ -0,0 +1,7 @@
+class NoBrainer::QueryRunner::ConnectionLock < NoBrainer::QueryRunner::Middleware
+  @@lock = Mutex.new
+
+  def call(env)
+    @@lock.synchronize { @runner.call(env) }
+  end
+end