From 5cd3eeb8f820cf02daacbe8ee3b50e574a343604 Mon Sep 17 00:00:00 2001
From: DreaMer963 <wxd963996380@gmail.com>
Date: Sat, 22 Jan 2022 15:59:30 +0800
Subject: [PATCH] fix a cte block with same name for many times

---
 datafusion/src/sql/planner.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/datafusion/src/sql/planner.rs b/datafusion/src/sql/planner.rs
index 8be5d7c7b061..f06004440492 100644
--- a/datafusion/src/sql/planner.rs
+++ b/datafusion/src/sql/planner.rs
@@ -216,6 +216,14 @@ impl<'a, S: ContextProvider> SqlToRel<'a, S> {
             // Process CTEs from top to bottom
             // do not allow self-references
             for cte in &with.cte_tables {
+                // A `WITH` block can't use the same name for many times
+                let cte_name: &str = cte.alias.name.value.as_ref();
+                if ctes.contains_key(cte_name) {
+                    return Err(DataFusionError::SQL(ParserError(format!(
+                        "WITH query name {:?} specified more than once",
+                        cte_name
+                    ))));
+                }
                 // create logical plan & pass backreferencing CTEs
                 let logical_plan = self.query_to_plan_with_alias(
                     &cte.query,
@@ -3882,6 +3890,14 @@ mod tests {
                                     \n      TableScan: lineitem projection=None";
         quick_test(sql, expected);
     }
+
+    #[test]
+    fn cte_use_same_name_multiple_times() {
+        let sql = "with a as (select * from person), a as (select * from orders) select * from a;";
+        let expected = "SQL error: ParserError(\"WITH query name \\\"a\\\" specified more than once\")";
+        let result = logical_plan(sql).err().unwrap();
+        assert_eq!(expected, format!("{}", result));
+    }
 }
 
 fn parse_sql_number(n: &str) -> Result<Expr> {