Optimize MapNormalize

kaikalur · Mar 14, 2024 · fb7780c · fb7780c
1 parent 2aa1acf
commit fb7780c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/presto-main/src/main/java/com/facebook/presto/operator/scalar/sql/MapNormalizeFunction.java b/presto-main/src/main/java/com/facebook/presto/operator/scalar/sql/MapNormalizeFunction.java
@@ -28,6 +28,8 @@ private MapNormalizeFunction() {}
     @SqlType("map<varchar, double>")
     public static String arraySumDouble()
     {
-        return "RETURN transform_values(input, (k, v) -> (v / reduce(map_values(input), double '0.0', (s, x) -> (s + coalesce(x, double '0.0')), (s) -> s)))";
+        return "RETURN " +
+                "transform(array[ROW(input, reduce(map_values(input), double '0.0', (s, x) -> (s + coalesce(x, double '0.0')), (s) -> s))], " +
+                "x->transform_values(x[1], (k, v) -> (v / x[2])))[1]";
     }
 }