activerecord-cockroachdb-adapter 6.1.7 → 6.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 95b9157a969cd938c7500ed5a755120ec9f9bdce6e5e11316d32e2e434e8d7b2
-  data.tar.gz: 96f79790d7f0fbbb95b32e3e07425fe8df2d10830165e31302f0d9fa8f27405a
+  metadata.gz: 72a654c4b06b42fbaae48bc9610420b2103f38b72d4170b7ae59983860d6e9d6
+  data.tar.gz: fb400dec89ffb39d202a081ff09a34330a498e3e9ce4363433d55de255a0288a
 SHA512:
-  metadata.gz: ed654032304e706dcab7ab22514b5e1e93facfa946686a9455676fa220c5aff0859044645b778d0ef63f8482809a21440846d733f263ec007c548a4bba18c891
-  data.tar.gz: 8a1d4ae16ac33747a80b2a1af1d04095358265d7949e43b66ba3e77aa02d310c9ebc32891d315ce9351197223e6f04e238cb1bd1eb57c8ffef2ecf94a120ed31
+  metadata.gz: f39c8a1fb5da46833e2ebb25a7224fd4c7d6bf1b71649d06550e60bd00e6dc084ef6675d9314995a002a507aa3e382efb7947fdc951221612fbc12b730b95fed
+  data.tar.gz: b133873824c59d673b9aed7a8cb45933aa7298bf9b5dc533240e9e43e99820ab1abe0be09874a9d0f5b6f7ce599441b7c1e462ecc83754f0de87dab9ca8ce50b

data/CHANGELOG.md

@@ -1,5 +1,24 @@
 # Changelog
 
+## 6.1.10 - 2022-05-06
+
+- Disable supports_expression_index regardless of CockroachDB version until
+  `ON CONFLICT expression` is supported.
+
+  See https://github.com/cockroachdb/cockroach/issues/67893.
+
+## 6.1.9 - 2022-04-26
+
+- Fix bug where duplicate `rowid` columns would be created when loading
+  a schema dump of a table that was not created with an explicit primary key.
+- Support the NOT VISIBLE syntax from CockroachDB, by using the `hidden`
+  column modifier in the Rails schema.
+
+## 6.1.8 - 2022-03-14
+
+- Add a test helper from https://github.com/rails/rails/pull/40822
+  to be able to test against Rails upstream.
+
 ## 6.1.7 - 2022-03-01
 
 - Fix query to get the CockroachDB version so it does not require any privileges.

data/lib/active_record/connection_adapters/cockroachdb/column.rb

@@ -6,9 +6,10 @@ module ActiveRecord
         # https://github.com/rgeo/activerecord-postgis-adapter/blob/master/lib/active_record/connection_adapters/postgis/spatial_column.rb
         def initialize(name, default, sql_type_metadata = nil, null = true,
                        default_function = nil, collation: nil, comment: nil,
-                       serial: nil, spatial: nil)
+                       serial: nil, spatial: nil, generated: nil, hidden: nil)
           @sql_type_metadata = sql_type_metadata
           @geographic = !!(sql_type_metadata.sql_type =~ /geography\(/i)
+          @hidden = hidden
 
           if spatial
             # This case comes from an entry in the geometry_columns table
@@ -29,7 +30,7 @@ module ActiveRecord
             build_from_sql_type(sql_type_metadata.sql_type)
           end
           super(name, default, sql_type_metadata, null, default_function,
-            collation: collation, comment: comment, serial: serial)
+            collation: collation, comment: comment, serial: serial, generated: generated)
           if spatial? && @srid
             @limit = { srid: @srid, type: to_type_name(geometric_type) }
             @limit[:has_z] = true if @has_z
@@ -52,6 +53,14 @@ module ActiveRecord
           spatial? ? @limit : super
         end
 
+        def virtual?
+          @generated.present?
+        end
+
+        def hidden?
+          @hidden
+        end
+
         def spatial?
           %i[geometry geography].include?(@sql_type_metadata.type)
         end

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

@@ -0,0 +1,18 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module ConnectionAdapters
+    module CockroachDB
+      class SchemaCreation < PostgreSQL::SchemaCreation # :nodoc:
+        private
+          def add_column_options!(sql, options)
+            if options[:hidden]
+              sql << " NOT VISIBLE"
+            end
+            super
+          end
+      end
+    end
+  end
+end
+

data/lib/active_record/connection_adapters/cockroachdb/schema_dumper.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module ActiveRecord
+  module ConnectionAdapters
+    module CockroachDB
+      class SchemaDumper < ConnectionAdapters::PostgreSQL::SchemaDumper # :nodoc:
+        private
+          def prepare_column_options(column)
+            spec = super
+            if column.hidden?
+              spec[:hidden] = true
+            end
+            spec
+          end
+      end
+    end
+  end
+end
+

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

@@ -77,7 +77,7 @@ module ActiveRecord
         # override
         # https://github.com/rails/rails/blob/6-0-stable/activerecord/lib/active_record/connection_adapters/postgresql/schema_statements.rb#L624
         def new_column_from_field(table_name, field)
-          column_name, type, default, notnull, oid, fmod, collation, comment = field
+          column_name, type, default, notnull, oid, fmod, collation, comment, generated, hidden = field
           type_metadata = fetch_type_metadata(column_name, type, oid.to_i, fmod.to_i)
           default_value = extract_value_from_default(default)
           default_function = extract_default_function(default_value, default)
@@ -99,7 +99,9 @@ module ActiveRecord
             collation: collation,
             comment: comment.presence,
             serial: serial,
