active_record_shards 3.17.0 → 4.0.0.beta1

data/lib/active_record_shards/connection_handler.rb

@@ -1,8 +0,0 @@
-# frozen_string_literal: true
-
-ActiveRecord::ConnectionAdapters::ConnectionHandler.class_eval do
-  remove_method :retrieve_connection_pool
-  def retrieve_connection_pool(klass)
-    class_to_pool[klass.connection_pool_name] ||= pool_for(klass)
-  end
-end

data/lib/active_record_shards/connection_pool.rb

@@ -1,36 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecordShards
-  ConnectionPoolNameDecorator = Struct.new(:name)
-
-  # It overrides given connection handler methods (they differ depend on
-  # Rails version).
-  #
-  # It takes the first argument, ActiveRecord::Base object or
-  # String (connection_pool_name), converts it in Struct object and
-  # passes to the original method.
-  #
-  # Example:
-  #   methods_to_override = [:establish_connection, :remove_connection]
-  #   ActiveRecordShards.override_connection_handler_methods(methods_to_override)
-  #
-  def self.override_connection_handler_methods(method_names)
-    method_names.each do |method_name|
-      ActiveRecord::ConnectionAdapters::ConnectionHandler.class_eval do
-        define_method("#{method_name}_with_connection_pool_name") do |*args|
-          unless args[0].is_a? ConnectionPoolNameDecorator
-            name = if args[0].is_a? String
-                     args[0]
-                   else
-                     args[0].connection_pool_name
-                   end
-            args[0] = ConnectionPoolNameDecorator.new(name)
-          end
-          send("#{method_name}_without_connection_pool_name", *args)
-        end
-        alias_method :"#{method_name}_without_connection_pool_name", method_name
-        alias_method method_name, :"#{method_name}_with_connection_pool_name"
-      end
-    end
-  end
-end

data/lib/active_record_shards/connection_specification.rb

@@ -1,19 +0,0 @@
-# frozen_string_literal: true
-
-class << ActiveRecord::Base
-  remove_method :establish_connection if ActiveRecord::VERSION::MAJOR >= 5
-  def establish_connection(spec = ENV["DATABASE_URL"])
-    spec ||= ActiveRecord::ConnectionHandling::DEFAULT_ENV.call
-    spec = spec.to_sym if spec.is_a?(String)
-    resolver = ActiveRecordShards::ConnectionSpecification::Resolver.new configurations
-    spec = resolver.spec(spec)
-
-    unless respond_to?(spec.adapter_method)
-      raise AdapterNotFound, "database configuration specifies nonexistent #{spec.config[:adapter]} adapter"
-    end
-
-    remove_connection
-    specification_cache[connection_pool_name] = spec
-    connection_handler.establish_connection self, spec
-  end
-end

data/lib/active_record_shards/connection_switcher-4-2.rb

@@ -1,56 +0,0 @@
-module ActiveRecordShards
-  module ConnectionSwitcher
-    # Name of the connection pool. Used by ConnectionHandler to retrieve the current connection pool.
-    def connection_pool_name # :nodoc:
-      name = current_shard_selection.shard_name(self)
-
-      # e.g. if "production_slave" is not defined in `Configuration`, fall back to "production"
-      if configurations[name].nil? && on_slave?
-        current_shard_selection.shard_name(self, false)
-      else
-        name
-      end
-    end
-
-    private
-
-    def ensure_shard_connection
-      establish_shard_connection unless connected_to_shard?
-    end
-
-    def establish_shard_connection
-      name = connection_pool_name
-      spec = configurations[name]
-
-      if spec.nil?
-        raise ActiveRecord::AdapterNotSpecified, "No database defined by #{name} in your database config. (configurations: #{configurations.keys.inspect})"
-      end
-
-      specification_cache[name] ||= begin
-        resolver = ActiveRecordShards::ConnectionSpecification::Resolver.new configurations
-        resolver.spec(spec)
-      end
-
-      connection_handler.establish_connection(self, specification_cache[name])
-    end
-
-    def specification_cache
-      @@specification_cache ||= {}
-    end
-
-    # Helper method to clear global state when testing.
-    def clear_specification_cache
-      @@specification_cache = {}
-    end
-
-    def connection_pool_key
-      specification_cache[connection_pool_name]
-    end
-
-    def connected_to_shard?
-      specs_to_pools = Hash[connection_handler.connection_pool_list.map { |pool| [pool.spec, pool] }]
-
-      specs_to_pools.key?(connection_pool_key)
-    end
-  end
-end

data/lib/active_record_shards/migration.rb

