diff --git a/core/src/main/java/org/apache/calcite/sql/validate/SqlValidatorImpl.java b/core/src/main/java/org/apache/calcite/sql/validate/SqlValidatorImpl.java index 706c8e6be0e5..461a1ecec588 100644 --- a/core/src/main/java/org/apache/calcite/sql/validate/SqlValidatorImpl.java +++ b/core/src/main/java/org/apache/calcite/sql/validate/SqlValidatorImpl.java @@ -2941,6 +2941,12 @@ private void registerQuery( SqlSelect.HAVING_OPERAND); registerSubQueries(selectScope2, SqlNonNullableAccessors.getSelectList(select)); + + if (enclosingNode.getKind() == SqlKind.UPDATE) { + registerSubQueries(selectScope2, + ((SqlUpdate) enclosingNode).getSourceExpressionList()); + } + final SqlNodeList orderList = select.getOrderList(); if (orderList != null) { // If the query is 'SELECT DISTINCT', restrict the columns diff --git a/core/src/test/java/org/apache/calcite/test/SqlToRelConverterTest.java b/core/src/test/java/org/apache/calcite/test/SqlToRelConverterTest.java index 47efce21c62d..bb1e2a1ae95c 100644 --- a/core/src/test/java/org/apache/calcite/test/SqlToRelConverterTest.java +++ b/core/src/test/java/org/apache/calcite/test/SqlToRelConverterTest.java @@ -3119,6 +3119,13 @@ void checkCorrelatedMapSubQuery(boolean expand) { sql(sql).ok(); } + @Test void testUpdateSubQueryWithCast() { + final String sql = "update emp\n" + + "set empno = (\n" + + " select cast(min(empno) as BIGINT) from emp as e where e.deptno = emp.deptno)"; + sql(sql).ok(); + } + /** * Test case for * [CALCITE-3229] diff --git a/core/src/test/resources/org/apache/calcite/test/SqlToRelConverterTest.xml b/core/src/test/resources/org/apache/calcite/test/SqlToRelConverterTest.xml index 5d6f6f2f0ff6..1939c2593a2b 100644 --- a/core/src/test/resources/org/apache/calcite/test/SqlToRelConverterTest.xml +++ b/core/src/test/resources/org/apache/calcite/test/SqlToRelConverterTest.xml @@ -8933,6 +8933,24 @@ LogicalTableModify(table=[[CATALOG, SALES, EMP]], operation=[UPDATE], updateColu LogicalAggregate(group=[{0}], EXPR$0=[MIN($1)]) LogicalProject(DEPTNO=[$7], EMPNO=[$0]) LogicalTableScan(table=[[CATALOG, SALES, EMP]]) +]]> + + + + + + + +