Merge pull request #142 from mixify/sum-aggregation

Adds Sum Aggregation support
jillesvangurp · May 21, 2024 · 6345bef · 6345bef
2 parents 68e29a0 + f3d5e72
commit 6345bef
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 0 deletions.
diff --git a/search-client/src/commonMain/kotlin/com/jillesvangurp/ktsearch/SearchResponse.kt b/search-client/src/commonMain/kotlin/com/jillesvangurp/ktsearch/SearchResponse.kt
@@ -413,3 +413,14 @@ fun Aggregations?.topHitResult(name: String, json: Json = DEFAULT_JSON): TopHits
 fun Aggregations?.topHitResult(name: Enum<*>, json: Json = DEFAULT_JSON): TopHitsAggregationResult =
     getAggResult(name.name, json)
 
+@Serializable
+data class SumAggregationResult(
+    val value: Double,
+)
+
+fun Aggregations?.sumAggregationResult(name: String, json: Json = DEFAULT_JSON): SumAggregationResult =
+    getAggResult(name, json)
+
+fun Aggregations?.sumAggregationResult(name: Enum<*>, json: Json = DEFAULT_JSON): TopHitsAggregationResult =
+    getAggResult(name.name, json)
+
diff --git a/search-client/src/commonTest/kotlin/com/jillesvangurp/ktsearch/AggQueryTest.kt b/search-client/src/commonTest/kotlin/com/jillesvangurp/ktsearch/AggQueryTest.kt
@@ -259,4 +259,14 @@ class AggQueryTest : SearchTestBase() {
             b.topHitResult("top").hits.hits.size shouldBeGreaterThan 0
         }
     }
+
+    @Test
+    fun sumAgg() = coRun {
+        before()
+        val response = repository.search {
+            resultSize = 0
+            agg("by_value", SumAgg(MockDoc::value)){}
+        }
+        response.aggregations.sumAggregationResult("by_value").value shouldBe 10.0
+    }
 }
diff --git a/...h-dsls/src/commonMain/kotlin/com/jillesvangurp/searchdsls/querydsl/aggregation-queries.kt b/...h-dsls/src/commonMain/kotlin/com/jillesvangurp/searchdsls/querydsl/aggregation-queries.kt
@@ -327,5 +327,19 @@ class FilterAgg(filter: ESQuery): AggQuery("filter") {
     }
 }
 
+class SumAggConfig : JsonDsl() {
+    var field by property<String>()
+}
+
+class SumAgg(val field: String, block: (SumAggConfig.() -> Unit)? = null) : AggQuery("sum") {
+    constructor(field: KProperty<*>, block: (SumAggConfig.() -> Unit)? = null) : this(field.name, block)
+
+    init {
+        val config = SumAggConfig()
+        config.field = field
+        block?.invoke(config)
+        put(name, config)
+    }
+}