code0-zero_track 0.0.9 → 0.0.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5af11d8b9714d7d557f46cfe5ea9dbd6702d3bc85f5aa67e9ce09e44c6963238
-  data.tar.gz: 5fc99a08521c502aec93ebe725d9c82c18d9b49e6a7120fba5127db9bb620f6b
+  metadata.gz: b7bc699d8f0bb3f1237eeb8ff75493624930bc8cce4b97ba1666709c494d4d04
+  data.tar.gz: aac0a33b9d880e0fd9f5a6297b76f0dd32c5f42f6beccad0afeea412caa78bd9
 SHA512:
-  metadata.gz: fded2cb69f93aff4704e89b6c3094a0c9dd47a9891748ddd40af68e72eb2d432c150ecd62bf104cd864ceee65a5e782b3eef656fd7ac69ea338898f2cae713a7
-  data.tar.gz: db2a0b2c23fa93a22442e7f8e0c956ddae670a05bf4105c0a77ea98968c5827541b442b119331bdfb6f8b564b151ccedb9b1418cf45a8dd2fe243c7402748bb8
+  metadata.gz: 60edef2bb7d288aeb3127068d6d58099cb409b659a5401a62db2844317cd4aa7048f7b8146ae59234f0e2c3c20e4af1a2e94ef91ac0561c62987f9977193e1c6
+  data.tar.gz: 02f209d8121437c829735e11bb9896c0991f2927d484da5c5f8f1f52c1cb7dd6d2363581f4d9886f285052339685ae942972666d497db6e03e6f25a0ed5b0c86

data/README.md

@@ -189,6 +189,9 @@ When tables have foreign key relationships, registration order and retention con
 - A child table's `retain_for` must be less than or equal to the parent table's `retain_for`.
   If a child retains partitions longer than its parent, dropping the parent partition will
   fail because the child's foreign key still references it.
+- Foreign keys between the partitioned tables must be declared with `drop_foreign_keys_on_detach`
+  on the model. Otherwise, the foreign key of a detached partition will block detaching partitions
+  of the referenced table.
 
 #### Schema Cleaner Integration
 

data/lib/code0/zero_track/database/partitioning/partition_manager.rb

@@ -64,6 +64,8 @@ module Code0
               model.partitioning_strategy.partitions_to_detach.each do |partition|
                 detach_partition(partition, connection)
               end
+
+              drop_declared_foreign_keys_from_detached_partitions(connection)
             end
           end
 
@@ -109,6 +111,73 @@ module Code0
             )
           end
 
+          def drop_declared_foreign_keys_from_detached_partitions(connection)
+            constraint_names = model.foreign_keys_to_drop_on_detach || []
+            return if constraint_names.empty?
+
+            warn_about_unknown_foreign_keys(constraint_names, connection)
+
+            detached_foreign_keys(constraint_names, connection).each do |qualified_table, constraint_name|
+              connection.execute(
+                "ALTER TABLE #{qualified_table} " \
+                "DROP CONSTRAINT #{connection.quote_column_name(constraint_name)}"
+              )
+
+              logger.info(
+                message: 'Dropped foreign key from detached partition',
+                table_name: model.table_name,
+                partition_name: qualified_table,
+                constraint_name: constraint_name
+              )
+            end
+          end
+
+          # We check the parent table because we drop the FK from the partition
+          # when detaching. A missing FK on the detached partition is an expected state
+          def warn_about_unknown_foreign_keys(constraint_names, connection)
+            existing = parent_foreign_key_names(connection)
+            unknown = constraint_names - existing
+            return if unknown.empty?
+
+            logger.warn(
+              message: 'Configured foreign keys to drop on detach do not exist on the partitioned table',
+              table_name: model.table_name,
+              constraint_names: unknown
+            )
+          end
+
+          def parent_foreign_key_names(connection)
+            connection.select_values(<<~SQL.squish)
+              SELECT con.conname
+              FROM pg_catalog.pg_constraint con
+              WHERE con.contype = 'f'
+                AND con.conrelid = #{connection.quote(model.table_name)}::regclass
+            SQL
+          end
+
+          def detached_foreign_keys(constraint_names, connection)
+            schema = Rails.application.config.zero_track.db_partitioning.dynamic_partition_schema
+            quoted_names = constraint_names.map { |name| connection.quote(name) }.join(', ')
+
+            rows = connection.select_rows(<<~SQL.squish)
+              SELECT (quote_ident(n.nspname) || '.' || quote_ident(c.relname)) AS qualified_table,
+                     con.conname AS constraint_name
+              FROM pg_catalog.pg_constraint con
+              JOIN pg_catalog.pg_class c ON c.oid = con.conrelid
+              JOIN pg_catalog.pg_namespace n ON n.oid = c.relnamespace
+              WHERE con.contype = 'f'
+                AND con.conname IN (#{quoted_names})
+                AND n.nspname = #{connection.quote(schema)}
+                AND c.relkind = 'r'
+                AND NOT c.relispartition
+                AND NOT EXISTS (
+                  SELECT 1 FROM pg_catalog.pg_inherits i WHERE i.inhrelid = c.oid
+                )
+            SQL
+
+            rows.map { |qualified_table, constraint_name| [qualified_table, constraint_name] }
+          end
+
           def drop_partition(detached_partition, connection)
             schema_name = connection.quote_table_name(detached_partition.schema)
             partition_name = connection.quote_table_name(detached_partition.name)

data/lib/code0/zero_track/database/partitioning/partitioned_table.rb

@@ -25,6 +25,16 @@ module Code0
 
               @partitioning_strategy = strategy_class.new(self, column, **kwargs)
             end
+
+            def drop_foreign_keys_on_detach(*constraint_names)
+              @foreign_keys_to_drop_on_detach ||= []
+              @foreign_keys_to_drop_on_detach.concat(constraint_names.flatten.map(&:to_s))
+              @foreign_keys_to_drop_on_detach
+            end
+
+            def foreign_keys_to_drop_on_detach
+              @foreign_keys_to_drop_on_detach || []
+            end
           end
         end
       end

data/lib/code0/zero_track/version.rb

@@ -2,6 +2,6 @@
 
 module Code0
   module ZeroTrack
-    VERSION = '0.0.9'
+    VERSION = '0.0.10'
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: code0-zero_track
 version: !ruby/object:Gem::Version
-  version: 0.0.9
+  version: 0.0.10
 platform: ruby
 authors:
 - Niklas van Schrick