dd expand method

TouK · Oct 2, 2024 · 0420297 · 0420297
1 parent 0b0373c
commit 0420297
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...c/main/scala/pl/touk/nussknacker/engine/flink/util/transformer/aggregate/aggregates.scala b/...c/main/scala/pl/touk/nussknacker/engine/flink/util/transformer/aggregate/aggregates.scala
@@ -346,6 +346,12 @@ object aggregates {
       _.computeStoredType(_)
     )
 
+    def expand(key: String, aggregator: Aggregator): MapAggregator = {
+      val expandedMap: Map[String, Aggregator] = (scalaFields + (key -> aggregator))
+
+      new MapAggregator(expandedMap.asJava)
+    }
+
     private def computeTypeByFields(
         input: TypingResult,
         objType: TypedClass,