Refactor solution

OpenGeoMetadata · Aug 6, 2024 · bd6a122 · bd6a122
1 parent 4f7133b
commit bd6a122
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/lib/geo_combine/indexer.rb b/lib/geo_combine/indexer.rb
@@ -35,16 +35,10 @@ def index(docs)
       start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
 
       # Index in batches; set batch size via BATCH_SIZE
-      batch = []
-      docs.each do |doc, path|
-        if batch.size < @batch_size
-          batch << [doc, path]
-        else
-          total_indexed += index_batch(batch)
-          batch = [[doc, path]]
-        end
+      docs.each_slice(@batch_size) do |slice|
+        batch = slice.map { |doc, path| [doc, path] }
+        total_indexed += index_batch(batch)
       end
-      total_indexed += index_batch(batch) unless batch.empty?
 
       # Issue a commit to make sure all documents are indexed
       @solr.commit