Contents

# frozen_string_literal: true
require 'concurrent'
require 'singleton'

module FreshConnection
  class ReplicaConnectionHandler
    include Singleton

    def initialize
      @owner_to_pool = Concurrent::Map.new(initial_capacity: 2) do |h, k|
        h[k] = Concurrent::Map.new(initial_capacity: 2)
      end
    end

    def refresh_all
      owner_to_pool.clear
    end

    def refresh_connection(spec_name)
      remove_connection(spec_name.to_s)
    end

    def connection(spec_name)
      detect_connection_manager(spec_name).replica_connection
    end

    def clear_all_connections!
      all_connection_managers do |connection_manager|
        connection_manager.clear_all_connections!
      end
    end

    def recovery?(spec_name)
      detect_connection_manager(spec_name).recovery?
    end

    def put_aside!
      all_connection_managers do |connection_manager|
        connection_manager.put_aside!
      end
    end

    private

    def remove_connection(spec_name)
      pool = owner_to_pool.delete(spec_name.to_s)
      return unless pool

      pool.clear_all_connections!
    end

    def all_connection_managers
      owner_to_pool.each_value do |connection_manager|
        yield(connection_manager)
      end
    end

    def detect_connection_manager(spec_name)
      owner_to_pool.fetch(spec_name.to_s) do
        refresh_connection(spec_name)

        message_bus = ActiveSupport::Notifications.instrumenter
        payload = {
          connection_id: object_id,
          spec_name: spec_name
        }

        message_bus.instrument("!connection.active_record", payload) do
          FreshConnection.connection_manager.new(spec_name)
        end
      end
    end

    def owner_to_pool
      @owner_to_pool[Process.pid]
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
fresh_connection-3.0.1	lib/fresh_connection/replica_connection_handler.rb
fresh_connection-3.0.1.rc2	lib/fresh_connection/replica_connection_handler.rb
fresh_connection-3.0.1.rc1	lib/fresh_connection/replica_connection_handler.rb