separate methods find_by_id and find_by_ids

datacite · Feb 18, 2020 · 2ce9d90 · 2ce9d90
1 parent a094bea
commit 2ce9d90
Show file tree

Hide file tree

Showing 6 changed files with 17 additions and 6 deletions.
diff --git a/app/controllers/dois_controller.rb b/app/controllers/dois_controller.rb
@@ -49,7 +49,7 @@ def index
       if params[:id].present?
         response = Doi.find_by_id(params[:id])
       elsif params[:ids].present?
-        response = Doi.find_by_id(params[:ids], page: page, sort: sort)
+        response = Doi.find_by_ids(params[:ids], page: page, sort: sort)
       else
         response = Doi.query(params[:query],
                             state: params[:state],

diff --git a/app/controllers/works_controller.rb b/app/controllers/works_controller.rb
@@ -31,7 +31,7 @@ def index
       if params[:id].present?
         response = Doi.find_by_id(params[:id])
       elsif params[:ids].present?
-        response = Doi.find_by_id(params[:ids], page: page, sort: sort)
+        response = Doi.find_by_ids(params[:ids], page: page, sort: sort)
       else
         response = Doi.query(params[:query],
                             exclude_registration_agencies: true,

diff --git a/app/graphql/elasticsearch_loader.rb b/app/graphql/elasticsearch_loader.rb
@@ -6,7 +6,7 @@ def initialize(model)
   end
 
   def perform(ids)
-    @model.find_by_id(ids).results.each { |record| fulfill(record.uid, record) }
+    @model.find_by_ids(ids).results.each { |record| fulfill(record.uid, record) }
     ids.each { |id| fulfill(id, nil) unless fulfilled?(id) }
   end
 end
diff --git a/app/graphql/types/query_type.rb b/app/graphql/types/query_type.rb
@@ -154,7 +154,7 @@ def people(query: nil, first: nil)
   def creative_works(query: nil, ids: nil, client_id: nil, provider_id: nil, first: nil)
     if ids.present?
       dois = ids.split(",").map { |i| doi_from_url(i) }
-      Doi.find_by_id(dois, page: { number: 1, size: first }).results.to_a
+      Doi.find_by_ids(dois, page: { number: 1, size: first }).results.to_a
     else
       Doi.query(query, client_id: client_id, provider_id: provider_id, state: "findable", page: { number: 1, size: first }).results.to_a
     end

diff --git a/app/models/concerns/batch_loader_helper.rb b/app/models/concerns/batch_loader_helper.rb
@@ -5,7 +5,7 @@ module BatchLoaderHelper
     def load_doi(object)
       BatchLoader.for(object.uuid).batch do |dois, loader|
         dois = object.doi
-        results = Doi.find_by_id(dois).results
+        results = Doi.find_by_ids(dois).results
         loader.call(object.uuid, results) 
       end
     end

diff --git a/app/models/doi.rb b/app/models/doi.rb
@@ -598,7 +598,7 @@ def self.query_fields
   end
 
   # return results for one or more ids
-  def self.find_by_id(ids, options={})
+  def self.find_by_ids(ids, options={})
     ids = ids.split(",") if ids.is_a?(String)
 
     options[:page] ||= {}
@@ -624,6 +624,17 @@ def self.find_by_id(ids, options={})
     })
   end
 
+  # return results for one doi
+  def self.find_by_id(id)
+    __elasticsearch__.search(
+      query: {
+        match: {
+          uid: id,
+        },
+      },
+    )
+  end
+
   def self.query(query, options={})
     # support scroll api
     # map function is small performance hit