show connection counts within one type. #299

datacite · Jun 30, 2019 · 026d8de · 026d8de
1 parent 37f10cb
commit 026d8de
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/app/graphql/types/dataset_connection_with_meta_type.rb b/app/graphql/types/dataset_connection_with_meta_type.rb
@@ -5,6 +5,7 @@ class DatasetConnectionWithMetaType < BaseConnection
   field_class GraphQL::Cache::Field
 
   field :total_count, Integer, null: false, cache: true
+  field :dataset_connection_count, Integer, null: false, cache: true
   field :publication_connection_count, Integer, null: false, cache: true
   field :software_connection_count, Integer, null: false, cache: true
   field :researcher_connection_count, Integer, null: false, cache: true
@@ -16,6 +17,10 @@ def total_count
     Doi.query(args[:query], resource_type_id: "Dataset", state: "findable", page: { number: 1, size: args[:first] }).results.total
   end
 
+  def dataset_connection_count
+    Event.query(nil, citation_type: "Dataset-Dataset").results.total
+  end
+
   def publication_connection_count
     Event.query(nil, citation_type: "Dataset-ScholarlyArticle").results.total
   end

diff --git a/app/graphql/types/publication_connection_with_meta_type.rb b/app/graphql/types/publication_connection_with_meta_type.rb
@@ -5,6 +5,7 @@ class PublicationConnectionWithMetaType < BaseConnection
   field_class GraphQL::Cache::Field
 
   field :total_count, Integer, null: false, cache: true
+  field :publication_connection_count, Integer, null: false, cache: true
   field :dataset_connection_count, Integer, null: false, cache: true
   field :software_connection_count, Integer, null: false, cache: true
   field :researcher_connection_count, Integer, null: false, cache: true
@@ -16,6 +17,10 @@ def total_count
     Doi.query(args[:query], resource_type_id: "Text", state: "findable", page: { number: 1, size: args[:first] }).results.total
   end
 
+  def publication_connection_count
+    Event.query(nil, citation_type: "ScholarlyArticle-ScholarlyArticle").results.total
+  end
+
   def dataset_connection_count
     Event.query(nil, citation_type: "Dataset-ScholarlyArticle").results.total
   end

diff --git a/app/graphql/types/software_connection_with_meta_type.rb b/app/graphql/types/software_connection_with_meta_type.rb
@@ -5,6 +5,7 @@ class SoftwareConnectionWithMetaType < BaseConnection
   field_class GraphQL::Cache::Field
 
   field :total_count, Integer, null: false, cache: true
+  field :software_connection_count, Integer, null: false, cache: true
   field :publication_connection_count, Integer, null: false, cache: true
   field :dataset_connection_count, Integer, null: false, cache: true
   field :researcher_connection_count, Integer, null: false, cache: true
@@ -16,6 +17,10 @@ def total_count
     Doi.query(args[:query], resource_type_id: "Software", state: "findable", page: { number: 1, size: args[:first] }).results.total
   end
 
+  def software_connection_count
+    Event.query(nil, citation_type: "SoftwareSourceCode-SoftwareSourceCode").results.total
+  end
+
   def publication_connection_count
     Event.query(nil, citation_type: "ScholarlyArticle-SoftwareSourceCode").results.total
   end