1 files changed, 44 insertions, 0 deletions
diff --git a/db/post_migrate/20210331105335_drop_non_partitioned_audit_events.rb b/db/post_migrate/20210331105335_drop_non_partitioned_audit_events.rb
new file mode 100644
index 00000000000..7c32fc61711
--- /dev/null
+++ b/db/post_migrate/20210331105335_drop_non_partitioned_audit_events.rb
@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+class DropNonPartitionedAuditEvents < ActiveRecord::Migration[6.0]
+  include Gitlab::Database::MigrationHelpers
+  include Gitlab::Database::PartitioningMigrationHelpers::TableManagementHelpers
+
+  DOWNTIME = false
+
+  def up
+    drop_nonpartitioned_archive_table(:audit_events)
+  end
+
+  def down
+    execute(<<~SQL)
+      CREATE TABLE audit_events_archived (
+        id integer NOT NULL,
+        author_id integer NOT NULL,
+        entity_id integer NOT NULL,
+        entity_type character varying NOT NULL,
+        details text,
+        created_at timestamp without time zone,
+        ip_address inet,
+        author_name text,
+        entity_path text,
+        target_details text,
+        target_type text,
+        target_id bigint,
+        CONSTRAINT check_492aaa021d CHECK ((char_length(entity_path) <= 5500)),
+        CONSTRAINT check_82294106dd CHECK ((char_length(target_type) <= 255)),
+        CONSTRAINT check_83ff8406e2 CHECK ((char_length(author_name) <= 255)),
+        CONSTRAINT check_d493ec90b5 CHECK ((char_length(target_details) <= 5500))
+      );
+
+      ALTER TABLE ONLY audit_events_archived ADD CONSTRAINT audit_events_archived_pkey PRIMARY KEY (id);
+
+      CREATE INDEX analytics_index_audit_events_on_created_at_and_author_id ON audit_events_archived USING btree (created_at, author_id);
+      CREATE INDEX idx_audit_events_on_entity_id_desc_author_id_created_at ON audit_events_archived USING btree (entity_id, entity_type, id DESC, author_id, created_at);
+    SQL
+
+    with_lock_retries do
+      create_trigger_to_sync_tables(:audit_events, :audit_events_archived, 'id')
+    end
+  end
+end