File: 20241023134305_modify_update_vulnerability_reads_trigger_function.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (40 lines) | stat: -rw-r--r-- 1,020 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# frozen_string_literal: true

class ModifyUpdateVulnerabilityReadsTriggerFunction < Gitlab::Database::Migration[2.2]
  include Gitlab::Database::SchemaHelpers

  milestone '17.6'

  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,
          auto_resolved = NEW.auto_resolved
        WHERE vulnerability_id = NEW.id;
        RETURN NULL;
      SQL
    end
  end

  def down
    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
  end
end