Contents

# frozen_string_literal: true

require 'em-synchrony'
require 'active_model'
require 'active_record/errors'
require 'active_record/connection_adapters/em_mysql2_adapter'

module FiberedMysql2
  module FiberedMysql2Adapter_4_2
    def lease
      synchronize do
        unless in_use?
          @owner = Fiber.current
        end
      end
    end
  end

  module FiberedMysql2Adapter_5_2
    def lease
      if in_use?
        msg = "Cannot lease connection, ".dup
        if @owner == Fiber.current
          msg << "it is already leased by the current fiber."
        else
          msg << "it is already in use by a different fiber: #{@owner}. " \
                  "Current fiber: #{Fiber.current}."
        end
        raise ::ActiveRecord::ActiveRecordError, msg
      end

      @owner = Fiber.current
    end

    def expire
      if in_use?
        # Because we are actively releasing connections from dead fibers, we only want
        # to enforce that we're expiring the current fibers connection, iff the owner
        # of the connection is still alive.
        if @owner.alive? && @owner != Fiber.current
          raise ::ActiveRecord::ActiveRecordError, "Cannot expire connection, " \
            "it is owned by a different fiber: #{@owner}. " \
            "Current fiber: #{Fiber.current}."
        end

        @idle_since = ::Concurrent.monotonic_time
        @owner = nil
      else
        raise ::ActiveRecord::ActiveRecordError, "Cannot expire connection, it is not currently leased."
      end
    end
  end

  class FiberedMysql2Adapter < ::ActiveRecord::ConnectionAdapters::EMMysql2Adapter
    case ::Rails::VERSION::MAJOR
    when 4
      include FiberedMysql2Adapter_4_2
    when 5, 6
      include FiberedMysql2Adapter_5_2
    end

    def initialize(*args)
      super
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
fibered_mysql2-0.1.0	lib/active_record/connection_adapters/fibered_mysql2_adapter.rb
fibered_mysql2-0.1.0.pre.5	lib/active_record/connection_adapters/fibered_mysql2_adapter.rb
fibered_mysql2-0.1.0.pre.4	lib/active_record/connection_adapters/fibered_mysql2_adapter.rb
fibered_mysql2-0.1.0.pre.3	lib/active_record/connection_adapters/fibered_mysql2_adapter.rb
fibered_mysql2-0.1.0.pre.2	lib/fibered_mysql2/fibered_mysql2_adapter.rb