Merge 4f719e5 into a86c58b

src/ast/mod.rs

@@ -793,6 +793,11 @@ pub enum Expr {
     OuterJoin(Box<Expr>),
     /// A reference to the prior level in a CONNECT BY clause.
     Prior(Box<Expr>),
+    /// Scalar variable creation e.g. `[@]foo INT`
+    ScalarVariable {
+        data_type: DataType,
+        name: String,
+    },
 }
 
 impl fmt::Display for CastFormat {
@@ -1241,6 +1246,9 @@ impl fmt::Display for Expr {
                 write!(f, "{expr} (+)")
             }
             Expr::Prior(expr) => write!(f, "PRIOR {expr}"),
+            Expr::ScalarVariable { data_type, name } => {
+                write!(f, "{name} {data_type}")
+            }
         }
     }
 }
@@ -6441,6 +6449,24 @@ mod tests {
         assert_eq!("CUBE (a, (b, c), d)", format!("{cube}"));
     }
 
+    #[test]
+    fn test_scalar_variable_display() {
+        let scalar_variable = Expr::ScalarVariable {
+            data_type: DataType::Boolean,
+            name: "foo".to_string(),
+        };
+        assert_eq!("foo BOOLEAN", format!("{scalar_variable}"));
+    }
+
+    #[test]
+    fn test_scalar_variable_display_int() {
+        let scalar_variable = Expr::ScalarVariable {
+            data_type: DataType::Int(None),
+            name: "foo".to_string(),
+        };
+        assert_eq!("foo INT", format!("{scalar_variable}"));
+    }
+
     #[test]
     fn test_interval_display() {
         let interval = Expr::Interval(Interval {