add 'size' option to Agg

rcarver · Jan 30, 2025 · b6b4264 · b6b4264
1 parent d8f8b3c
commit b6b4264
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/Sources/ElasticsearchQueryBuilder/Components.swift b/Sources/ElasticsearchQueryBuilder/Components.swift
@@ -146,9 +146,13 @@ extension esb {
     public struct Agg: DictComponent {
         var name: String
         var term: QueryDict
-        public init(_ name: String, field: String) {
+        public init(_ name: String, field: String, size: Int? = nil) {
             self.name = name
-            self.term = [ "field" : .string(field) ]
+            if let size {
+                self.term = [ "field" : .string(field), "size": .int(size) ]
+            } else {
+                self.term = [ "field" : .string(field) ]
+            }
         }
         public init(_ name: String, term: QueryDict) {
             self.name = name

diff --git a/Tests/ElasticsearchQueryBuilderTests/ComponentTests.swift b/Tests/ElasticsearchQueryBuilderTests/ComponentTests.swift
@@ -186,13 +186,17 @@ final class AggsTests: XCTestCase {
     func testBuild() throws {
         @ElasticsearchQueryBuilder func build() -> some esb.QueryDSL {
             esb.Aggs {
-                esb.Agg("name", field: "name")
+                esb.Agg("names", field: "name")
+                esb.Agg("ages", field: "age", size: 100)
             }
         }
         expectNoDifference(build().makeQuery(), [
             "aggs": [
-                "name": [
+                "names": [
                     "terms": [ "field": "name" ]
+                ],
+                "ages": [
+                    "terms": [ "field": "age", "size": 100 ]
                 ]
             ]
         ])