activerecord-spanner-adapter 1.6.3 → 2.0.0

data/lib/active_record/connection_adapters/spanner/database_statements.rb

@@ -21,15 +21,15 @@ module ActiveRecord
           internal_execute sql, name, binds
         end
 
-        def internal_exec_query sql, name = "SQL", binds = [], prepare: false, async: false
-          result = internal_execute sql, name, binds, prepare: prepare, async: async
+        def internal_exec_query sql, name = "SQL", binds = [], prepare: false, async: false, allow_retry: false
+          result = internal_execute sql, name, binds, prepare: prepare, async: async, allow_retry: allow_retry
           ActiveRecord::Result.new(
             result.fields.keys.map(&:to_s), result.rows.map(&:values)
           )
         end
 
         def internal_execute sql, name = "SQL", binds = [],
-                             prepare: false, async: false # rubocop:disable Lint/UnusedMethodArgument
+                             prepare: false, async: false, allow_retry: false # rubocop:disable Lint/UnusedMethodArgument, /
           statement_type = sql_statement_type sql
           # Call `transform` to invoke any query transformers that might have been registered.
           sql = transform sql
@@ -65,7 +65,7 @@ module ActiveRecord
           end
 
           log_args = [sql, name]
-          log_args.concat [binds, type_casted_binds(binds)] if log_statement_binds
+          log_args.push binds, type_casted_binds(binds) if log_statement_binds
 
           log(*log_args) do
             types, params = to_types_and_params binds
@@ -83,12 +83,16 @@ module ActiveRecord
         end
 
         def append_request_tag_from_query_logs sql, binds
-          legacy_formatter_prefix = "/*request_tag:true,"
-          sql_commenter_prefix = "/*request_tag='true',"
-          if sql.start_with? legacy_formatter_prefix
-            append_request_tag_from_query_logs_with_format sql, binds, legacy_formatter_prefix
-          elsif sql.start_with? sql_commenter_prefix
-            append_request_tag_from_query_logs_with_format sql, binds, sql_commenter_prefix
+          possible_prefixes = [
+            "/*request_tag:true,",
+            "/*_request_tag='true',",
+            "/*_request_tag:true,",
+            "/*_request_tag='true',"
+          ]
+          possible_prefixes.each do |prefix|
+            if sql.start_with? prefix
+              append_request_tag_from_query_logs_with_format sql, binds, prefix
+            end
           end
         end
 
@@ -101,7 +105,7 @@ module ActiveRecord
           if options.request_tag == ""
             options.request_tag = request_tag
           else
-            options.request_tag += "," + request_tag
+            options.request_tag += ",#{request_tag}"
           end
 
           binds.append options
@@ -192,7 +196,7 @@ module ActiveRecord
           # and this RPC can return multiple partial result sets for DML as well. Only the last partial
           # result set will contain the statistics. Although there will never be any rows, this makes
           # sure that the stream is fully consumed.
