Add index on group_id column concurrentlyfix-events-migration

This fixes previously added migration which caused timeouts on
big events table.

1 files changed, 14 insertions, 2 deletions

diff --git a/db/migrate/20190826100605_add_group_column_to_events.rb b/db/migrate/20190826100605_add_group_column_to_events.rb
index cd7b2b1d96a..a940764bc9f 100644
--- a/db/migrate/20190826100605_add_group_column_to_events.rb
+++ b/db/migrate/20190826100605_add_group_column_to_events.rb
@@ -1,9 +1,21 @@
 # frozen_string_literal: true
 
 class AddGroupColumnToEvents < ActiveRecord::Migration[5.2]
+  include Gitlab::Database::MigrationHelpers
+
   DOWNTIME = false
 
-  def change
-    add_reference :events, :group, index: true, foreign_key: { to_table: :namespaces, on_delete: :cascade }
+  disable_ddl_transaction!
+
+  def up
+    add_column(:events, :group_id, :bigint) unless column_exists?(:events, :group_id)
+    add_concurrent_foreign_key(:events, :namespaces, column: :group_id, on_delete: :cascade)
+    add_concurrent_index(:events, :group_id)
+  end
+
+  def down
+    remove_foreign_key_without_error(:events, column: :group_id)
+    remove_concurrent_index(:events, :group_id) if index_exists?(:events, :group_id)
+    remove_column(:events, :group_id) if column_exists?(:events, :group_id)
   end
 end