Faster import for multiple document types

Andrew Kane
1 parent c168c3fa
Showing 1 changed file with 8 additions and 16 deletions Show diff stats
lib/searchkick/index.rb
@@ -45,12 +45,7 @@ module Searchkick
     # record based
     def store(record)
-      client.index(
-        index: name,
-        type: document_type(record),
-        id: search_id(record),
-        body: search_data(record)
-      )
+      import([record])
     end
     def remove(record)
@@ -65,16 +60,13 @@ module Searchkick
     end
     def import(records)
-      records.group_by { |r| document_type(r) }.each do |type, batch|
-        response =
-          client.bulk(
-            index: name,
-            type: type,
-            body: batch.map { |r| {index: {_id: search_id(r), data: search_data(r)}} }
-          )
-        if response["errors"]
-          raise Searchkick::ImportError, response["items"].first["index"]["error"]
-        end
+      response =
+        client.bulk(
+          index: name,
+          body: records.map { |r| {index: {_id: search_id(r), _type: document_type(r), data: search_data(r)}} }
+        )
+      if response["errors"]
+        raise Searchkick::ImportError, response["items"].first["index"]["error"]
       end
     end