Github user viirya commented on a diff in the pull request:
https://github.com/apache/spark/pull/21405#discussion_r191372561
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/codegen/javaCode.scala
---
@@ -149,6 +152,34 @@ trait Block extends JavaCode {
// Concatenates this block with other block.
def + (other: Block): Block
+
+ /**
+ * Apply a map function to each java expression codes present in this
java code, and return a new
+ * java code based on the mapped java expression codes.
+ */
+ def transformExprValues(f: PartialFunction[ExprValue, ExprValue]):
this.type = {
--- End diff --
Added test for it.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
