Merge pull request BKBASE-Plugin#1 from groobyming/feat/#122344757

feat: sqlparser-rs 扩展访问切入点 --story=122344757

Author: wking1986

src/ast/mod.rs

@@ -153,6 +153,7 @@ where
 #[derive(Debug, Clone, PartialOrd, Ord)]
 #[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
 #[cfg_attr(feature = "visitor", derive(Visit, VisitMut))]
+#[cfg_attr(feature = "visitor", visit(with = "visit_ident"))]
 pub struct Ident {
     /// The value of the identifier without quotes.
     pub value: String,
@@ -6305,6 +6306,7 @@ impl fmt::Display for CloseCursor {
 #[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_function"))]
 pub struct Function {
     pub name: ObjectName,
     /// Flags whether this function call uses the [ODBC syntax].

src/ast/query.rs

@@ -317,6 +317,7 @@ pub struct Select {
     /// [ClickHouse](https://clickhouse.com/docs/en/sql-reference/statements/select/prewhere)
     pub prewhere: Option<Expr>,
     /// WHERE
+    #[cfg_attr(feature = "visitor", visit(with = "visit_where"))]
     pub selection: Option<Expr>,
     /// GROUP BY
     pub group_by: GroupByExpr,
@@ -629,6 +630,7 @@ pub enum SelectItemQualifiedWildcardKind {
 #[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_item"))]
 pub enum SelectItem {
     /// Any expression, not followed by `[ AS ] alias`
     UnnamedExpr(Expr),
@@ -2235,6 +2237,7 @@ pub enum OrderByKind {
 #[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_order_by"))]
 pub struct OrderBy {
     pub kind: OrderByKind,
 
@@ -2626,6 +2629,7 @@ impl fmt::Display for GroupByWithModifier {
 #[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_group_by_expr"))]
 pub enum GroupByExpr {
     /// ALL syntax of [Snowflake], [DuckDB] and [ClickHouse].
     ///

src/ast/visitor.rs

@@ -17,8 +17,12 @@
 
 //! Recursive visitors for ast Nodes. See [`Visitor`] for more details.
 
-use crate::ast::{Expr, ObjectName, Query, Statement, TableFactor, Value};
+use crate::ast::{
+    Expr, Function, GroupByExpr, ObjectName, OrderBy, Query, SelectItem, Statement, TableFactor,
+    Value,
+};
 use core::ops::ControlFlow;
+use sqlparser::ast::Ident;
 
 /// A type that can be visited by a [`Visitor`]. See [`Visitor`] for
 /// recursively visiting parsed SQL statements.
@@ -243,6 +247,66 @@ pub trait Visitor {
     fn post_visit_value(&mut self, _value: &Value) -> ControlFlow<Self::Break> {
         ControlFlow::Continue(())
     }
+
+    /// Invoked for any Identifier that appear in the AST before visiting children
+    fn pre_visit_ident(&mut self, _ident: &Ident) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any Identifier that appear in the AST after visiting children
+    fn post_visit_ident(&mut self, _ident: &Ident) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any functions that appear in the AST before visiting children
+    fn pre_visit_function(&mut self, _function: &Function) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any functions that appear in the AST after visiting children
+    fn post_visit_function(&mut self, _function: &Function) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any SelectItem that appear in the AST before visiting children
+    fn pre_visit_select_item(&mut self, _select_item: &SelectItem) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any SelectItem that appear in the AST after visiting children
+    fn post_visit_select_item(&mut self, _select_item: &SelectItem) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for Where Expr that appear in the AST before visiting children
+    fn pre_visit_where(&mut self, _where: &Expr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for Where Expr that appear in the AST after visiting children
+    fn post_visit_where(&mut self, _where: &Expr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any GroupBy Expr that appear in the AST before visiting children
+    fn pre_visit_group_by_expr(&mut self, _group_by: &GroupByExpr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any GroupBy Expr that appear in the AST after visiting children
+    fn post_visit_group_by_expr(&mut self, _group_by: &GroupByExpr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any OrderBy Expr that appear in the AST before visiting children
+    fn pre_visit_order_by(&mut self, _order_by: &OrderBy) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any OrderBy Expr that appear in the AST after visiting children
+    fn post_visit_order_by(&mut self, _order_by: &OrderBy) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
 }
 
 /// A visitor that can be used to mutate an AST tree.
@@ -357,6 +421,62 @@ pub trait VisitorMut {
     fn post_visit_value(&mut self, _value: &mut Value) -> ControlFlow<Self::Break> {
         ControlFlow::Continue(())
     }
+
+    /// Invoked for any functions that appear in the AST before visiting children
+    fn pre_visit_function(&mut self, _function: &mut Function) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any functions that appear in the AST after visiting children
+    fn post_visit_function(&mut self, _function: &mut Function) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any SelectItem that appear in the AST before visiting children
+    fn pre_visit_select_item(&mut self, _select_item: &mut SelectItem) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any SelectItem that appear in the AST after visiting children
+    fn post_visit_select_item(
+        &mut self,
+        _select_item: &mut SelectItem,
+    ) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for Where Expr that appear in the AST before visiting children
+    fn pre_visit_where(&mut self, _where: &mut Expr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for Where Expr that appear in the AST after visiting children
+    fn post_visit_where(&mut self, _where: &mut Expr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any GroupBy Expr that appear in the AST before visiting children
+    fn pre_visit_group_by_expr(&mut self, _group_by: &mut GroupByExpr) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any GroupBy Expr that appear in the AST after visiting children
+    fn post_visit_group_by_expr(
+        &mut self,
+        _group_by: &mut GroupByExpr,
+    ) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any OrderBy Expr that appear in the AST before visiting children
+    fn pre_visit_order_by(&mut self, _order_by: &mut OrderBy) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
+
+    /// Invoked for any OrderBy Expr that appear in the AST after visiting children
+    fn post_visit_order_by(&mut self, _order_by: &mut OrderBy) -> ControlFlow<Self::Break> {
+        ControlFlow::Continue(())
+    }
 }
 
 struct RelationVisitor<F>(F);