Add ExactCountStrategy.

1 files changed, 13 insertions, 0 deletions

diff --git a/lib/gitlab/database/count.rb b/lib/gitlab/database/count.rb
index fc61ee48832..7b7fd3148fa 100644
--- a/lib/gitlab/database/count.rb
+++ b/lib/gitlab/database/count.rb
@@ -57,6 +57,19 @@ module Gitlab
         ReltuplesCountStrategy.new(table_names).count
       end
 
+      class ExactCountStrategy
+        attr_reader :models
+        def initialize(models)
+          @models = models
+        end
+
+        def count
+          models.each_with_object({}) do |model, data|
+            data[model] = model.count
+          end
+        end
+      end
+
       class ReltuplesCountStrategy
         attr_reader :table_names