41 lines
1.1 KiB
Ruby
41 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class RecreateLooseFkInsertFunction < Gitlab::Database::Migration[1.0]
|
||
|
|
include Gitlab::Database::MigrationHelpers::LooseForeignKeyHelpers
|
||
|
|
|
||
|
|
def up
|
||
|
|
execute(<<~SQL)
|
||
|
|
CREATE OR REPLACE FUNCTION #{DELETED_RECORDS_INSERT_FUNCTION_NAME}()
|
||
|
|
RETURNS TRIGGER AS
|
||
|
|
$$
|
||
|
|
BEGIN
|
||
|
|
INSERT INTO loose_foreign_keys_deleted_records
|
||
|
|
(partition, fully_qualified_table_name, primary_key_value)
|
||
|
|
SELECT 1, TG_TABLE_SCHEMA || '.' || TG_TABLE_NAME, old_table.id FROM old_table
|
||
|
|
ON CONFLICT DO NOTHING;
|
||
|
|
|
||
|
|
RETURN NULL;
|
||
|
|
END
|
||
|
|
$$ LANGUAGE PLPGSQL
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
# old function
|
||
|
|
execute(<<~SQL)
|
||
|
|
CREATE OR REPLACE FUNCTION #{DELETED_RECORDS_INSERT_FUNCTION_NAME}()
|
||
|
|
RETURNS TRIGGER AS
|
||
|
|
$$
|
||
|
|
BEGIN
|
||
|
|
INSERT INTO loose_foreign_keys_deleted_records
|
||
|
|
(deleted_table_name, deleted_table_primary_key_value)
|
||
|
|
SELECT TG_TABLE_NAME, old_table.id FROM old_table
|
||
|
|
ON CONFLICT DO NOTHING;
|
||
|
|
|
||
|
|
RETURN NULL;
|
||
|
|
END
|
||
|
|
$$ LANGUAGE PLPGSQL
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
end
|