feat(query): support tag_reference table function (#19221)

* feat(query): support tag_reference table function

* fix stage privilege check

* fix unit test

Author: TCeason

src/query/ast/src/parser/parser.rs

@@ -20,18 +20,22 @@ use pretty_assertions::assert_eq;
 use crate::ParseError;
 use crate::Range;
 use crate::Result;
+use crate::ast::DatabaseRef;
 use crate::ast::ExplainKind;
 use crate::ast::Expr;
 use crate::ast::Identifier;
 use crate::ast::Literal;
 use crate::ast::SelectTarget;
 use crate::ast::Statement;
 use crate::ast::StatementWithFormat;
+use crate::ast::TableRef;
 use crate::parser::Backtrace;
 use crate::parser::common::IResult;
 use crate::parser::common::comma_separated_list0;
 use crate::parser::common::comma_separated_list1;
+use crate::parser::common::database_ref;
 use crate::parser::common::ident;
+use crate::parser::common::table_ref;
 use crate::parser::common::transform_span;
 use crate::parser::error::display_parser_error;
 use crate::parser::expr::expr;
@@ -66,6 +70,20 @@ pub fn parse_expr(tokens: &[Token], dialect: Dialect) -> Result<Expr> {
     run_parser(tokens, dialect, ParseMode::Default, false, expr)
 }
 
+/// Parse a table reference string like "table", "db.table", or "catalog.db.table".
+/// Correctly handles quoted identifiers like `"my.weird.table"`.
+pub fn parse_table_ref(sql: &str, dialect: Dialect) -> Result<TableRef> {
+    let tokens = tokenize_sql(sql)?;
+    run_parser(&tokens, dialect, ParseMode::Default, false, table_ref)
+}
+
+/// Parse a database reference string like "db" or "catalog.db".
+/// Correctly handles quoted identifiers.
+pub fn parse_database_ref(sql: &str, dialect: Dialect) -> Result<DatabaseRef> {
+    let tokens = tokenize_sql(sql)?;
+    run_parser(&tokens, dialect, ParseMode::Default, false, database_ref)
+}
+
 pub fn parse_comma_separated_exprs(tokens: &[Token], dialect: Dialect) -> Result<Vec<Expr>> {
     run_parser(tokens, dialect, ParseMode::Default, true, |i| {
         comma_separated_list0(expr)(i)

src/query/service/src/table_functions/mod.rs

@@ -31,6 +31,7 @@ mod sync_crash_me;
 mod system;
 mod table_function;
 mod table_function_factory;
+mod tag_references;
 mod temporary_tables_table;
 mod udf_table;
 
@@ -46,5 +47,6 @@ pub use system::get_fuse_table_snapshot;
 pub use system::get_fuse_table_statistics;
 pub use table_function::TableFunction;
 pub use table_function_factory::TableFunctionFactory;
+pub use tag_references::TagReferencesTable;
 pub use temporary_tables_table::TemporaryTablesTable;
 pub use udf_table::UDTFTable;

src/query/service/src/table_functions/table_function_factory.rs

@@ -66,6 +66,7 @@ use crate::table_functions::show_variables::ShowVariables;
 use crate::table_functions::srf::RangeTable;
 use crate::table_functions::sync_crash_me::SyncCrashMeTable;
 use crate::table_functions::system::TableStatisticsFunc;
+use crate::table_functions::tag_references::TagReferencesTable;
 type TableFunctionCreators = RwLock<HashMap<String, (MetaId, Arc<dyn TableFunctionCreator>)>>;
 
 pub trait TableFunctionCreator: Send + Sync {
@@ -324,6 +325,11 @@ impl TableFunctionFactory {
             (next_id(), Arc::new(PolicyReferencesTable::create)),
         );
 
+        creators.insert(
+            "tag_references".to_string(),
+            (next_id(), Arc::new(TagReferencesTable::create)),
+        );
+
         creators.insert(
             "udf_echo".to_string(),
             (next_id(), Arc::new(UdfEchoTable::create)),

src/query/service/src/table_functions/tag_references/mod.rs

@@ -0,0 +1,17 @@
+// Copyright 2021 Datafuse Labs
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+mod tag_references_table;
+
+pub use tag_references_table::TagReferencesTable;