Contents

# frozen_string_literal: true

module Switchman
  module ActiveRecord
    module ConnectionPool
      if ::Rails.version < "7.1"
        def get_schema_cache(connection)
          self.schema_cache ||= SharedSchemaCache.get_schema_cache(connection)
          self.schema_cache.connection = connection

          self.schema_cache
        end

        # rubocop:disable Naming/AccessorMethodName override method
        def set_schema_cache(cache)
          schema_cache = get_schema_cache(cache.connection)

          cache.instance_variables.each do |x|
            schema_cache.instance_variable_set(x, cache.instance_variable_get(x))
          end
        end
        # rubocop:enable Naming/AccessorMethodName override method
      end

      def default_schema
        connection unless @schemas
        # default shard will not switch databases immediately, so it won't be set yet
        @schemas ||= connection.current_schemas
        @schemas.first
      end

      def checkout_new_connection
        conn = super
        conn.shard = current_shard
        conn
      end

      def connection(switch_shard: true)
        conn = super()
        raise Errors::NonExistentShardError if current_shard.new_record?

        switch_database(conn) if conn.shard != current_shard && switch_shard
        conn
      end

      def release_connection(with_id = Thread.current)
        super(with_id)

        flush
      end

      def switch_database(conn)
        if !@schemas && conn.adapter_name == "PostgreSQL" && !current_shard.database_server.config[:shard_name]
          @schemas = conn.current_schemas
        end

        conn.shard = current_shard
      end

      private

      def current_shard
        (::Rails.version < "7.0") ? connection_klass.current_switchman_shard : connection_class.current_switchman_shard
      end

      def tls_key
        :"#{object_id}_shard"
      end
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
switchman-3.6.7	lib/switchman/active_record/connection_pool.rb
switchman-3.6.6	lib/switchman/active_record/connection_pool.rb
switchman-3.6.5	lib/switchman/active_record/connection_pool.rb
switchman-3.6.3	lib/switchman/active_record/connection_pool.rb
switchman-3.6.2	lib/switchman/active_record/connection_pool.rb
switchman-3.6.1	lib/switchman/active_record/connection_pool.rb
switchman-3.6.0	lib/switchman/active_record/connection_pool.rb