Support cardinality metric aggregator (#876)

JP Rosevear · Andrew Kane
1 parent 64615cb6
Showing 2 changed files with 20 additions and 0 deletions Show diff stats
lib/searchkick/query.rb
test/aggs_test.rb
@@ -643,6 +643,12 @@ module Searchkick
               interval: interval
             }
           }
+        elsif agg_options[:cardinality]
+          payload[:aggs][field] = {
+            cardinality: {
+              field: agg_options[:cardinality][:field] || field
+            }
+          }
         else
           payload[:aggs][field] = {
             terms: {
@@ -116,6 +116,20 @@ class AggsTest &lt; Minitest::Test
     assert_equal 4, products.aggs["products_per_year"]["buckets"].size
   end
+  def test_aggs_cardinality
+    products =
+      Product.search("*", {
+        aggs: {
+          total_stores: {
+            cardinality: {
+              field: :store_id
+            }
+          }
+        }
+      })
+    assert_equal 3, products.aggs["total_stores"]["value"]
+  end
+
   protected
   def buckets_as_hash(agg)
	@@ -643,6 +643,12 @@ module Searchkick		@@ -643,6 +643,12 @@ module Searchkick
643	interval: interval	643	interval: interval
644	}	644	}
645	}	645	}
		646	+ elsif agg_options[:cardinality]
		647	+ payload[:aggs][field] = {
		648	+ cardinality: {
		649	+ field: agg_options[:cardinality][:field] \|\| field
		650	+ }
		651	+ }
646	else	652	else
647	payload[:aggs][field] = {	653	payload[:aggs][field] = {
648	terms: {	654	terms: {
	@@ -116,6 +116,20 @@ class AggsTest < Minitest::Test		@@ -116,6 +116,20 @@ class AggsTest < Minitest::Test
116	assert_equal 4, products.aggs["products_per_year"]["buckets"].size	116	assert_equal 4, products.aggs["products_per_year"]["buckets"].size
117	end	117	end
118		118
		119	+ def test_aggs_cardinality
		120	+ products =
		121	+ Product.search("*", {
		122	+ aggs: {
		123	+ total_stores: {
		124	+ cardinality: {
		125	+ field: :store_id
		126	+ }
		127	+ }
		128	+ }
		129	+ })
		130	+ assert_equal 3, products.aggs["total_stores"]["value"]
		131	+ end
		132	+
119	protected	133	protected
120		134
121	def buckets_as_hash(agg)	135	def buckets_as_hash(agg)