gitlab-org--gitlab-foss/db/migrate/20210906100021_delete_proje...

32 lines
840 B
Ruby

# frozen_string_literal: true
class DeleteProjectNamespaceTrigger < Gitlab::Database::Migration[1.0]
include Gitlab::Database::SchemaHelpers
TRIGGER_NAME = "trigger_delete_project_namespace_on_project_delete"
FUNCTION_NAME = 'delete_associated_project_namespace'
def up
create_trigger_function(FUNCTION_NAME, replace: true) do
<<~SQL
DELETE FROM namespaces
WHERE namespaces.id = OLD.project_namespace_id AND
namespaces.type = 'Project';
RETURN NULL;
SQL
end
execute(<<~SQL.squish)
CREATE TRIGGER #{TRIGGER_NAME}
AFTER DELETE ON projects FOR EACH ROW
WHEN (OLD.project_namespace_id IS NOT NULL)
EXECUTE FUNCTION #{FUNCTION_NAME}();
SQL
end
def down
drop_trigger(:projects, TRIGGER_NAME)
drop_function(FUNCTION_NAME)
end
end