[CALCITE-7268] SqlToRelConverter throws exception if lambda contains IN

Author: dssysolyatin

core/src/main/java/org/apache/calcite/sql2rel/SqlToRelConverter.java

@@ -2100,6 +2100,9 @@ private void findSubQueries(
     }
     final SqlKind kind = node.getKind();
     switch (kind) {
+    // A lambda has its own scope, which is not part of the blackboard.
+    case LAMBDA:
+      return;
     case EXISTS:
     case UNIQUE:
     case SELECT:
@@ -2282,6 +2285,7 @@ private RexNode convertLambda(Blackboard bb, SqlNode node) {
 
     final Blackboard lambdaBb = createBlackboard(scope, nameToNodeMap, false);
     lambdaBb.setRoot(castNonNull(bb.inputs));
+    replaceSubQueries(lambdaBb, call.getExpression(), RelOptUtil.Logic.TRUE_FALSE_UNKNOWN);
     final RexNode expr = lambdaBb.convertExpression(call.getExpression());
     return rexBuilder.makeLambdaCall(expr, parameters);
   }

core/src/test/java/org/apache/calcite/test/SqlToRelConverterTest.java

@@ -143,6 +143,19 @@ public static void checkActualAndReferenceFiles() {
         .ok();
   }
 
+  /** Test case for
+   * <a href="https://issues.apache.org/jira/browse/CALCITE-7268">[CALCITE-7268]
+   * SqlToRelConverter throws exception if lambda contains IN</a>. */
+  @Test void testLambdaExpressionContainsIn() {
+    final String sql = "select \"EXISTS\"(ARRAY[1,2,3,4], (n) -> n IN (1,3))";
+    fixture()
+        .withFactory(c ->
+            c.withOperatorTable(t -> SqlValidatorTest.operatorTableFor(SqlLibrary.SPARK)))
+        .withSql(sql)
+        .ok();
+
+  }
+
   /** Test case for
    * <a href="https://issues.apache.org/jira/browse/CALCITE-3679">[CALCITE-3679]
    * Allow lambda expressions in SQL queries</a>. */

core/src/test/resources/org/apache/calcite/test/SqlToRelConverterTest.xml

@@ -4694,6 +4694,17 @@ LogicalProject(EXPR$0=[HIGHER_ORDER_FUNCTION2(1, () -> -1)])
       <![CDATA[
 LogicalProject(EXPR$0=[HIGHER_ORDER_FUNCTION($7, (X, DEPTNO) -> +(DEPTNO, 1))])
   LogicalTableScan(table=[[CATALOG, SALES, EMP]])
+]]>
+    </Resource>
+  </TestCase>
+  <TestCase name="testLambdaExpressionContainsIn">
+    <Resource name="sql">
+      <![CDATA[select "EXISTS"(ARRAY[1,2,3,4], (n) -> n IN (1,3))]]>
+    </Resource>
+    <Resource name="plan">
+      <![CDATA[
+LogicalProject(EXPR$0=[EXISTS(ARRAY(1, 2, 3, 4), (N) -> OR(=(N, 1), =(N, 3)))])
+  LogicalValues(tuples=[[{ 0 }]])
 ]]>
     </Resource>
   </TestCase>