41 lines
1.1 KiB
Ruby
41 lines
1.1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
class AddUpdateVulnerabilityReadsTrigger < Gitlab::Database::Migration[1.0]
|
||
|
|
include Gitlab::Database::SchemaHelpers
|
||
|
|
|
||
|
|
TRIGGER_NAME = 'trigger_update_vulnerability_reads_on_vulnerability_update'
|
||
|
|
FUNCTION_NAME = 'update_vulnerability_reads_from_vulnerability'
|
||
|
|
|
||
|
|
def up
|
||
|
|
create_trigger_function(FUNCTION_NAME, replace: true) do
|
||
|
|
<<~SQL
|
||
|
|
UPDATE
|
||
|
|
vulnerability_reads
|
||
|
|
SET
|
||
|
|
severity = NEW.severity,
|
||
|
|
state = NEW.state,
|
||
|
|
resolved_on_default_branch = NEW.resolved_on_default_branch
|
||
|
|
WHERE vulnerability_id = NEW.id;
|
||
|
|
RETURN NULL;
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
execute(<<~SQL)
|
||
|
|
CREATE TRIGGER #{TRIGGER_NAME}
|
||
|
|
AFTER UPDATE ON vulnerabilities
|
||
|
|
FOR EACH ROW
|
||
|
|
WHEN (
|
||
|
|
OLD.severity IS DISTINCT FROM NEW.severity OR
|
||
|
|
OLD.state IS DISTINCT FROM NEW.state OR
|
||
|
|
OLD.resolved_on_default_branch IS DISTINCT FROM NEW.resolved_on_default_branch
|
||
|
|
)
|
||
|
|
EXECUTE PROCEDURE #{FUNCTION_NAME}();
|
||
|
|
SQL
|
||
|
|
end
|
||
|
|
|
||
|
|
def down
|
||
|
|
drop_trigger(:vulnerabilities, TRIGGER_NAME)
|
||
|
|
drop_function(FUNCTION_NAME)
|
||
|
|
end
|
||
|
|
end
|