fibered_mysql2 0.4.0.pre.tstarck.2 → 0.4.0.pre.tstarck.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5804c1cde27f4f2a0d3c222c5f49a207930c140ebc911e71b0b4655925b83917
-  data.tar.gz: dd33b3c3dfba5b858fc2b85832b2d65650661e53513311f9b99a62089dade8c4
+  metadata.gz: 0fcddd19ebaa9bb4c2bcfba7bcb7862b9fb07a31ef4ecd122d46c345a563f8ec
+  data.tar.gz: 65666da2f04f234f15e8945f8b693341599c478a471756963091c19c6391f770
 SHA512:
-  metadata.gz: 72b39a4bb7c035e1ab5d7ea7929f51c08d7fd0f7cac223d97444c3352e7bcd0f078d7146a5e2a0f446c1bae69774b0e5246dd77d425a83530e92be02f4215579
-  data.tar.gz: e4ef21a8acd88205f612d0fe23924527d4ed65f78d54ec5a397317c27ab8a1454f17b9979b830a5ffe90ab2478851114305af3706c958f3cbd559ece21adeeb5
+  metadata.gz: fdcc099c7c98ec0860a198d7ede37859c1c060c791f2fa33acb8659b02fabdc64f675fcbdc3cb4b555a3f40a3466945b15ebe586a14311ff70eb443e92ad30be
+  data.tar.gz: 755c58904cdd0405c464ba49f662b00c1747a30c61672472a683d17a8c6432e20d3c8c695ca82f34d2d4797c66741d7381ead66084478d634a04f187112c1b02

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    fibered_mysql2 (0.4.0.pre.tstarck.2)
+    fibered_mysql2 (0.4.0.pre.tstarck.3)
       em-synchrony (~> 1.0)
       rails (>= 7.0, < 7.2)
 

data/lib/active_record/connection_adapters/fibered_mysql2_adapter.rb

@@ -54,6 +54,98 @@ module FiberedMysql2
       end
     end
 
+    def reset_transaction #:nodoc:
+      @transaction_manager = ::FiberedMysql2::FiberedMysql2Adapter_7_0::TransactionManager.new(self)
+    end
+
+    class TransactionManager < ::ActiveRecord::ConnectionAdapters::TransactionManager
+      def initialize(...)
+        super
+        @stack = Hash.new { |h, k| h[k] = [] }
+      end
+
+      def current_transaction #:nodoc:
+        _current_stack.last || ::ActiveRecord::ConnectionAdapters::TransactionManager::NULL_TRANSACTION
+      end
+
+      def open_transactions
+        _current_stack.size
+      end
+
+      def begin_transaction(isolation: nil, joinable: true, _lazy: true)
+        @connection.lock.synchronize do
+          run_commit_callbacks = !current_transaction.joinable?
+          transaction =
+            if _current_stack.empty?
+              ::ActiveRecord::ConnectionAdapters::RealTransaction.new(@connection, isolation:, joinable:, run_commit_callbacks: run_commit_callbacks)
+            else
+              ::ActiveRecord::ConnectionAdapters::SavepointTransaction.new(@connection, "active_record_#{Fiber.current.object_id}_#{open_transactions}", _current_stack.last, isolation:, joinable:, run_commit_callbacks: run_commit_callbacks)
+            end
+
+          if @connection.supports_lazy_transactions? && lazy_transactions_enabled? && _lazy
+            @has_unmaterialized_transactions = true
+          else
+            transaction.materialize!
+          end
+          _current_stack.push(transaction)
+          transaction
+        end
+      end
+
+      # Overriding the ActiveRecord::TransactionManager#materialize_transactions method to use
+      # fiber safe the _current_stack instead of the @stack instance variable. when marterializing
+      # transactions.
+      def materialize_transactions
+        return if @materializing_transactions
+        return unless @has_unmaterialized_transactions
+
+        @connection.lock.synchronize do
+          begin
+            @materializing_transactions = true
+            _current_stack.each { |t| t.materialize! unless t.materialized? }
+          ensure
+            @materializing_transactions = false
+          end
+          @has_unmaterialized_transactions = false
+        end
+      end
+
+      # Overriding the ActiveRecord::TransactionManager#commit_transaction method to use
+      # fiber safe the _current_stack instead of the @stack instance variable. when marterializing
+      # transactions.
+      def commit_transaction
+        @connection.lock.synchronize do
+          transaction = _current_stack.last
+
+          begin
+            transaction.before_commit_records
+          ensure
+            _current_stack.pop
+          end
+
+          transaction.commit
+          transaction.commit_records
+        end
+      end
+
+      # Overriding the ActiveRecord::TransactionManager#rollback_transaction method to use
+      # fiber safe the _current_stack instead of the @stack instance variable. when marterializing
+      # transactions.
+      def rollback_transaction(transaction = nil)
+        @connection.lock.synchronize do
+          transaction ||= _current_stack.pop
+          transaction.rollback
+          transaction.rollback_records
+        end
+      end
+
+      private
+
+      def _current_stack
+        @stack[Fiber.current.object_id]
+      end
+    end
+
     private
 
     def owner_fiber

data/lib/fibered_mysql2/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module FiberedMysql2
-  VERSION = "0.4.0.pre.tstarck.2"
+  VERSION = "0.4.0.pre.tstarck.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: fibered_mysql2
 version: !ruby/object:Gem::Version
-  version: 0.4.0.pre.tstarck.2
+  version: 0.4.0.pre.tstarck.3
 platform: ruby
 authors:
 - Invoca Development