diff options
Diffstat (limited to 'db/post_migrate')
-rw-r--r-- | db/post_migrate/20230317151841_remove_from_to_state_constraint.rb | 21 | ||||
-rw-r--r-- | db/post_migrate/20230322085041_remove_user_namespace_records_from_vsa_aggregation.rb | 26 |
2 files changed, 47 insertions, 0 deletions
diff --git a/db/post_migrate/20230317151841_remove_from_to_state_constraint.rb b/db/post_migrate/20230317151841_remove_from_to_state_constraint.rb new file mode 100644 index 00000000000..21913d394d1 --- /dev/null +++ b/db/post_migrate/20230317151841_remove_from_to_state_constraint.rb @@ -0,0 +1,21 @@ +# frozen_string_literal: true + +class RemoveFromToStateConstraint < Gitlab::Database::Migration[2.1] + disable_ddl_transaction! + + def up + constraint_name = check_constraint_name( + 'vulnerability_state_transitions', + 'fully_qualified_table_name', + 'state_not_equal') + remove_check_constraint(:vulnerability_state_transitions, constraint_name) + end + + def down + constraint_name = check_constraint_name( + 'vulnerability_state_transitions', + 'fully_qualified_table_name', + 'state_not_equal') + add_check_constraint(:vulnerability_state_transitions, '(from_state != to_state)', constraint_name) + end +end diff --git a/db/post_migrate/20230322085041_remove_user_namespace_records_from_vsa_aggregation.rb b/db/post_migrate/20230322085041_remove_user_namespace_records_from_vsa_aggregation.rb new file mode 100644 index 00000000000..6fc23c742b9 --- /dev/null +++ b/db/post_migrate/20230322085041_remove_user_namespace_records_from_vsa_aggregation.rb @@ -0,0 +1,26 @@ +# frozen_string_literal: true + +class RemoveUserNamespaceRecordsFromVsaAggregation < Gitlab::Database::Migration[2.1] + BATCH_SIZE = 100 + + disable_ddl_transaction! + + restrict_gitlab_migration gitlab_schema: :gitlab_main + + def up + aggregations_model = define_batchable_model('analytics_cycle_analytics_aggregations') + namespaces_model = define_batchable_model('namespaces') + + aggregations_model.each_batch(of: BATCH_SIZE) do |relation| + inner_query = namespaces_model + .where(type: 'Group') + .where(aggregations_model.arel_table[:group_id].eq(namespaces_model.arel_table[:id])) + + relation.where('NOT EXISTS (?)', inner_query).delete_all + end + end + + def down + # noop + end +end |