Move query limit exception to EE only

No longer needed in FOSS because queries are again below the threshold.

app/controllers/projects/merge_requests/creations_controller.rb

@@ -6,7 +6,6 @@ class Projects::MergeRequests::CreationsController < Projects::MergeRequests::Ap
   include RendersCommits
 
   skip_before_action :merge_request
-  before_action :disable_query_limiting, only: [:create]
   before_action :authorize_create_merge_request_from!
   before_action :apply_diff_view_cookie!, only: [:diffs, :diff_for_path]
   before_action :build_merge_request, except: [:create]
@@ -133,13 +132,11 @@ class Projects::MergeRequests::CreationsController < Projects::MergeRequests::Ap
   end
   # rubocop: enable CodeReuse/ActiveRecord
 
-  def disable_query_limiting
-    Gitlab::QueryLimiting.disable!('https://gitlab.com/gitlab-org/gitlab/-/issues/20801')
-  end
-
   def incr_count_webide_merge_request
     return if params[:nav_source] != 'webide'
 
     Gitlab::UsageDataCounters::WebIdeCounter.increment_merge_requests_count
   end
 end
+
+Projects::MergeRequests::CreationsController.prepend_ee_mod

ee/app/controllers/ee/projects/merge_requests/creations_controller.rb

+# frozen_string_literal: true
+
+module EE
+  module Projects
+    module MergeRequests
+      module CreationsController
+        extend ActiveSupport::Concern
+
+        prepended do
+          before_action :disable_query_limiting, only: [:create]
+        end
+
+        private
+
+        def disable_query_limiting
+          ::Gitlab::QueryLimiting.disable!('https://gitlab.com/gitlab-org/gitlab/-/issues/20801')
+        end
+      end
+    end
+  end
+end

ee/spec/controllers/projects/merge_requests/creations_controller_spec.rb

@@ -139,5 +139,11 @@ RSpec.describe Projects::MergeRequests::CreationsController do
         end
       end
     end
+
+    it 'disables query limiting' do
+      expect(Gitlab::QueryLimiting).to receive(:disable!)
+
+      create_merge_request
+    end
   end
 end