1 files changed, 5 insertions, 1 deletions
diff --git a/lib/gitlab/database/postgres_hll/batch_distinct_counter.rb b/lib/gitlab/database/postgres_hll/batch_distinct_counter.rb
index 2e3f674cf82..4e973efebca 100644
--- a/lib/gitlab/database/postgres_hll/batch_distinct_counter.rb
+++ b/lib/gitlab/database/postgres_hll/batch_distinct_counter.rb
@@ -57,7 +57,7 @@ module Gitlab
         # @param finish final pkey range
         # @return [Gitlab::Database::PostgresHll::Buckets] HyperLogLog data structure instance that can estimate number of unique elements
         def execute(batch_size: nil, start: nil, finish: nil)
-          raise 'BatchCount can not be run inside a transaction' if ActiveRecord::Base.connection.transaction_open? # rubocop: disable Database/MultipleDatabases
+          raise 'BatchCount can not be run inside a transaction' if transaction_open?
 
           batch_size ||= DEFAULT_BATCH_SIZE
           start = actual_start(start)
@@ -79,6 +79,10 @@ module Gitlab
 
         private
 
+        def transaction_open?
+          @relation.connection.transaction_open?
+        end
+
         def unwanted_configuration?(start, finish, batch_size)
           batch_size <= MIN_REQUIRED_BATCH_SIZE ||
             (finish - start) >= MAX_DATA_VOLUME ||