use active record for elasticsearch results. #274

datacite · May 16, 2019 · 8ed26e6 · 8ed26e6
1 parent 7c2c733
commit 8ed26e6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/graphql/elasticsearch_loader.rb b/app/graphql/elasticsearch_loader.rb
@@ -6,7 +6,11 @@ def initialize(model)
   end
 
   def perform(ids)
-    @model.find_by_id(ids).results.each { |record| fulfill(record.uid, record) }
+    if @model.name == "Doi"
+      @model.find_by_id(ids).results.each { |record| fulfill(record.uid, record) }
+    else
+      @model.find_by_id(ids).records.each { |record| fulfill(record.uid, record) }
+    end
     ids.each { |id| fulfill(id, nil) unless fulfilled?(id) }
   end
 end