Merge pull request #3 from groobyming/refactor/#122360719

refactor: sqlparser-rs 扩展 plain select --story=122360719

Author: wking1986

src/ast/query.rs

@@ -294,6 +294,7 @@ pub enum SelectFlavor {
 #[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
 #[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
+#[cfg_attr(feature = "visitor", visit(with = "visit_select"))]
 pub struct Select {
     /// Token for the `SELECT` keyword
     pub select_token: AttachedToken,
@@ -304,6 +305,7 @@ pub struct Select {
     /// Whether the top was located before `ALL`/`DISTINCT`
     pub top_before_distinct: bool,
     /// projection expressions
+    #[cfg_attr(feature = "visitor", visit(with = "visit_projection"))]
     pub projection: Vec<SelectItem>,
     /// INTO
     pub into: Option<SelectInto>,
@@ -541,6 +543,7 @@ impl fmt::Display for NamedWindowDefinition {
 #[derive(Debug, Clone, PartialEq, PartialOrd, Eq, Ord, Hash)]
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
 #[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
+#[cfg_attr(feature = "visitor", visit(with = "visit_with"))]
 pub struct With {
     /// Token for the "WITH" keyword
     pub with_token: AttachedToken,

src/ast/visitor.rs

@@ -18,8 +18,8 @@
 //! Recursive visitors for ast Nodes. See [`Visitor`] for more details.
 
 use crate::ast::{
-    Expr, Function, GroupByExpr, ObjectName, OrderBy, Query, SelectItem, Statement, TableFactor,
-    Value,
+    Expr, Function, GroupByExpr, ObjectName, OrderBy, Query, Select, SelectItem, Statement,
+    TableFactor, Value, With,
 };
 use core::ops::ControlFlow;
 use sqlparser::ast::Ident;
@@ -307,6 +307,36 @@ pub trait Visitor {
     fn post_visit_order_by(&mut self, _order_by: &OrderBy) -> ControlFlow<Self::Break> {
         ControlFlow::Continue(())
     }
+
+    /// Invoked for any Select that appear in the AST before visiting children
+    fn pre_visit_select(&mut self, _select: &Select) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Select that appear in the AST after visiting children
+    fn post_visit_select(&mut self, _select: &Select) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any With that appear in the AST before visiting children
+    fn pre_visit_with(&mut self, _select: &With) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any With that appear in the AST after visiting children
+    fn post_visit_with(&mut self, _select: &With) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Projection that appear in the AST before visiting children
+    fn pre_visit_projection(&mut self, _select: &Vec<SelectItem>) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Projection that appear in the AST after visiting children
+    fn post_visit_projection(&mut self, _select: &Vec<SelectItem>) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
 }
 
 /// A visitor that can be used to mutate an AST tree.
@@ -487,6 +517,36 @@ pub trait VisitorMut {
     fn post_visit_order_by(&mut self, _order_by: &mut OrderBy) -> ControlFlow<Self::Break> {
         ControlFlow::Continue(())
     }
+
+    /// Invoked for any Select Expr that appear in the AST before visiting children
+    fn pre_visit_select(&mut self, _select: &mut Select) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Select Expr that appear in the AST after visiting children
+    fn post_visit_select(&mut self, _select: &mut Select) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any With Expr that appear in the AST before visiting children
+    fn pre_visit_with(&mut self, _select: &mut With) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any With Expr that appear in the AST after visiting children
+    fn post_visit_with(&mut self, _select: &mut With) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Projection that appear in the AST before visiting children
+    fn pre_visit_projection(&mut self, _select: &mut Vec<SelectItem>) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Projection that appear in the AST after visiting children
+    fn post_visit_projection(&mut self, _select: &mut Vec<SelectItem>) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
 }
 
 struct RelationVisitor<F>(F);