Allow the BibReader to return the size of the entire bibliography.

sul-dlss · Oct 3, 2017 · 2c6ffb5 · 2c6ffb5
1 parent b6da25d
commit 2c6ffb5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/app/services/bibliography_builder.rb b/app/services/bibliography_builder.rb
@@ -7,8 +7,10 @@ class BibliographyBuilder < Spotlight::SolrDocumentBuilder
   include ActiveSupport::Benchmarkable
   delegate :logger, to: :Rails
 
+  delegate :size, to: :traject_reader
+
   def to_solr
-    return to_enum(:to_solr) unless block_given?
+    return to_enum(:to_solr) { size } unless block_given?
 
     benchmark "Indexing resource #{inspect}" do
       base_doc = super
@@ -37,8 +39,7 @@ def traject_indexer
   private
 
   def convert_id(doc)
-    return unless doc['id']
-    doc[:id] = doc['id'].first
+    doc[:id] = doc['id'].try(:first)
     doc
   end
 end
diff --git a/lib/traject/bib_reader.rb b/lib/traject/bib_reader.rb
@@ -4,6 +4,9 @@
 class BibReader
   # @param input_stream [File]
   # @param settings [Traject::Indexer::Settings]
+
+  delegate :size, to: :bibtex
+
   def initialize(input_stream, settings)
     @settings = Traject::Indexer::Settings.new settings
     @input_stream = input_stream