lib/pheme/queue_poller.rb in pheme-4.0.1 vs lib/pheme/queue_poller.rb in pheme-4.0.2

@@ -163,12 +163,12 @@
         Pheme.logger.info("Unsupported custom data type")
         value["Value"]
       end
     end
 
-    def with_optional_connection_pool_block
+    def with_optional_connection_pool_block(&block)
       if connection_pool_block
-        ActiveRecord::Base.connection_pool.with_connection { yield }
+        ActiveRecord::Base.connection_pool.with_connection(&block)
       else
         yield
       end
     end