Merge branch 'samuel02-stats-facets'

Andrew Kane
2 parents e9b70245 98c9b5ab
Showing 3 changed files with 23 additions and 1 deletions Show diff stats
README.md
lib/searchkick/query.rb
test/facets_test.rb
@@ -443,6 +443,12 @@ price_ranges = [{to: 20}, {from: 20, to: 50}, {from: 50}]
 Product.search "*", facets: {price: {ranges: price_ranges}}
 ```
+Use the `stats` option to get to max, min, mean, and total scores for each facet
+
+```ruby
+Product.search "*", facets: {store_id: {stats: true}}
+```
+
 ### Highlight
 Highlight the search query in the results.
@@ -219,6 +219,7 @@ module Searchkick
         facets.each do |field, facet_options|
           # ask for extra facets due to
           # https://github.com/elasticsearch/elasticsearch/issues/1305
+          size = facet_options[:limit] ? facet_options[:limit] + 150 : 100000
           if facet_options[:ranges]
             payload[:facets][field] = {
@@ -226,11 +227,19 @@ module Searchkick
                 field.to_sym => facet_options[:ranges]
               }
             }
+          elsif facet_options[:stats]
+            payload[:facets][field] = {
+              terms_stats: {
+                key_field: field,
+                value_script: "doc.score",
+                size: size
+              }
+            }
           else
             payload[:facets][field] = {
               terms: {
                 field: field,
-                size: facet_options[:limit] ? facet_options[:limit] + 150 : 100000
+                size: size
               }
             }
           end
@@ -59,6 +59,13 @@ class TestFacets &lt; Minitest::Unit::TestCase
     assert_equal ({1 => 1, 2 => 1}), store_facet(where: {store_id: 2, price: {gt: 5}}, facets: [:store_id], smart_facets: true)
   end
+  def test_stats_facets
+    options = {where: {store_id: 2}, facets: {store_id: {stats: true}}}
+    facets = Product.search("Product", options).facets["store_id"]["terms"]
+    expected_facets_keys = %w[term count total_count min max total mean]
+    assert_equal expected_facets_keys, facets.first.keys
+  end
+
   protected
   def store_facet(options)
	@@ -443,6 +443,12 @@ price_ranges = [{to: 20}, {from: 20, to: 50}, {from: 50}]		@@ -443,6 +443,12 @@ price_ranges = [{to: 20}, {from: 20, to: 50}, {from: 50}]
443	Product.search "*", facets: {price: {ranges: price_ranges}}	443	Product.search "*", facets: {price: {ranges: price_ranges}}
444	```	444	```
445		445
		446	+Use the `stats` option to get to max, min, mean, and total scores for each facet
		447	+
		448	+```ruby
		449	+Product.search "*", facets: {store_id: {stats: true}}
		450	+```
		451	+
446	### Highlight	452	### Highlight
447		453
448	Highlight the search query in the results.	454	Highlight the search query in the results.
	@@ -59,6 +59,13 @@ class TestFacets < Minitest::Unit::TestCase		@@ -59,6 +59,13 @@ class TestFacets < Minitest::Unit::TestCase
59	assert_equal ({1 => 1, 2 => 1}), store_facet(where: {store_id: 2, price: {gt: 5}}, facets: [:store_id], smart_facets: true)	59	assert_equal ({1 => 1, 2 => 1}), store_facet(where: {store_id: 2, price: {gt: 5}}, facets: [:store_id], smart_facets: true)
60	end	60	end
61		61
		62	+ def test_stats_facets
		63	+ options = {where: {store_id: 2}, facets: {store_id: {stats: true}}}
		64	+ facets = Product.search("Product", options).facets["store_id"]["terms"]
		65	+ expected_facets_keys = %w[term count total_count min max total mean]
		66	+ assert_equal expected_facets_keys, facets.first.keys
		67	+ end
		68	+
62	protected	69	protected
63		70
64	def store_facet(options)	71	def store_facet(options)