Move disable_prepared_statements to Core

We use this method in initializer for Load Balancing

ee/lib/ee/gitlab/database.rb

@@ -17,12 +17,6 @@ module EE
           !Postgresql::ReplicationSlot.lag_too_great?
         end
 
-        # Disables prepared statements for the current database connection.
-        def disable_prepared_statements
-          config = ::Gitlab::Database.config.merge(prepared_statements: false)
-          ActiveRecord::Base.establish_connection(config)
-        end
-
         def geo_uncached_queries(&block)
           raise 'No block given' unless block_given?
 

ee/spec/lib/ee/gitlab/database_spec.rb

@@ -60,28 +60,6 @@ RSpec.describe Gitlab::Database do
     end
   end
 
-  describe '.disable_prepared_statements' do
-    around do |example|
-      original_config = ::Gitlab::Database.config
-
-      example.run
-
-      ActiveRecord::Base.establish_connection(original_config)
-    end
-
-    it 'disables prepared statements' do
-      ActiveRecord::Base.establish_connection(::Gitlab::Database.config.merge(prepared_statements: true))
-      expect(ActiveRecord::Base.connection.prepared_statements).to eq(true)
-
-      expect(ActiveRecord::Base).to receive(:establish_connection)
-        .with(a_hash_including({ 'prepared_statements' => false })).and_call_original
-
-      described_class.disable_prepared_statements
-
-      expect(ActiveRecord::Base.connection.prepared_statements).to eq(false)
-    end
-  end
-
   describe '.geo_uncached_queries' do
     context 'when no block is given' do
       it 'raises error' do

lib/gitlab/database.rb

@@ -89,6 +89,11 @@ module Gitlab
       end
     end
 
+    # Disables prepared statements for the current database connection.
+    def self.disable_prepared_statements
+      ActiveRecord::Base.establish_connection(config.merge(prepared_statements: false))
+    end
+
     # @deprecated
     def self.postgresql?
       adapter_name.casecmp('postgresql') == 0

spec/lib/gitlab/database_spec.rb

@@ -65,6 +65,28 @@ RSpec.describe Gitlab::Database do
     end
   end
 
+  describe '.disable_prepared_statements' do
+    around do |example|
+      original_config = ::Gitlab::Database.config
+
+      example.run
+
+      ActiveRecord::Base.establish_connection(original_config)
+    end
+
+    it 'disables prepared statements' do
+      ActiveRecord::Base.establish_connection(::Gitlab::Database.config.merge(prepared_statements: true))
+      expect(ActiveRecord::Base.connection.prepared_statements).to eq(true)
+
+      expect(ActiveRecord::Base).to receive(:establish_connection)
+        .with(a_hash_including({ 'prepared_statements' => false })).and_call_original
+
+      described_class.disable_prepared_statements
+
+      expect(ActiveRecord::Base.connection.prepared_statements).to eq(false)
+    end
+  end
+
   describe '.postgresql?' do
     subject { described_class.postgresql? }