Update Search#images to return an enumerable

projectblacklight · Dec 10, 2015 · 9711549 · 9711549
1 parent 412e723
commit 9711549
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/app/models/spotlight/search.rb b/app/models/spotlight/search.rb
@@ -28,8 +28,10 @@ def count
     end
 
     def images
-      documents.map do |doc|
-        [
+      return enum_for(:images) unless block_given?
+
+      documents.each do |doc|
+        yield [
           doc.first(blacklight_config.document_model.unique_key),
           doc.first(blacklight_config.index.title_field),
           doc.first(blacklight_config.index.thumbnail_field)