@@ -1,125 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecord
-  class Migrator
-    def self.shards_migration_context
-      if ActiveRecord::VERSION::MAJOR >= 6
-        ActiveRecord::MigrationContext.new(ActiveRecord::Migrator.migrations_paths, ActiveRecord::SchemaMigration)
-      elsif ActiveRecord::VERSION::STRING >= '5.2.0'
-        ActiveRecord::MigrationContext.new(ActiveRecord::Migrator.migrations_paths)
-      else
-        self
-      end
-    end
-
-    def initialize_with_sharding(*args)
-      initialize_without_sharding(*args)
-
-      # Rails creates the internal tables on the unsharded DB. We make them
-      # manually on the sharded DBs.
-      ActiveRecord::Base.on_all_shards do
-        ActiveRecord::SchemaMigration.create_table
-        if ActiveRecord::VERSION::MAJOR >= 5
-          ActiveRecord::InternalMetadata.create_table
-        end
-      end
-    end
-    alias_method :initialize_without_sharding, :initialize
-    alias_method :initialize, :initialize_with_sharding
-
-    def run_with_sharding
-      ActiveRecord::Base.on_shard(nil) { run_without_sharding }
-      ActiveRecord::Base.on_all_shards { run_without_sharding }
-    end
-    alias_method :run_without_sharding, :run
-    alias_method :run, :run_with_sharding
-
-    def migrate_with_sharding
-      ActiveRecord::Base.on_shard(nil) { migrate_without_sharding }
-      ActiveRecord::Base.on_all_shards { migrate_without_sharding }
-    end
-    alias_method :migrate_without_sharding, :migrate
-    alias_method :migrate, :migrate_with_sharding
-
-    # don't allow Migrator class to cache versions
-    undef migrated
-    def migrated
-      self.class.shards_migration_context.get_all_versions
-    end
-
-    # list of pending migrations is any migrations that haven't run on all shards.
-    undef pending_migrations
-    def pending_migrations
-      pending, _missing = self.class.shard_status(migrations.map(&:version))
-      pending = pending.values.flatten
-      migrations.select { |m| pending.include?(m.version) }
-    end
-
-    # public
-    # list of pending and missing versions per shard
-    # [{1 => [1234567]}, {1 => [2345678]}]
-    def self.shard_status(versions)
-      pending = {}
-      missing = {}
-
-      collect = lambda do |shard|
-        migrated = shards_migration_context.get_all_versions
-
-        p = versions - migrated
-        pending[shard] = p if p.any?
-
-        m = migrated - versions
-        missing[shard] = m if m.any?
-      end
-
-      ActiveRecord::Base.on_shard(nil) { collect.call(nil) }
-      ActiveRecord::Base.on_all_shards { |shard| collect.call(shard) }
-
-      [pending, missing]
-    end
-  end
-end
-
-module ActiveRecordShards
-  module MigrationClassExtension
-    attr_accessor :migration_shard
-
-    def shard(arg = nil)
-      self.migration_shard = arg
-    end
-  end
-
-  module ActualMigrationExtension
-    def migrate_with_forced_shard(direction)
-      if migration_shard.blank?
-        raise "#{name}: Can't run migrations without a shard spec: this may be :all, :none,
-                 or a specific shard (for data-fixups).  please call shard(arg) in your migration."
-      end
-
-      shard = ActiveRecord::Base.current_shard_selection.shard
-
-      if shard.nil?
-        return if migration_shard != :none
-      else
-        return if migration_shard == :none
-        return if migration_shard != :all && migration_shard.to_s != shard.to_s
-      end
-
-      migrate_without_forced_shard(direction)
-    end
-
-    def migration_shard
-      self.class.migration_shard
-    end
-  end
-end
-
-ActiveRecord::Migration.class_eval do
-  extend ActiveRecordShards::MigrationClassExtension
-  include ActiveRecordShards::ActualMigrationExtension
-
-  alias_method :migrate_without_forced_shard, :migrate
-  alias_method :migrate, :migrate_with_forced_shard
-end
-
-ActiveRecord::MigrationProxy.delegate :migration_shard, to: :migration

data/lib/active_record_shards/patches-4-2.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-require 'active_record_shards/connection_pool'
-require 'active_record_shards/connection_handler'
-require 'active_record_shards/connection_specification'
-
-ActiveRecordShards::ConnectionSpecification = ActiveRecord::ConnectionAdapters::ConnectionSpecification
-methods_to_override = [:establish_connection, :remove_connection, :pool_for, :pool_from_any_process_for]
-ActiveRecordShards.override_connection_handler_methods(methods_to_override)

data/lib/active_record_shards/schema_dumper_extension.rb

@@ -1,42 +0,0 @@
-# frozen_string_literal: true
-
-module ActiveRecordShards
-  module SchemaDumperExtension
-    def dump(stream)
-      stream = super(stream)
-      original_connection = @connection
-
-      if ActiveRecord::Base.supports_sharding?
-        ActiveRecord::Base.on_first_shard do
-          @connection = ActiveRecord::Base.connection
-          shard_header(stream)
-          extensions(stream)
-          tables(stream)
-          shard_trailer(stream)
-        end
-      end
-
-      stream
-    ensure
-      @connection = original_connection
-    end
-
-    def shard_header(stream)
-      define_params = @version ? "version: #{@version}" : ""
-
-      stream.puts <<~HEADER
-
-
-        # This section generated by active_record_shards
-
-        ActiveRecord::Base.on_all_shards do
-        ActiveRecord::Schema.define(#{define_params}) do
-
-      HEADER
-    end
-
-    def shard_trailer(stream)
-      stream.puts "end\nend"
-    end
-  end
-end