Add setPostFilter method to Elastica\Query

Author: krzaczek

changes.txt

@@ -1,5 +1,8 @@
 CHANGES
 
+2014-07-02
+- Add setPostFilter method to Elastica\Query (http://www.elasticsearch.org/guide/en/elasticsearch/guide/current/_filtering_queries_and_aggregations.html#_post_filter)
+
 2014-06-30
 - Add Reverse Nested aggregation (http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/search-aggregations-bucket-reverse-nested-aggregation.html).
 

lib/Elastica/Query.php

@@ -348,6 +348,10 @@ public function toArray()
             unset($this->_params['facets']);
         }
 
+        if (isset($this->_params['post_filter']) && 0 === count($this->_params['post_filter'])) {
+            unset($this->_params['post_filter']);
+        }
+
         return $this->_params;
     }
 
@@ -399,6 +403,17 @@ public function setSource(array $fields)
     {
         return $this->setParam('_source', $fields);
     }
+
+    /**
+     * Sets post_filter argument for the query. The filter is applied after the query has executed
+     * @param   array $post
+     * @return  \Elastica\Query Current object
+     * @link    http://www.elasticsearch.org/guide/en/elasticsearch/guide/current/_filtering_queries_and_aggregations.html#_post_filter
+     */
+    public function setPostFilter(array $post)
+    {
+        return $this->setParam("post_filter", $post);
+    }
 }
 
 

test/lib/Elastica/Test/QueryTest.php

@@ -9,6 +9,7 @@
 use Elastica\Query\Text;
 use Elastica\Query;
 use Elastica\Facet\Terms;
+use Elastica\Filter\Term as FilterTerm;
 use Elastica\Test\Base as BaseTest;
 
 class QueryTest extends BaseTest
@@ -191,4 +192,21 @@ public function testSetFacets()
 
         $this->assertArrayNotHasKey('facets', $query->toArray());
     }
+
+    public function testSetPostQuery()
+    {
+        $query = new Query();
+
+        $post_filter = new FilterTerm(['color' => 'green']);
+        $query->setPostFilter($post_filter->toArray());
+
+        $data = $query->toArray();
+
+        $this->assertArrayHasKey('post_filter', $data);
+        $this->assertEquals(array('term' => array('color' => 'green')), $data['post_filter']);
+
+        $query->setPostFilter(array());
+
+        $this->assertArrayNotHasKey('post_filter', $query->toArray());
+    }
 }