chore(binder): fix bind cte (databendlabs#17183)

* chore(binder): fix bind cte

* chore(test): add sqllogictest

* chore(binder): refine code

* chore(binder): remove materialized cte in create table stmt with

* chore(binder): fix m_cte_to_temp_table

* chore(code): refine code

* chore(code): refine code

Author: Dousir9

src/query/sql/src/planner/binder/bind_query/bind.rs

@@ -101,7 +101,7 @@ impl Binder {
             };
             // If the CTE is materialized, we'll construct a temp table for it.
             if cte.materialized {
-                self.m_cte_to_temp_table(cte)?;
+                self.m_cte_to_temp_table(cte, idx, with.clone())?;
             }
             bind_context
                 .cte_context
@@ -180,7 +180,7 @@ impl Binder {
         ))
     }
 
-    fn m_cte_to_temp_table(&mut self, cte: &CTE) -> Result<()> {
+    fn m_cte_to_temp_table(&mut self, cte: &CTE, cte_index: usize, mut with: With) -> Result<()> {
         let engine = if self.ctx.get_settings().get_persist_materialized_cte()? {
             Engine::Fuse
         } else {
@@ -207,6 +207,13 @@ impl Binder {
 
         let expr_replacer = ExprReplacer::new(database.clone(), self.m_cte_table_name.clone());
         let mut as_query = cte.query.clone();
+        with.ctes.truncate(cte_index);
+        with.ctes.retain(|cte| !cte.materialized);
+        as_query.with = if !with.ctes.is_empty() {
+            Some(with)
+        } else {
+            None
+        };
         expr_replacer.replace_query(&mut as_query);
 
         let create_table_stmt = CreateTableStmt {

tests/sqllogictests/suites/query/cte/cte.test

@@ -557,4 +557,18 @@ statement ok
 drop table if exists t1;
 
 statement ok
-drop table if exists t2;
+drop table if exists t2;
+
+query T
+with t(tt) as (select number as tt from numbers(10)),  t2(tt) AS MATERIALIZED (SELECT tt FROM t) select * from t2;
+----
+0
+1
+2
+3
+4
+5
+6
+7
+8
+9