n_plus_one_control 0.1.3 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f39c7dee63a881210d82f9459af15c81969f6346
-  data.tar.gz: 1a303ce2a4b52354cf3b4036c1f925c017308097
+  metadata.gz: 942639e351ee999dc4a113c237d9604878d296cc
+  data.tar.gz: 76d9b2b2fd28b02925ce4d219e3b5c6ac316d1fb
 SHA512:
-  metadata.gz: 4bd630b55700e20ba3cb00523d7e2b023819ffdfd802a1daae90e7532dfa86b8c4ac170378fde819b5810c694268df07d08b361c0fae84bbcbbfeff198a3e5fc
-  data.tar.gz: 6bc7a3c38fda5837c6f8a3e0cabc3a118ef0c95bb3ffc1066fae066fe9a86a7c14ac06876704326520594b91230b5e78c1015e00f6160ee1e5e3c58953e46396
+  metadata.gz: 980d3feab9ebd9d9b51da847526a83ae0ed2b36e50509494d14bfad75578086b4c2ca6a6da60d5f518a6a2deb87c80af848c792046804e96b62249404f09535c
+  data.tar.gz: ef2d79c707712f2a578ba4ff513c1f03b6c4cf351b72381875188ac6a3730c1f73b3fd11ab7a0538f5d76f2d67a5b6fb4f57775218e1ac8b1a0877e6936bd0a9

data/.rubocop.yml

@@ -34,6 +34,9 @@ Style/StringLiterals:
 Style/RegexpLiteral:
   Enabled: false
 
+Style/Lambda:
+  Enabled: false
+
 Style/SpaceInsideStringInterpolation:
   EnforcedStyle: no_space
 

data/README.md

@@ -167,6 +167,19 @@ NPlusOneControl.ignore = /^(BEGIN|COMMIT|SAVEPOINT|RELEASE)/
 # We track ActiveRecord event by default,
 # but can also track rom-rb events ('sql.rom') as well.
 NPlusOneControl.event = 'sql.active_record'
+
+# configure transactional behavour for populate method
+# in case of use multiple database connections
+NPlusOneControl::Executor.tap do |executor|
+  connections = ActiveRecord::Base.connection_handler.connection_pool_list.map(&:connection)
+
+  executor.transaction_begin = -> do
+    connections.each { |connection| connection.begin_transaction(joinable: false) }
+  end
+  executor.transaction_rollback = -> do
+    connections.each(&:rollback_transaction)
+  end
+end
 ```
 
 ## How does it work?
@@ -180,7 +193,7 @@ It may be useful to provide more matchers/assertions, for example:
 ```ruby
 
 # Actually, that means that it is N+1))
-assert_linear_number_of_queries { ... } 
+assert_linear_number_of_queries { ... }
 
 # But we can tune it with `coef` and handle such cases as selecting in batches
 assert_linear_number_of_queries(coef: 0.1) do
@@ -211,4 +224,3 @@ Bug reports and pull requests are welcome on GitHub at https://github.com/palkan
 ## License
 
 The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
-

data/lib/n_plus_one_control/executor.rb

@@ -26,6 +26,9 @@ module NPlusOneControl
     end
 
     class << self
+      attr_accessor :transaction_begin
+      attr_accessor :transaction_rollback
+
       def call(population:, scale_factors: nil, matching: nil)
         raise ArgumentError, "Block is required!" unless block_given?
 
@@ -44,12 +47,18 @@ module NPlusOneControl
       private
 
       def with_transaction
-        return yield unless defined?(ActiveRecord)
-        ActiveRecord::Base.connection.begin_transaction(joinable: false)
+        transaction_begin.call
         yield
       ensure
-        ActiveRecord::Base.connection.rollback_transaction
+        transaction_rollback.call
       end
     end
+
+    self.transaction_begin = -> do
+      ActiveRecord::Base.connection.begin_transaction(joinable: false)
+    end
+    self.transaction_rollback = -> do
+      ActiveRecord::Base.connection.rollback_transaction
+    end
   end
 end

data/lib/n_plus_one_control/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module NPlusOneControl
-  VERSION = "0.1.3"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: n_plus_one_control
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.2.1
 platform: ruby
 authors:
 - palkan
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-10-27 00:00:00.000000000 Z
+date: 2017-10-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler