add expand method

TouK · Sep 26, 2024 · 8470bc1 · 8470bc1
1 parent a2e3279
commit 8470bc1
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(newField: (String, Aggregator)): MapAggregator = {
+      val expandedMap: Map[String, Aggregator] = (scalaFields + newField)
+
+      new MapAggregator(expandedMap.asJava)
+    }
+
     private def computeTypeByFields(
         input: TypingResult,
         objType: TypedClass,