Add latest changes from gitlab-org/gitlab@15-4-stable-eev15.4.0-rc42

1 files changed, 48 insertions, 0 deletions

diff --git a/lib/gitlab/database/partitioning_migration_helpers/table_management_helpers.rb b/lib/gitlab/database/partitioning_migration_helpers/table_management_helpers.rb
index a541ecf5316..695a5d7ec77 100644
--- a/lib/gitlab/database/partitioning_migration_helpers/table_management_helpers.rb
+++ b/lib/gitlab/database/partitioning_migration_helpers/table_management_helpers.rb
@@ -251,6 +251,54 @@ module Gitlab
           create_sync_trigger(source_table_name, trigger_name, function_name)
         end
 
+        def prepare_constraint_for_list_partitioning(table_name:, partitioning_column:, parent_table_name:, initial_partitioning_value:)
+          validate_not_in_transaction!(:prepare_constraint_for_list_partitioning)
+
+          Gitlab::Database::Partitioning::ConvertTableToFirstListPartition
+            .new(migration_context: self,
+                 table_name: table_name,
+                 parent_table_name: parent_table_name,
+                 partitioning_column: partitioning_column,
+                 zero_partition_value: initial_partitioning_value
+                ).prepare_for_partitioning
+        end
+
+        def revert_preparing_constraint_for_list_partitioning(table_name:, partitioning_column:, parent_table_name:, initial_partitioning_value:)
+          validate_not_in_transaction!(:revert_preparing_constraint_for_list_partitioning)
+
+          Gitlab::Database::Partitioning::ConvertTableToFirstListPartition
+            .new(migration_context: self,
+                 table_name: table_name,
+                 parent_table_name: parent_table_name,
+                 partitioning_column: partitioning_column,
+                 zero_partition_value: initial_partitioning_value
+                ).revert_preparation_for_partitioning
+        end
+
+        def convert_table_to_first_list_partition(table_name:, partitioning_column:, parent_table_name:, initial_partitioning_value:)
+          validate_not_in_transaction!(:convert_table_to_first_list_partition)
+
+          Gitlab::Database::Partitioning::ConvertTableToFirstListPartition
+            .new(migration_context: self,
+                 table_name: table_name,
+                 parent_table_name: parent_table_name,
+                 partitioning_column: partitioning_column,
+                 zero_partition_value: initial_partitioning_value
+                ).partition
+        end
+
+        def revert_converting_table_to_first_list_partition(table_name:, partitioning_column:, parent_table_name:, initial_partitioning_value:)
+          validate_not_in_transaction!(:revert_converting_table_to_first_list_partition)
+
+          Gitlab::Database::Partitioning::ConvertTableToFirstListPartition
+            .new(migration_context: self,
+                 table_name: table_name,
+                 parent_table_name: parent_table_name,
+                 partitioning_column: partitioning_column,
+                 zero_partition_value: initial_partitioning_value
+                ).revert_partitioning
+        end
+
         private
 
         def assert_table_is_allowed(table_name)