Get started with Rails 7.1 support

active_record_shards.gemspec

@@ -8,8 +8,8 @@ Gem::Specification.new "active_record_shards", "5.5.0" do |s|
 
   s.required_ruby_version = ">= 2.6"
 
-  s.add_runtime_dependency("activerecord", ">= 5.1", "< 7.1")
-  s.add_runtime_dependency("activesupport", ">= 5.1", "< 7.1")
+  s.add_runtime_dependency("activerecord", ">= 5.1", "<= 7.1")
+  s.add_runtime_dependency("activesupport", ">= 5.1", "<= 7.1")
 
   s.add_development_dependency("bump")
   s.add_development_dependency("minitest", ">= 5.10.0")

gemfiles/rails7.1.gemfile

@@ -0,0 +1,3 @@
+eval_gemfile 'common.rb'
+
+gem 'activerecord', '~> 7.1.0'

lib/active_record_shards.rb

@@ -75,7 +75,7 @@ def self.reset_app_env!
 
   # https://github.com/rails/rails/blob/v5.2.6/activerecord/lib/active_record/associations/preloader/association.rb#L96
   ActiveRecord::Associations::Preloader::Association.prepend(ActiveRecordShards::DefaultReplicaPatches::AssociationsPreloaderAssociationLoadRecordsPatch)
-when '6.0', '6.1', '7.0'
+when '6.0', '6.1', '7.0', '7.1'
   # https://github.com/rails/rails/blob/v6.0.4/activerecord/lib/active_record/type_caster/connection.rb#L28
   ActiveRecord::TypeCaster::Connection.prepend(ActiveRecordShards::DefaultReplicaPatches::TypeCasterConnectionConnectionPatch)
 

lib/active_record_shards/connection_switcher.rb

@@ -271,7 +271,7 @@ def method_missing(method, *args, &block) # rubocop:disable Style/MethodMissingS
   require 'active_record_shards/connection_switcher-6-0'
 when '6.1'
   require 'active_record_shards/connection_switcher-6-1'
-when '7.0'
+when '7.0', '7.1'
   require 'active_record_shards/connection_switcher-7-0'
 else
   raise "ActiveRecordShards is not compatible with #{ActiveRecord::VERSION::STRING}"

lib/active_record_shards/shard_selection.rb

@@ -35,7 +35,8 @@ def resolve_connection_name(sharded:, configurations:)
         name << "_shard_#{resolved_shard}" if resolved_shard
         replica_config = begin
           case "#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}"
-          when '7.0'
+          when '7.0', '7.1'
+            binding.pry
             configurations.configs_for(env_name: "#{name}_replica", include_hidden: true).any?
           when '6.1'
             configurations.configs_for(env_name: "#{name}_replica", include_replicas: true).any?
@@ -48,6 +49,10 @@ def resolve_connection_name(sharded:, configurations:)
         else
           # ActiveRecord always names its default connection pool 'primary'
           # while everything else is named by the configuration name
+
+          # NOTE: for some reason this might need to return the environment.
+          # my 7.1 specs fail because  see `connection_switcher-7-0.rb:8`
+          # configurations.configs_for(env_name: "primiary", include_hidden: true).any? == false
           resolved_shard ? name : PRIMARY
         end
       end