-            spatial: spatial
+            spatial: spatial,
+            generated: generated,
+            hidden: hidden
           )
         end
 
@@ -186,6 +188,14 @@ module ActiveRecord
           @spatial_column_info ||= {}
           @spatial_column_info[table_name.to_sym] ||= SpatialColumnInfo.new(self, table_name.to_s)
         end
+
+        def create_schema_dumper(options)
+          CockroachDB::SchemaDumper.create(self, options)
+        end
+
+        def schema_creation
+          CockroachDB::SchemaCreation.new(self)
+        end
       end
     end
   end

data/lib/active_record/connection_adapters/cockroachdb_adapter.rb

@@ -1,7 +1,10 @@
 require "rgeo/active_record"
 
 require "active_record/connection_adapters/postgresql_adapter"
+require "active_record/connection_adapters/cockroachdb/attribute_methods"
 require "active_record/connection_adapters/cockroachdb/column_methods"
+require "active_record/connection_adapters/cockroachdb/schema_creation"
+require "active_record/connection_adapters/cockroachdb/schema_dumper"
 require "active_record/connection_adapters/cockroachdb/schema_statements"
 require "active_record/connection_adapters/cockroachdb/referential_integrity"
 require "active_record/connection_adapters/cockroachdb/transaction_manager"
@@ -9,7 +12,6 @@ require "active_record/connection_adapters/cockroachdb/database_statements"
 require "active_record/connection_adapters/cockroachdb/table_definition"
 require "active_record/connection_adapters/cockroachdb/quoting"
 require "active_record/connection_adapters/cockroachdb/type"
-require "active_record/connection_adapters/cockroachdb/attribute_methods"
 require "active_record/connection_adapters/cockroachdb/column"
 require "active_record/connection_adapters/cockroachdb/spatial_column_info"
 require "active_record/connection_adapters/cockroachdb/setup"
@@ -179,7 +181,10 @@ module ActiveRecord
       end
 
       def supports_expression_index?
-        @crdb_version >= 2122
+        # Expression indexes are partially supported by CockroachDB v21.2,
+        # but activerecord requires "ON CONFLICT expression" support.
+        # See https://github.com/cockroachdb/cockroach/issues/67893
+        false
       end
 
       def supports_datetime_with_precision?
@@ -199,7 +204,7 @@ module ActiveRecord
       end
 
       def supports_virtual_columns?
-        false
+        @crdb_version >= 2110
       end
 
       def supports_string_to_array_coercion?
@@ -474,7 +479,8 @@ module ActiveRecord
               SELECT a.attname, format_type(a.atttypid, a.atttypmod),
                      pg_get_expr(d.adbin, d.adrelid), a.attnotnull, a.atttypid, a.atttypmod,
                      c.collname, NULL AS comment,
-                     #{supports_virtual_columns? ? 'attgenerated' : quote('')} as attgenerated
+                     #{supports_virtual_columns? ? 'attgenerated' : quote('')} as attgenerated,
+                     NULL as is_hidden
                 FROM pg_attribute a
                 LEFT JOIN pg_attrdef d ON a.attrelid = d.adrelid AND a.attnum = d.adnum
                 LEFT JOIN pg_type t ON a.atttypid = t.oid
@@ -500,8 +506,18 @@ module ActiveRecord
             dtype = field[1]
             field[1] = crdb_fields[field[0]][2].downcase if re.match(dtype)
             field[7] = crdb_fields[field[0]][1]&.gsub!(/^\'|\'?$/, '')
+            field[9] = true if crdb_fields[field[0]][3]
             field
           end
+          fields.delete_if do |field|
+            # Don't include rowid column if it is hidden and the primary key
+            # is not defined (meaning CRDB implicitly created it).
+            if field[0] == CockroachDBAdapter::DEFAULT_PRIMARY_KEY
+              field[9] && !primary_key(table_name)
+            else
+              false # Keep this entry.
+            end
+          end
         end
 
         # Fetch the column comment because it's faster this way
@@ -512,7 +528,7 @@ module ActiveRecord
         def crdb_column_definitions(table_name)
           fields = \
           query(<<~SQL, "SCHEMA")
-            SELECT c.column_name, c.column_comment, c.crdb_sql_type
+            SELECT c.column_name, c.column_comment, c.crdb_sql_type, c.is_hidden::BOOLEAN
               FROM information_schema.columns c
             WHERE c.table_name = #{quote(table_name)}
           SQL

data/lib/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module ActiveRecord
-  COCKROACH_DB_ADAPTER_VERSION = "6.1.7"
+  COCKROACH_DB_ADAPTER_VERSION = "6.1.10"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: activerecord-cockroachdb-adapter
 version: !ruby/object:Gem::Version
-  version: 6.1.7
+  version: 6.1.10
 platform: ruby
 authors:
 - Cockroach Labs
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-03-01 00:00:00.000000000 Z
+date: 2022-05-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activerecord
@@ -87,6 +87,8 @@ files:
 - lib/active_record/connection_adapters/cockroachdb/oid/type_map_initializer.rb
 - lib/active_record/connection_adapters/cockroachdb/quoting.rb
 - lib/active_record/connection_adapters/cockroachdb/referential_integrity.rb
+- lib/active_record/connection_adapters/cockroachdb/schema_creation.rb
+- lib/active_record/connection_adapters/cockroachdb/schema_dumper.rb
 - lib/active_record/connection_adapters/cockroachdb/schema_statements.rb
 - lib/active_record/connection_adapters/cockroachdb/setup.rb
 - lib/active_record/connection_adapters/cockroachdb/spatial_column_info.rb