Patch a different method to prevent re-iteration

ee/lib/ee/peek/views/active_record.rb

@@ -17,16 +17,14 @@ module EE
           detail
         end
 
-        override :summary
-        def summary
+        override :count_summary
+        def count_summary(item, count)
+          super
+
           if ::Gitlab::Database::LoadBalancing.enable?
-            detail_store.each_with_object(super) do |item, count|
             count[item[:db_role]] ||= 0
             count[item[:db_role]] += 1
           end
-          else
-            super
-          end
         end
       end
     end

lib/peek/views/active_record.rb

@@ -29,6 +29,11 @@ module Peek
 
       def summary
         detail_store.each_with_object({}) do |item, count|
+          count_summary(item, count)
+        end
+      end
+
+      def count_summary(item, count)
         if item[:cached].present?
           count[item[:cached]] ||= 0
           count[item[:cached]] += 1
@@ -39,7 +44,6 @@ module Peek
           count[item[:transaction]] += 1
         end
       end
-      end
 
       def setup_subscribers
         super