Merge pull request ClickHouse#88843 from ClickHouse/backport/25.8/87789

Backport ClickHouse#87789 to 25.8: Fix Insert Select with CTE

Author: clickhouse-gh[bot]

src/Interpreters/InterpreterInsertQuery.cpp

@@ -9,6 +9,8 @@
 #include <Core/ServerSettings.h>
 #include <DataTypes/DataTypeNullable.h>
 #include <IO/ReadBuffer.h>
+#include <Interpreters/ApplyWithAliasVisitor.h>
+#include <Interpreters/ApplyWithSubqueryVisitor.h>
 #include <Interpreters/DatabaseCatalog.h>
 #include <Interpreters/InterpreterSelectWithUnionQuery.h>
 #include <Interpreters/InterpreterWatchQuery.h>
@@ -80,6 +82,7 @@ namespace Setting
     extern const SettingsBool async_query_sending_for_remote;
     extern const SettingsBool async_socket_for_remote;
     extern const SettingsUInt64 max_distributed_depth;
+    extern const SettingsBool enable_global_with_statement;
 }
 
 namespace MergeTreeSetting
@@ -760,6 +763,10 @@ InterpreterInsertQuery::distributedWriteIntoReplicatedMergeTreeFromClusterStorag
     {
         if (auto * select_query = select.list_of_selects->children.at(0)->as<ASTSelectQuery>())
         {
+            if (local_context->getSettingsRef()[Setting::enable_global_with_statement])
+                ApplyWithAliasVisitor::visit(select.list_of_selects->children.at(0));
+            ApplyWithSubqueryVisitor(local_context).visit(select.list_of_selects->children.at(0));
+
             JoinedTables joined_tables(Context::createCopy(local_context), *select_query);
             if (joined_tables.tablesCount() == 1)
                 src_storage = joined_tables.getLeftTableStorage();

tests/queries/0_stateless/03632_insert_select_cte_bug.reference

@@ -0,0 +1 @@
+1

tests/queries/0_stateless/03632_insert_select_cte_bug.sql

@@ -0,0 +1,17 @@
+SET enable_analyzer=1; -- parallel distributed insert select for replicated tables works only with analyzer
+SET parallel_distributed_insert_select=2;
+SET enable_global_with_statement=1;
+
+DROP TABLE IF EXISTS test_insert SYNC;
+
+CREATE TABLE test_insert (c1 String, c2 UInt8)
+ENGINE = ReplicatedMergeTree('/clickhouse/tables/{database}/test_03632/tables/test_insert', '{replica}')
+ORDER BY ();
+
+INSERT INTO test_insert
+WITH cte_test AS (SELECT '1234', 1)
+SELECT * FROM cte_test;
+
+SELECT count() FROM test_insert;
+
+DROP TABLE test_insert;