Add Rails 7.2 compatibility

Also improves 7.1 compatibility

Close #252

Ref:
- rails/rails@009c7e7
- rails/rails#50064

lib/active_record/connection_adapters/chronomodel_adapter.rb

@@ -3,16 +3,19 @@
 require 'chrono_model'
 
 module ActiveRecord
+  # TODO: Remove when dropping Rails < 7.2 compatibility
   module ConnectionHandling
     def chronomodel_adapter_class
-      ConnectionAdapters::PostgreSQLAdapter
+      ChronoModel::Adapter
     end
 
     # Install the new adapter in ActiveRecord. This approach is required because
     # the PG adapter defines +add_column+ itself, thus making impossible to use
     # super() in overridden Module methods.
     #
     def chronomodel_connection(config) # :nodoc:
+      return chronomodel_adapter_class.new(config) if ActiveRecord::VERSION::STRING >= '7.1'
+
       conn_params = config.symbolize_keys
 
       conn_params.delete_if { |_, v| v.nil? }
@@ -29,9 +32,6 @@ def chronomodel_connection(config) # :nodoc:
 
       adapter = ChronoModel::Adapter.new(conn, logger, conn_params, config)
 
-      # Rails 7.2.0, see ifad/chronomodel#236
-      adapter.connect! if adapter.respond_to?(:connect!)
-
       unless adapter.chrono_supported?
         raise ChronoModel::Error, 'Your database server is not supported by ChronoModel. ' \
                                   'Currently, only PostgreSQL >= 9.3 is supported.'

lib/chrono_model.rb

@@ -64,3 +64,7 @@ def self.history_models
 
   raise 'In order to use ChronoModel, set `config.active_record.schema_format` to `:sql`'
 end
+
+if ActiveRecord::ConnectionAdapters.respond_to?(:register)
+  ActiveRecord::ConnectionAdapters.register 'chronomodel', 'ChronoModel::Adapter', 'chrono_model/adapter'
+end

lib/chrono_model/adapter.rb

@@ -33,6 +33,21 @@ class Adapter < ActiveRecord::ConnectionAdapters::PostgreSQLAdapter
     # The schema holding historical data
     HISTORY_SCHEMA  = 'history'
 
+    if ActiveRecord::VERSION::STRING >= '7.1'
+      def initialize(*)
+        super
+
+        connect!
+
+        unless chrono_supported?
+          raise ChronoModel::Error, 'Your database server is not supported by ChronoModel. ' \
+                                    'Currently, only PostgreSQL >= 9.3 is supported.'
+        end
+
+        chrono_setup!
+      end
+    end
+
     # Returns true whether the connection adapter supports our
     # implementation of temporal tables. Currently, Chronomodel
     # is supported starting with PostgreSQL 9.3 (90300 in PostgreSQL's

spec/aruba/rake_task_spec.rb

@@ -13,7 +13,10 @@
   describe 'bundle exec rake -T' do
     subject { last_command_started }
 
-    before { run_command_and_stop('bundle exec rake -T') }
+    before do
+      copy_db_config
+      run_command_and_stop('bundle exec rake -T')
+    end
 
     it { is_expected.to have_output(load_schema_task(as_regexp: true)) }
   end