summaryrefslogtreecommitdiff
path: root/db/migrate/20211011141242_create_namespaces_sync_trigger.rb
blob: 91f64709f28680927f77da50cc0fedb8b0098cb8 (plain)
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
# frozen_string_literal: true

class CreateNamespacesSyncTrigger < Gitlab::Database::Migration[1.0]
  include Gitlab::Database::SchemaHelpers

  enable_lock_retries!

  TABLE_NAME = 'namespaces'
  EVENT_TABLE_NAME = 'namespaces_sync_events'
  FUNCTION_NAME = 'insert_namespaces_sync_event'
  TRIGGER_ON_INSERT = 'trigger_namespaces_parent_id_on_insert'
  TRIGGER_ON_UPDATE = 'trigger_namespaces_parent_id_on_update'

  def up
    create_trigger_function(FUNCTION_NAME) do
      <<~SQL
        INSERT INTO #{EVENT_TABLE_NAME} (namespace_id)
        VALUES(COALESCE(NEW.id, OLD.id));
        RETURN NULL;
      SQL
    end

    create_trigger(TABLE_NAME, TRIGGER_ON_INSERT, FUNCTION_NAME, fires: 'AFTER INSERT')

    create_trigger(TABLE_NAME, TRIGGER_ON_UPDATE, FUNCTION_NAME, fires: 'AFTER UPDATE') do
      <<~SQL
        WHEN (OLD.parent_id IS DISTINCT FROM NEW.parent_id)
      SQL
    end
  end

  def down
    drop_trigger(TABLE_NAME, TRIGGER_ON_INSERT)
    drop_trigger(TABLE_NAME, TRIGGER_ON_UPDATE)
    drop_function(FUNCTION_NAME)
  end
end