-          result.rows.each { |_| }
+          result.rows.each { |_| } # rubocop:disable Lint/EmptyBlock
           return result.row_count if result.row_count
 
           raise ActiveRecord::StatementInvalid.new(
@@ -282,12 +286,12 @@ module ActiveRecord
         #
         def begin_isolated_db_transaction isolation
           if isolation.is_a? Hash
-            raise "Unsupported isolation level: #{isolation}" unless \
+            raise "Unsupported isolation level: #{isolation}" unless
               isolation[:timestamp] || isolation[:staleness] || isolation[:strong]
             raise "Only one option is supported. It must be one of `timestamp`, `staleness` or `strong`." \
               if isolation.count != 1
           else
-            raise "Unsupported isolation level: #{isolation}" unless \
+            raise "Unsupported isolation level: #{isolation}" unless
               [:serializable, :read_only, :buffered_mutations, :pdml].include? isolation
           end
 
@@ -312,24 +316,35 @@ module ActiveRecord
 
         # Translates binds to Spanner types and params.
         def to_types_and_params binds
-          types = binds.enum_for(:each_with_index).map do |bind, i|
+          types = to_types binds
+          params = to_params binds
+          [types, params]
+        end
+
+        def to_types binds
+          binds.enum_for(:each_with_index).to_h do |bind, i|
             type = :INT64
             if bind.respond_to? :type
               type = ActiveRecord::Type::Spanner::SpannerActiveRecordConverter
                      .convert_active_model_type_to_spanner(bind.type)
-            elsif bind.class == Symbol
+            elsif bind.instance_of? Symbol
               # This ensures that for example :environment is sent as the string 'environment' to Cloud Spanner.
               type = :STRING
+            elsif bind.instance_of?(TrueClass) || bind.instance_of?(FalseClass)
+              type = :BOOL
             end
             [
               # Generates binds for named parameters in the format `@p1, @p2, ...`
               "p#{i + 1}", type
             ]
-          end.to_h
-          params = binds.enum_for(:each_with_index).map do |bind, i|
+          end
+        end
+
+        def to_params binds
+          binds.enum_for(:each_with_index).to_h do |bind, i|
             type = if bind.respond_to? :type
                      bind.type
-                   elsif bind.class == Symbol
+                   elsif bind.instance_of? Symbol
                      # This ensures that for example :environment is sent as the string 'environment' to Cloud Spanner.
                      :STRING
                    else
@@ -341,8 +356,7 @@ module ActiveRecord
                     .serialize_with_transaction_isolation_level(type, bind_value, :dml)
 
             ["p#{i + 1}", value]
-          end.to_h
-          [types, params]
+          end
         end
 
         # An insert/update/delete statement could use mutations in some specific circumstances.
@@ -351,7 +365,7 @@ module ActiveRecord
         def should_use_mutation arel
           !@connection.current_transaction.nil? \
             && @connection.current_transaction.isolation == :buffered_mutations \
-            && can_use_mutation(arel) \
+            && can_use_mutation(arel)
         end
 
         def can_use_mutation arel
@@ -381,7 +395,7 @@ module ActiveRecord
           )
         end
 
-        COMMENT_REGEX = %r{(?:--.*\n)*|/\*(?:[^*]|\*[^/])*\*/}m.freeze \
+        COMMENT_REGEX = %r{(?:--.*\n)*|/\*(?:[^*]|\*[^/])*\*/}m \
             unless defined? ActiveRecord::ConnectionAdapters::AbstractAdapter::COMMENT_REGEX
         COMMENT_REGEX = ActiveRecord::ConnectionAdapters::AbstractAdapter::COMMENT_REGEX \
             if defined? ActiveRecord::ConnectionAdapters::AbstractAdapter::COMMENT_REGEX

data/lib/active_record/connection_adapters/spanner/quoting.rb

@@ -30,20 +30,33 @@
 
 module ActiveRecord
   module ConnectionAdapters
+    QUOTED_COLUMN_NAMES = Concurrent::Map.new # :nodoc:
+    QUOTED_TABLE_NAMES = Concurrent::Map.new # :nodoc:
+
+    module Quoting
+      module ClassMethods
+        # This is used for ActiveRecord v8 and higher.
+        def quote_column_name name
+          QUOTED_COLUMN_NAMES[name] ||= "`#{name.to_s.gsub '`', '``'}`".freeze
+        end
+
+        def quote_table_name name
+          QUOTED_TABLE_NAMES[name] ||= "`#{name.to_s.gsub '.', '`.`'}`".freeze
+        end
+      end
+    end
+
     module Spanner
       module Quoting
-        QUOTED_COLUMN_NAMES = Concurrent::Map.new # :nodoc:
-        QUOTED_TABLE_NAMES = Concurrent::Map.new # :nodoc:
-
         def quote_column_name name
-          QUOTED_COLUMN_NAMES[name] ||= "`#{super.gsub '`', '``'}`".freeze
+          QUOTED_COLUMN_NAMES[name] ||= "`#{name.to_s.gsub '`', '``'}`".freeze
         end
 
         def quote_table_name name
-          QUOTED_TABLE_NAMES[name] ||= super.gsub(".", "`.`").freeze
+          QUOTED_TABLE_NAMES[name] ||= "`#{name.to_s.gsub '.', '`.`'}`".freeze
         end
 
