Merge pull request #19 from matthuhiggins/multi-percolate

Percolate query supports multiple documents
matthuhiggins · Aug 2, 2018 · e350ca0 · e350ca0
2 parents 0ac173a + 94389e2
commit e350ca0
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/lib/arelastic/queries/percolate.rb b/lib/arelastic/queries/percolate.rb
@@ -11,11 +11,15 @@ def initialize(field, document, options = {})
       def as_elastic
         {
           "percolate" => {
-            "field"     => field,
-            "document"  => document
+            "field"                  => field,
+            document_parameter_name  => document
           }.merge(options)
         }
       end
+
+      def document_parameter_name
+        document.is_a?(Array) ? "documents" : "document"
+      end
     end
   end
 end
diff --git a/test/arelastic/queries/percolate_test.rb b/test/arelastic/queries/percolate_test.rb
@@ -12,6 +12,20 @@ def test_as_elastic
     assert_equal expected, percolate.as_elastic
   end
 
+  def test_as_elastic_with_multiple_documents
+    percolate = Arelastic::Queries::Percolate.new(
+      "query",
+      [{ "color" => "red" }, { "color" => "blue" }]
+    )
+    expected = {
+      "percolate" => {
+        "field" => "query",
+        "documents" => [{ "color" => "red" }, { "color" => "blue" }]
+      }
+    }
+    assert_equal expected, percolate.as_elastic
+  end
+
   def test_as_elastic_with_options
     percolate = Arelastic::Queries::Percolate.new("query", {"color" => "red"}, {"document_type" => "widget"})
     expected = {