Github user kiszk commented on a diff in the pull request:
https://github.com/apache/spark/pull/19733#discussion_r150485352
--- Diff:
sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/expressions/predicates.scala
---
@@ -236,24 +236,38 @@ case class In(value: Expression, list:
Seq[Expression]) extends Predicate {
override def doGenCode(ctx: CodegenContext, ev: ExprCode): ExprCode = {
val valueGen = value.genCode(ctx)
val listGen = list.map(_.genCode(ctx))
+ ctx.addMutableState("boolean", ev.value, "")
+ ctx.addMutableState("boolean", ev.isNull, "")
+ val valueArg = ctx.freshName("valueArg")
val listCode = listGen.map(x =>
s"""
if (!${ev.value}) {
${x.code}
if (${x.isNull}) {
${ev.isNull} = true;
- } else if (${ctx.genEqual(value.dataType, valueGen.value,
x.value)}) {
+ } else if (${ctx.genEqual(value.dataType, valueArg, x.value)}) {
${ev.isNull} = false;
${ev.value} = true;
}
}
- """).mkString("\n")
+ """)
+ val listCodes = if (ctx.currentVars == null) {
+ val args = if (ctx.INPUT_ROW != null) {
--- End diff --
Thanks, I follow the logid in `CodeGenerator.splitExpressions`.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
