1 files changed, 24 insertions, 0 deletions
diff --git a/app/graphql/types/countable_connection_type.rb b/app/graphql/types/countable_connection_type.rb
new file mode 100644
index 00000000000..2538366b786
--- /dev/null
+++ b/app/graphql/types/countable_connection_type.rb
@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module Types
+  # rubocop: disable Graphql/AuthorizeTypes
+  class CountableConnectionType < GraphQL::Types::Relay::BaseConnection
+    field :count, Integer, null: false,
+          description: 'Total count of collection'
+
+    def count
+      # rubocop: disable CodeReuse/ActiveRecord
+      relation = object.items
+
+      # sometimes relation is an Array
+      relation = relation.reorder(nil) if relation.respond_to?(:reorder)
+      # rubocop: enable CodeReuse/ActiveRecord
+
+      if relation.try(:group_values)&.present?
+        relation.size.keys.size
+      else
+        relation.size
+      end
+    end
+  end
+end