Added multi search

Andrew Kane
1 parent 2f16c8da
Showing 3 changed files with 24 additions and 1 deletions Show diff stats
lib/searchkick.rb
lib/searchkick/query.rb
test/multi_search_test.rb
@@ -138,6 +138,16 @@ module Searchkick
       query.execute
     end
   end
+
+  def self.multi_search(queries)
+    if queries.any?
+      responses = client.msearch(body: queries.flat_map { |q| [q.params.except(:body), q.body] })["responses"]
+      queries.each_with_index do |query, i|
+        query.send(:handle_response, responses[i])
+      end
+    end
+    true
+  end
 end
  
 # TODO find better ActiveModel hook
@@ -129,7 +129,7 @@ module Searchkick
         match_suffix: @match_suffix,
         highlighted_fields: @highlighted_fields || []
       }
-      Searchkick::Results.new(searchkick_klass, response, opts)
+      @execute = Searchkick::Results.new(searchkick_klass, response, opts)
     end
  
     def reindex_command
@@ -0,0 +1,13 @@
+require_relative "test_helper"
+
+class MultiSearchTest < Minitest::Test
+  def test_basic
+    store_names ["Product A"]
+    store_names ["Store A"], Store
+    products = Product.search("*", execute: false)
+    stores = Store.search("*", execute: false)
+    Searchkick.multi_search([products, stores])
+    assert_equal ["Product A"], products.map(&:name)
+    assert_equal ["Store A"], stores.map(&:name)
+  end
+end
...	...	@@ -138,6 +138,16 @@ module Searchkick
138	138	query.execute
139	139	end
140	140	end
	141	+
	142	+ def self.multi_search(queries)
	143	+ if queries.any?
	144	+ responses = client.msearch(body: queries.flat_map { \|q\| [q.params.except(:body), q.body] })["responses"]
	145	+ queries.each_with_index do \|query, i\|
	146	+ query.send(:handle_response, responses[i])
	147	+ end
	148	+ end
	149	+ true
	150	+ end
141	151	end
142	152
143	153	# TODO find better ActiveModel hook
...	...
...	...	@@ -129,7 +129,7 @@ module Searchkick
129	129	match_suffix: @match_suffix,
130	130	highlighted_fields: @highlighted_fields \|\| []
131	131	}
132		- Searchkick::Results.new(searchkick_klass, response, opts)
	132	+ @execute = Searchkick::Results.new(searchkick_klass, response, opts)
133	133	end
134	134
135	135	def reindex_command
...	...
...	...	@@ -0,0 +1,13 @@
	1	+require_relative "test_helper"
	2	+
	3	+class MultiSearchTest < Minitest::Test
	4	+ def test_basic
	5	+ store_names ["Product A"]
	6	+ store_names ["Store A"], Store
	7	+ products = Product.search("*", execute: false)
	8	+ stores = Store.search("*", execute: false)
	9	+ Searchkick.multi_search([products, stores])
	10	+ assert_equal ["Product A"], products.map(&:name)
	11	+ assert_equal ["Store A"], stores.map(&:name)
	12	+ end
	13	+end
...	...