Update elastic query DSL syntax

nyaruka · May 22, 2024 · 44581e5 · 44581e5
1 parent d00a4d4
commit 44581e5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 26 deletions.
diff --git a/elastic/query.go b/elastic/query.go
@@ -9,7 +9,7 @@ func Ids(values ...string) Query {
 
 // Term is a shortcut for a term query
 func Term(field string, value any) Query {
-	return Query{"term": map[string]any{field: value}}
+	return Query{"term": map[string]any{field: map[string]any{"value": value}}}
 }
 
 // Exists is a shortcut for an exists query
@@ -32,10 +32,7 @@ func GreaterThan(field string, value any) Query {
 	return Query{
 		"range": map[string]any{
 			field: map[string]any{
-				"from":          value,
-				"include_lower": false,
-				"include_upper": true,
-				"to":            nil,
+				"gt": value,
 			},
 		},
 	}
@@ -46,10 +43,7 @@ func GreaterThanOrEqual(field string, value any) Query {
 	return Query{
 		"range": map[string]any{
 			field: map[string]any{
-				"from":          value,
-				"include_lower": true,
-				"include_upper": true,
-				"to":            nil,
+				"gte": value,
 			},
 		},
 	}
@@ -60,10 +54,7 @@ func LessThan(field string, value any) Query {
 	return Query{
 		"range": map[string]any{
 			field: map[string]any{
-				"from":          nil,
-				"include_lower": true,
-				"include_upper": false,
-				"to":            value,
+				"lt": value,
 			},
 		},
 	}
@@ -74,10 +65,7 @@ func LessThanOrEqual(field string, value any) Query {
 	return Query{
 		"range": map[string]any{
 			field: map[string]any{
-				"from":          nil,
-				"include_lower": true,
-				"include_upper": true,
-				"to":            value,
+				"lte": value,
 			},
 		},
 	}
@@ -88,10 +76,8 @@ func Between(field string, from, to any) Query {
 	return Query{
 		"range": map[string]any{
 			field: map[string]any{
-				"from":          from,
-				"include_lower": true,
-				"include_upper": false,
-				"to":            to,
+				"gte": from,
+				"lt":  to,
 			},
 		},
 	}

diff --git a/elastic/query_test.go b/elastic/query_test.go
@@ -18,11 +18,11 @@ func TestQuery(t *testing.T) {
 		{elastic.Exists("age"), []byte(`{"exists": {"field": "age"}}`)},
 		{elastic.Match("name", "Bob"), []byte(`{"match": {"name": {"query": "Bob"}}}`)},
 		{elastic.MatchPhrase("name", "Bob"), []byte(`{"match_phrase": {"name": {"query": "Bob"}}}`)},
-		{elastic.GreaterThan("age", 45), []byte(`{"range": {"age": {"from": 45, "include_lower": false, "include_upper": true, "to": null}}}`)},
-		{elastic.GreaterThanOrEqual("age", 45), []byte(`{"range": {"age": {"from": 45, "include_lower": true, "include_upper": true, "to": null}}}`)},
-		{elastic.LessThan("age", 45), []byte(`{"range": {"age": {"from": null, "include_lower": true, "include_upper": false, "to": 45}}}`)},
-		{elastic.LessThanOrEqual("age", 45), []byte(`{"range": {"age": {"from": null, "include_lower": true, "include_upper": true, "to": 45}}}`)},
-		{elastic.Between("age", 20, 45), []byte(`{"range": {"age": {"from": 20, "include_lower": true, "include_upper": false, "to": 45}}}`)},
+		{elastic.GreaterThan("age", 45), []byte(`{"range": {"age": {"gt": 45}}}`)},
+		{elastic.GreaterThanOrEqual("age", 45), []byte(`{"range": {"age": {"gte": 45}}}`)},
+		{elastic.LessThan("age", 45), []byte(`{"range": {"age": {"lt": 45}}}`)},
+		{elastic.LessThanOrEqual("age", 45), []byte(`{"range": {"age": {"lte": 45}}}`)},
+		{elastic.Between("age", 20, 45), []byte(`{"range": {"age": {"gte": 20, "lt": 45}}}`)},
 		{
 			elastic.Any(elastic.Ids("235"), elastic.Term("age", 42)),
 			[]byte(`{"bool": {"should": [{"ids": {"values": ["235"]}}, {"term": {"age": 42}}]}}`),