-        STR_ESCAPE_REGX = /[\n\r'\\]/.freeze
+        STR_ESCAPE_REGX = /[\n\r'\\]/
         STR_ESCAPE_VALUES = {
           "\n" => "\\n", "\r" => "\\r", "'" => "\\'", "\\" => "\\\\"
         }.freeze

data/lib/active_record/connection_adapters/spanner/schema_creation.rb

@@ -44,10 +44,7 @@ module ActiveRecord
                                      else
                                        [o.options[:primary_key]]
                                      end
-                           pk_names = []
-                           columns.each do |c|
-                             pk_names.append c.to_s
-                           end
+                           pk_names = columns.map(&:to_s)
                            PrimaryKeyDefinition.new pk_names
                          else
                            pk_names = o.columns.each_with_object [] do |c, r|
@@ -93,8 +90,8 @@ module ActiveRecord
         end
 
         def visit_DropColumnDefinition o
-          "ALTER TABLE #{quote_table_name o.table_name} DROP" \
-           " COLUMN #{quote_column_name o.name}"
+          "ALTER TABLE #{quote_table_name o.table_name} DROP " \
+            "COLUMN #{quote_column_name o.name}"
         end
 
         def visit_ChangeColumnDefinition o
@@ -143,7 +140,7 @@ module ActiveRecord
 
           if !options[:allow_commit_timestamp].nil? &&
              options[:column].sql_type == "TIMESTAMP"
-            sql << " OPTIONS (allow_commit_timestamp = "\
+            sql << " OPTIONS (allow_commit_timestamp = " \
                    "#{options[:allow_commit_timestamp]})"
           end
 
@@ -153,8 +150,9 @@ module ActiveRecord
             sql << " STORED" if options[:stored]
             unless options[:stored]
               raise ArgumentError, "" \
-                "Cloud Spanner currently does not support generated columns without the STORED option." \
-                "Specify 'stored: true' option for `#{options[:column].name}`"
+                                   "Cloud Spanner currently does not support generated columns" \
+                                   "without the STORED option." \
+                                   "Specify 'stored: true' option for `#{options[:column].name}`"
             end
           end
 

data/lib/active_record/connection_adapters/spanner/schema_definitions.rb

@@ -5,7 +5,7 @@
 # https://opensource.org/licenses/MIT.
 
 module ActiveRecord
-  module ConnectionAdapters #:nodoc:
+  module ConnectionAdapters # :nodoc:
     module Spanner
       class TableDefinition < ActiveRecord::ConnectionAdapters::TableDefinition
         attr_reader :interleave_in_parent
@@ -62,6 +62,9 @@ module ActiveRecord
       DropIndexDefinition = Struct.new :name
 
       class ReferenceDefinition < ActiveRecord::ConnectionAdapters::ReferenceDefinition
+        # This constructor intentionally does not call super to prevent ActiveRecord
+        # from creating an additional secondary index for the foreign key.
+        # rubocop:disable Lint/MissingSuper
         def initialize \
             name,
             polymorphic: false,
@@ -81,6 +84,7 @@ module ActiveRecord
           return unless polymorphic && foreign_key
           raise ArgumentError, "Cannot add a foreign key to a polymorphic relation"
         end
+        # rubocop:enable Lint/MissingSuper
 
         private
 
@@ -96,8 +100,13 @@ module ActiveRecord
       end
 
       class IndexDefinition < ActiveRecord::ConnectionAdapters::IndexDefinition
-        attr_reader :null_filtered, :interleave_in, :storing, :orders
+        attr_reader :null_filtered
+        attr_reader :interleave_in
+        attr_reader :storing
+        attr_reader :orders
 
+        # This constructor intentionally does not call super.
+        # rubocop:disable Lint/MissingSuper
         def initialize \
             table_name,
             name,
@@ -123,6 +132,7 @@ module ActiveRecord
 
           @orders = @orders.symbolize_keys
         end
+        # rubocop:enable Lint/MissingSuper
 
         def columns_with_order
           columns.each_with_object({}) do |c, result|

data/lib/active_record/connection_adapters/spanner/schema_statements.rb

@@ -23,6 +23,7 @@ module ActiveRecord
       module SchemaStatements
         VERSION_6_1_0 = Gem::Version.create "6.1.0"
         VERSION_6_0_3 = Gem::Version.create "6.0.3"
+        VERSION_7_2 = Gem::Version.create "7.2.0"
 
         def current_database
           @connection.database_id
@@ -108,8 +109,7 @@ module ActiveRecord
         end
 
         def rename_table _table_name, _new_name
-          raise ActiveRecordSpannerAdapter::NotSupportedError, \
-                "rename_table is not implemented"
+          raise ActiveRecordSpannerAdapter::NotSupportedError, "rename_table is not implemented"
         end
 
         # Column
@@ -167,20 +167,14 @@ module ActiveRecord
           execute_schema_statements statements
         end
 
-        if ActiveRecord.gem_version < VERSION_6_1_0
-          def remove_columns table_name, *column_names
-            _remove_columns table_name, *column_names
-          end
-        else
-          def remove_columns table_name, *column_names, _type: nil, **_options
-            _remove_columns table_name, *column_names
-          end
+        def remove_columns table_name, *column_names, _type: nil, **_options
+          _remove_columns table_name, *column_names
         end
 
         def _remove_columns table_name, *column_names
           if column_names.empty?
-            raise ArgumentError, "You must specify at least one column name. "\
-              "Example: remove_columns(:people, :first_name)"
+            raise ArgumentError, "You must specify at least one column name. " \
+                                 "Example: remove_columns(:people, :first_name)"
           end
 
           statements = []
@@ -192,14 +186,8 @@ module ActiveRecord
           execute_schema_statements statements
         end
 
-        if ActiveRecord.gem_version < VERSION_6_1_0
-          def change_column table_name, column_name, type, options = {}
-            _change_column table_name, column_name, type, **options
-          end
-        else
-          def change_column table_name, column_name, type, **options
-            _change_column table_name, column_name, type, **options
-          end
+        def change_column table_name, column_name, type, **options
+          _change_column table_name, column_name, type, **options
         end
 
         def change_column_null table_name, column_name, null, _default = nil
@@ -207,14 +195,12 @@ module ActiveRecord
         end
 
         def change_column_default _table_name, _column_name, _default_or_changes
-          raise ActiveRecordSpannerAdapter::NotSupportedError, \
-                "change column with default value not supported."
+          raise ActiveRecordSpannerAdapter::NotSupportedError, "change column with default value not supported."
         end
 
         def rename_column table_name, column_name, new_column_name
           if ActiveRecord::Base.connection.ddl_batch?
-            raise ActiveRecordSpannerAdapter::NotSupportedError, \
-                  "rename_column in a DDL Batch is not supported."
+            raise ActiveRecordSpannerAdapter::NotSupportedError, "rename_column in a DDL Batch is not supported."
           end
           column = information_schema do |i|
             i.table_column table_name, column_name
@@ -279,16 +265,9 @@ module ActiveRecord
           execute_schema_statements schema_creation.accept(id)
         end
 
-        if ActiveRecord.gem_version < VERSION_6_1_0
-          def remove_index table_name, options = {}
-            index_name = index_name_for_remove table_name, options
-            execute "DROP INDEX #{quote_table_name index_name}"
-          end
-        else
-          def remove_index table_name, column_name = nil, **options
-            index_name = index_name_for_remove table_name, column_name, options
-            execute "DROP INDEX #{quote_table_name index_name}"
-          end
+        def remove_index table_name, column_name = nil, **options
+          index_name = index_name_for_remove table_name, column_name, options
+          execute "DROP INDEX #{quote_table_name index_name}"
         end
 
         def rename_index table_name, old_name, new_name
@@ -357,14 +336,8 @@ module ActiveRecord
           end
         end
 
-        if ActiveRecord.gem_version < VERSION_6_0_3
-          def add_foreign_key from_table, to_table, options = {}
-            _add_foreign_key from_table, to_table, **options
-          end
-        else
-          def add_foreign_key from_table, to_table, **options
-            _add_foreign_key from_table, to_table, **options
-          end
+        def add_foreign_key from_table, to_table, **options
+          _add_foreign_key from_table, to_table, **options
         end
 
         def _add_foreign_key from_table, to_table, **options
@@ -401,6 +374,16 @@ module ActiveRecord
           information_schema { |i| i.check_constraints table_name }
         end
 
+        if ActiveRecord.gem_version >= VERSION_7_2
+          def migration_context
+            pool.migration_context
+          end
+
+          def schema_migration
+            pool.schema_migration
+          end
+        end
+
         def assume_migrated_upto_version version
           version = version.to_i
           sm_table = quote_table_name schema_migration.table_name
@@ -497,8 +480,8 @@ module ActiveRecord
           type ||= column.type
           options[:null] = column.null unless options.key? :null
 
-          if ["STRING", "BYTES"].include? type
-            options[:limit] = column.limit unless options.key? :limit
+          if ["STRING", "BYTES"].include?(type) && !options.key?(:limit)
+            options[:limit] = column.limit
           end
 
           # Only timestamp type can set commit timestamp
@@ -631,8 +614,7 @@ module ActiveRecord
         end
 
         def information_schema
-          info_schema = \
-            ActiveRecordSpannerAdapter::Connection.information_schema @config
+          info_schema = ActiveRecordSpannerAdapter::Connection.information_schema @config
 
           return info_schema unless block_given?
 

data/lib/active_record/connection_adapters/spanner/type_metadata.rb

@@ -14,7 +14,9 @@ module ActiveRecord
 
         include Deduplicable if defined?(Deduplicable)
 
-        attr_reader :ordinal_position, :allow_commit_timestamp, :generated
+        attr_reader :ordinal_position
+        attr_reader :allow_commit_timestamp
+        attr_reader :generated
 
         def initialize type_metadata, ordinal_position: nil, allow_commit_timestamp: nil, generated: nil
           super type_metadata
@@ -33,11 +35,7 @@ module ActiveRecord
         alias eql? ==
 
         def hash
-          TypeMetadata.hash ^
-            __getobj__.hash ^
-            ordinal_position.hash ^
-            allow_commit_timestamp.hash ^
-            generated.hash
+          [TypeMetadata.name, __getobj__, ordinal_position, allow_commit_timestamp, generated].hash
         end
 
         private

data/lib/active_record/connection_adapters/spanner_adapter.rb

@@ -28,16 +28,18 @@ require "activerecord_spanner_adapter/primary_key"
 require "activerecord_spanner_adapter/transaction"
 
 module ActiveRecord
-  module ConnectionHandling # :nodoc:
-    def spanner_connection config
-      connection = ActiveRecordSpannerAdapter::Connection.new config
-      connection.connect!
-      ConnectionAdapters::SpannerAdapter.new connection, logger, nil, config
-    rescue Google::Cloud::Error => error
-      if error.instance_of? Google::Cloud::NotFoundError
-        raise ActiveRecord::NoDatabaseError
-      end
-      raise error
+  if ActiveRecord.version < Gem::Version.new("7.2")
+    module ConnectionHandling # :nodoc:
+      def spanner_connection config
+        connection = ActiveRecordSpannerAdapter::Connection.new config
+        connection.connect!
+        ConnectionAdapters::SpannerAdapter.new connection, logger, nil, config
+      rescue Google::Cloud::Error => error
+        if error.instance_of? Google::Cloud::NotFoundError
+          raise ActiveRecord::NoDatabaseError
+        end
+        raise error
+      end
     end
   end
 
@@ -69,11 +71,21 @@ module ActiveRecord
       # Determines whether or not to log query binds when executing statements
       class_attribute :log_statement_binds, instance_writer: false, default: false
 
-      def initialize connection, logger, connection_options, config
-        @connection = connection
-        @connection_options = connection_options
-        super connection, logger, config
-        @raw_connection ||= connection
+      def initialize config_or_deprecated_connection, deprecated_logger = nil,
+                     deprecated_connection_options = nil, deprecated_config = nil
+        if config_or_deprecated_connection.is_a? Hash
+          @connection = ActiveRecordSpannerAdapter::Connection.new config_or_deprecated_connection
+          @connection.connect!
+          super config_or_deprecated_connection
+          @raw_connection ||= @connection
+        else
+          @connection = config_or_deprecated_connection
+          @connection_options = deprecated_connection_options
+          super config_or_deprecated_connection, deprecated_logger, deprecated_config
+          @raw_connection ||= config_or_deprecated_connection
+        end
+        # Spanner does not support unprepared statements
+        @prepared_statements = true
       end
 
       def max_identifier_length
@@ -197,12 +209,14 @@ module ActiveRecord
           raise "ActiveRecordSpannerAdapter does not support insert_sql with buffered_mutations transaction."
         end
 
-        if insert.skip_duplicates? || insert.update_duplicates?
-          raise NotImplementedError, "CloudSpanner does not support skip_duplicates and update_duplicates."
-        end
-
         values_list, = insert.values_list
-        "INSERT #{insert.into} #{values_list}"
+        prefix = "INSERT"
+        if insert.update_duplicates?
+          prefix += " OR UPDATE"
+        elsif insert.skip_duplicates?
+          prefix += " OR IGNORE"
+        end
+        "#{prefix} #{insert.into} #{values_list}"
       end
 
       module TypeMapBuilder
@@ -262,7 +276,9 @@ module ActiveRecord
       end
 
       def transform sql
-        if ActiveRecord::VERSION::MAJOR >= 7
+        if ActiveRecord::VERSION::MAJOR >= 8
+          preprocess_query sql
+        elsif ActiveRecord::VERSION::MAJOR == 7
           transform_query sql
         else
           sql
@@ -270,12 +286,23 @@ module ActiveRecord
       end
 
       # Overwrite the standard log method to be able to translate exceptions.
-      def log sql, name = "SQL", binds = [], type_casted_binds = [], statement_name = nil, *args
-        super
-      rescue ActiveRecord::StatementInvalid
-        raise
-      rescue StandardError => e
-        raise translate_exception_class(e, sql, binds)
+      # sql, name = "SQL", binds = [], type_casted_binds = [], async: false, &block
+      if ActiveRecord::VERSION::MAJOR >= 8
+        def log sql, name = "SQL", binds = [], type_casted_binds = [], async: false, &block
+          super
+        rescue ActiveRecord::StatementInvalid
+          raise
+        rescue StandardError => e
+          raise translate_exception_class(e, sql, binds)
+        end
+      else
+        def log sql, name = "SQL", binds = [], type_casted_binds = [], statement_name = nil, *args
+          super
+        rescue ActiveRecord::StatementInvalid
+          raise
+        rescue StandardError => e
+          raise translate_exception_class(e, sql, binds)
+        end
       end
 
       def translate_exception exception, message:, sql:, binds:

data/lib/active_record/tasks/spanner_database_tasks.rb

@@ -31,7 +31,7 @@ module ActiveRecord
       def purge
         begin
           drop
-        rescue ActiveRecord::NoDatabaseError # rubocop:disable Lint/HandleExceptions
+        rescue ActiveRecord::NoDatabaseError
           # ignored; create the database
         end
 
@@ -51,8 +51,8 @@ module ActiveRecord
         ignore_tables = ActiveRecord::SchemaDumper.ignore_tables
 
         if ignore_tables.any?
-          index_regx = /^CREATE(.*)INDEX(.*)ON (#{ignore_tables.join "|"})\(/
-          table_regx = /^CREATE TABLE (#{ignore_tables.join "|"})/
+          index_regx = /^CREATE(.*)INDEX(.*)ON (#{ignore_tables.join '|'})\(/
+          table_regx = /^CREATE TABLE (#{ignore_tables.join '|'})/
         end
 
         @connection.database.ddl(force: true).each do |statement|
@@ -66,7 +66,7 @@ module ActiveRecord
       end
 
       def structure_load filename, _extra_flags
-        statements = File.read(filename).split(/;/).map(&:strip).reject(&:empty?)
+        statements = File.read(filename).split(";").map(&:strip).reject(&:empty?)
         ddls = statements.select { |s| s =~ /^(CREATE|ALTER|DROP|GRANT|REVOKE|ANALYZE)/ }
         @connection.execute_ddl ddls
 

data/lib/active_record/type/spanner/array.rb

@@ -9,11 +9,15 @@ module ActiveRecord
     module Spanner
       class Array < Type::Value
         attr_reader :element_type
+
         delegate :type, :user_input_in_time_zone, :limit, :precision, :scale, to: :element_type
 
+        # This constructor intentionally does not call super.
+        # rubocop:disable Lint/MissingSuper
         def initialize element_type
           @element_type = element_type
         end
+        # rubocop:enable Lint/MissingSuper
 
         def cast value
           return super if value.nil?

data/lib/active_record/type/spanner/bytes.rb

@@ -10,6 +10,16 @@ module ActiveRecord
   module Type
     module Spanner
       class Bytes < ActiveRecord::Type::Binary
+        def deserialize value
+          # Set this environment variable to disable de-serializing BYTES
+          # to a StringIO instance.
+          return super if ENV["SPANNER_BYTES_DESERIALIZE_DISABLED"]
+
+          return super value if value.nil?
+          return StringIO.new Base64.strict_decode64(value) if value.is_a? ::String
+          value
+        end
+
         def serialize value
           return super value if value.nil?
 

data/lib/activerecord-spanner-adapter.rb

@@ -15,7 +15,11 @@ if defined?(Rails)
         end
 
         ActiveSupport.on_load :active_record do
-          require "active_record/connection_adapters/spanner_adapter"
+          if Rails.version >= "7.2.0"
+            ActiveRecord::ConnectionAdapters.register("spanner", "ActiveRecord::ConnectionAdapters::SpannerAdapter")
+          else
+            require "active_record/connection_adapters/spanner_adapter"
+          end
         end
       end
     end