Add time zone to date histogram (#1136)

* Add time_zone as attribute to searchkick date_histogram query * Test for aggregation with time zone * Fix test, allow any args to date_histogram. - Fix test so it doesn't fail based on when it runs. - Allow any key/value pairs to be passed through to date_histogram.

Add time zone to date histogram (#1136)
* Add time_zone as attribute to searchkick date_histogram query * Test for aggregation with time zone * Fix test, allow any args to date_histogram. - Fix test so it doesn't fail based on when it runs. - Allow any key/value pairs to be passed through to date_histogram.
Josh Schwartzman · Andrew Kane
1 parent edfe444d
Showing 2 changed files with 44 additions and 5 deletions Show diff stats
lib/searchkick/query.rb
test/aggs_test.rb
@@ -742,12 +742,8 @@ module Searchkick
             }.merge(shared_agg_options)
           }
         elsif (histogram = agg_options[:date_histogram])
-          interval = histogram[:interval]
           payload[:aggs][field] = {
-            date_histogram: {
-              field: histogram[:field],
-              interval: interval
-            }
+            date_histogram: histogram
           }
         elsif (metric = @@metric_aggs.find { |k| agg_options.has_key?(k) })
           payload[:aggs][field] = {
@@ -116,6 +116,32 @@ class AggsTest &lt; Minitest::Test
     assert_equal 4, products.aggs["products_per_year"]["buckets"].size
   end
+  def test_aggs_with_time_zone
+    start_time = Time.at(1529366400)
+
+    store [
+      {name: "Opera House Pass", created_at: start_time},
+      {name: "London Eye Pass", created_at: start_time + 16.hours},
+      {name: "London Tube Pass", created_at: start_time + 16.hours}
+    ]
+
+    sydney_search = search_aggregate_by_day_with_time_zone('Pass', '+10:00') # Sydney
+    london_search = search_aggregate_by_day_with_time_zone('Pass', '+01:00') # London
+
+    # London search will return all 3 in one bucket because of time zone offset
+    expected_london_buckets = [
+      {"key_as_string" => "2018-06-19T00:00:00.000+01:00", "key" => 1529362800000, "doc_count" => 3}
+    ]
+    assert_equal expected_london_buckets, london_search.aggs["products_per_day"]["buckets"]
+
+    # Sydney search will return them in separate buckets due to time zone offset
+    expected_sydney_buckets = [
+      {"key_as_string" => "2018-06-19T00:00:00.000+10:00", "key" => 1529330400000, "doc_count" => 1},
+      {"key_as_string" => "2018-06-20T00:00:00.000+10:00", "key" => 1529416800000, "doc_count" => 2}
+    ]
+    assert_equal expected_sydney_buckets, sydney_search.aggs["products_per_day"]["buckets"]
+  end
+
   def test_aggs_avg
     products =
       Product.search("*", {
@@ -200,6 +226,23 @@ class AggsTest &lt; Minitest::Test
   protected
+  def search_aggregate_by_day_with_time_zone(query, time_zone = '-8:00')
+    Product.search(query, {
+      where: {
+        created_at: {lt: Time.now}
+      },
+      aggs: {
+        products_per_day: {
+          date_histogram: {
+            field: :created_at,
+            interval: :day,
+            time_zone: time_zone
+          }
+        }
+      }
+    })
+  end
+
   def buckets_as_hash(agg)
     Hash[agg["buckets"].map { |v| [v["key"], v["doc_count"]] }]
   end
	@@ -116,6 +116,32 @@ class AggsTest < Minitest::Test		@@ -116,6 +116,32 @@ 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_with_time_zone
		120	+ start_time = Time.at(1529366400)
		121	+
		122	+ store [
		123	+ {name: "Opera House Pass", created_at: start_time},
		124	+ {name: "London Eye Pass", created_at: start_time + 16.hours},
		125	+ {name: "London Tube Pass", created_at: start_time + 16.hours}
		126	+ ]
		127	+
		128	+ sydney_search = search_aggregate_by_day_with_time_zone('Pass', '+10:00') # Sydney
		129	+ london_search = search_aggregate_by_day_with_time_zone('Pass', '+01:00') # London
		130	+
		131	+ # London search will return all 3 in one bucket because of time zone offset
		132	+ expected_london_buckets = [
		133	+ {"key_as_string" => "2018-06-19T00:00:00.000+01:00", "key" => 1529362800000, "doc_count" => 3}
		134	+ ]
		135	+ assert_equal expected_london_buckets, london_search.aggs["products_per_day"]["buckets"]
		136	+
		137	+ # Sydney search will return them in separate buckets due to time zone offset
		138	+ expected_sydney_buckets = [
		139	+ {"key_as_string" => "2018-06-19T00:00:00.000+10:00", "key" => 1529330400000, "doc_count" => 1},
		140	+ {"key_as_string" => "2018-06-20T00:00:00.000+10:00", "key" => 1529416800000, "doc_count" => 2}
		141	+ ]
		142	+ assert_equal expected_sydney_buckets, sydney_search.aggs["products_per_day"]["buckets"]
		143	+ end
		144	+
119	def test_aggs_avg	145	def test_aggs_avg
120	products =	146	products =
121	Product.search("*", {	147	Product.search("*", {
	@@ -200,6 +226,23 @@ class AggsTest < Minitest::Test		@@ -200,6 +226,23 @@ class AggsTest < Minitest::Test
200		226
201	protected	227	protected
202		228
		229	+ def search_aggregate_by_day_with_time_zone(query, time_zone = '-8:00')
		230	+ Product.search(query, {
		231	+ where: {
		232	+ created_at: {lt: Time.now}
		233	+ },
		234	+ aggs: {
		235	+ products_per_day: {
		236	+ date_histogram: {
		237	+ field: :created_at,
		238	+ interval: :day,
		239	+ time_zone: time_zone
		240	+ }
		241	+ }
		242	+ }
		243	+ })
		244	+ end
		245	+
203	def buckets_as_hash(agg)	246	def buckets_as_hash(agg)
204	Hash[agg["buckets"].map { \|v\| [v["key"], v["doc_count"]] }]	247	Hash[agg["buckets"].map { \|v\| [v["key"], v["doc_count"]] }]
205	end	248	end