chore:(query): support [SET | USE] CATALOG <catalog_name> (#17844)

Author: TCeason

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

@@ -519,7 +519,7 @@ pub fn statement_body(i: Input) -> IResult<Statement> {
     );
     let use_catalog = map(
         rule! {
-            USE ~ CATALOG ~ #ident
+            (SET | USE)?  ~ CATALOG ~ #ident
         },
         |(_, _, catalog)| Statement::UseCatalog { catalog },
     );

src/query/ast/tests/it/parser.rs

@@ -147,6 +147,8 @@ fn test_statement() {
         r#"create catalog ctl type=hive connection=(url='<hive-meta-store>' thrift_protocol='binary' warehouse='default');"#,
         r#"select current_catalog();"#,
         r#"use catalog ctl;"#,
+        r#"catalog ctl;"#,
+        r#"set catalog ctl;"#,
         r#"create database if not exists a;"#,
         r#"create database ctl.t engine = Default;"#,
         r#"create database t engine = Default;"#,

src/query/ast/tests/it/testdata/stmt.txt

@@ -3495,6 +3495,40 @@ UseCatalog {
 }
 
 
+---------- Input ----------
+catalog ctl;
+---------- Output ---------
+USE CATALOG ctl
+---------- AST ------------
+UseCatalog {
+    catalog: Identifier {
+        span: Some(
+            8..11,
+        ),
+        name: "ctl",
+        quote: None,
+        ident_type: None,
+    },
+}
+
+
+---------- Input ----------
+set catalog ctl;
+---------- Output ---------
+USE CATALOG ctl
+---------- AST ------------
+UseCatalog {
+    catalog: Identifier {
+        span: Some(
+            12..15,
+        ),
+        name: "ctl",
+        quote: None,
+        ident_type: None,
+    },
+}
+
+
 ---------- Input ----------
 create database if not exists a;
 ---------- Output ---------

tests/sqllogictests/suites/tpch_iceberg/table_function.test

@@ -13,7 +13,10 @@ CONNECTION=(
 );
 
 statement ok
-use catalog ctl;
+SET catalog ctl;
+
+statement ok
+catalog ctl;
 
 statement ok
 select * from iceberg_snapshot('tpch', 'lineitem');