1 files changed, 47 insertions, 0 deletions
diff --git a/app/models/concerns/cross_database_ignored_tables.rb b/app/models/concerns/cross_database_ignored_tables.rb
new file mode 100644
index 00000000000..c97e405cce4
--- /dev/null
+++ b/app/models/concerns/cross_database_ignored_tables.rb
@@ -0,0 +1,47 @@
+# frozen_string_literal: true
+
+module CrossDatabaseIgnoredTables
+  extend ActiveSupport::Concern
+
+  class_methods do
+    def cross_database_ignore_tables(tables, options = {})
+      raise "missing issue url" if options[:url].blank?
+
+      options[:on] = %I[save destroy] if options[:on].blank?
+      events = Array.wrap(options[:on])
+      tables = Array.wrap(tables)
+
+      events.each do |event|
+        register_ignored_cross_database_event(tables, event, options)
+      end
+    end
+
+    private
+
+    def register_ignored_cross_database_event(tables, event, options)
+      case event
+      when :save
+        around_save(prepend: true) { |_, blk| temporary_ignore_cross_database_tables(tables, options, &blk) }
+      when :create
+        around_create(prepend: true) { |_, blk| temporary_ignore_cross_database_tables(tables, options, &blk) }
+      when :update
+        around_update(prepend: true) { |_, blk| temporary_ignore_cross_database_tables(tables, options, &blk) }
+      when :destroy
+        around_destroy(prepend: true) { |_, blk| temporary_ignore_cross_database_tables(tables, options, &blk) }
+      else
+        raise "Unknown #{event}"
+      end
+    end
+  end
+
+  private
+
+  def temporary_ignore_cross_database_tables(tables, options, &blk)
+    return yield unless options[:if].nil? || instance_eval(&options[:if])
+
+    url = options[:url]
+    Gitlab::Database::QueryAnalyzers::PreventCrossDatabaseModification.temporary_ignore_tables_in_transaction(
+      tables, url: url, &blk
+    )
+  end
+end