Merge branch 'main' into explain-join

Author: mergify[bot]

docs/doc/13-sql-reference/20-system-tables/system-configs.md

@@ -43,9 +43,6 @@ mysql> SELECT * FROM system.configs;
 | query   | management_mode                        | false                          |             |
 | query   | jwt_key_file                           |                                |             |
 | query   | jwt_key_files                          |                                |             |
-| query   | async_insert_max_data_size             | 10000                          |             |
-| query   | async_insert_busy_timeout              | 200                            |             |
-| query   | async_insert_stale_timeout             | 0                              |             |
 | query   | users                                  |                                |             |
 | query   | share_endpoint_address                 |                                |             |
 | query   | share_endpoint_auth_token_file         |                                |             |

docs/doc/90-contributing/01-rfcs/20220602-async-insert.md

@@ -2194,6 +2194,8 @@ impl<'ast> Visitor<'ast> for AstFormatVisitor {
                 table,
                 alias,
                 travel_point,
+                pivot,
+                unpivot,
             } => {
                 let mut name = String::new();
                 name.push_str("TableIdentifier ");
@@ -2207,6 +2209,16 @@ impl<'ast> Visitor<'ast> for AstFormatVisitor {
                 }
                 name.push_str(&table.to_string());
 
+                if let Some(pivot) = pivot {
+                    name.push(' ');
+                    name.push_str(&pivot.to_string());
+                }
+
+                if let Some(unpivot) = unpivot {
+                    name.push(' ');
+                    name.push_str(&unpivot.to_string());
+                }
+
                 let mut children = Vec::new();
                 if let Some(travel_point) = travel_point {
                     self.visit_time_travel_point(travel_point);

src/query/ast/src/ast/format/ast_format.rs

@@ -275,6 +275,8 @@ pub(crate) fn pretty_table(table: TableReference) -> RcDoc<'static> {
             table,
             alias,
             travel_point,
+            pivot,
+            unpivot,
         } => if let Some(catalog) = catalog {
             RcDoc::text(catalog.to_string()).append(RcDoc::text("."))
         } else {
@@ -286,6 +288,16 @@ pub(crate) fn pretty_table(table: TableReference) -> RcDoc<'static> {
             RcDoc::nil()
         })
         .append(RcDoc::text(table.to_string()))
+        .append(if let Some(pivot) = pivot {
+            RcDoc::text(format!(" {pivot}"))
+        } else {
+            RcDoc::nil()
+        })
+        .append(if let Some(unpivot) = unpivot {
+            RcDoc::text(format!(" {unpivot}"))
+        } else {
+            RcDoc::nil()
+        })
         .append(if let Some(TimeTravelPoint::Snapshot(sid)) = travel_point {
             RcDoc::text(format!(" AT (SNAPSHOT => {sid})"))
         } else if let Some(TimeTravelPoint::Timestamp(ts)) = travel_point {

src/query/ast/src/ast/format/syntax/query.rs

@@ -145,6 +145,26 @@ pub enum SelectTarget {
     },
 }
 
+impl SelectTarget {
+    pub fn is_star(&self) -> bool {
+        match self {
+            SelectTarget::AliasedExpr { .. } => false,
+            SelectTarget::QualifiedName { qualified, .. } => {
+                matches!(qualified.last(), Some(Indirection::Star(_)))
+            }
+        }
+    }
+
+    pub fn exclude(&mut self, exclude: Vec<Identifier>) {
+        match self {
+            SelectTarget::AliasedExpr { .. } => unreachable!(),
+            SelectTarget::QualifiedName { exclude: e, .. } => {
+                *e = Some(exclude);
+            }
+        }
+    }
+}
+
 pub type QualifiedName = Vec<Indirection>;
 
 /// Indirection of a select result, like a part of `db.table.column`.
@@ -164,6 +184,20 @@ pub enum TimeTravelPoint {
     Timestamp(Box<Expr>),
 }
 
+#[derive(Debug, Clone, PartialEq)]
+pub struct Pivot {
+    pub aggregate: Expr,
+    pub value_column: Identifier,
+    pub values: Vec<Expr>,
+}
+
+#[derive(Debug, Clone, PartialEq)]
+pub struct Unpivot {
+    pub value_column: Identifier,
+    pub column_name: Identifier,
+    pub names: Vec<Identifier>,
+}
+
 /// A table name or a parenthesized subquery with an optional alias
 #[derive(Debug, Clone, PartialEq)]
 pub enum TableReference {
@@ -175,6 +209,8 @@ pub enum TableReference {
         table: Identifier,
         alias: Option<TableAlias>,
         travel_point: Option<TimeTravelPoint>,
+        pivot: Option<Box<Pivot>>,
+        unpivot: Option<Box<Unpivot>>,
     },
     // `TABLE(expr)[ AS alias ]`
     TableFunction {
@@ -202,6 +238,22 @@ pub enum TableReference {
     },
 }
 
+impl TableReference {
+    pub fn pivot(&self) -> Option<&Pivot> {
+        match self {
+            TableReference::Table { pivot, .. } => pivot.as_ref().map(|b| b.as_ref()),
+            _ => None,
+        }
+    }
+
+    pub fn unpivot(&self) -> Option<&Unpivot> {
+        match self {
+            TableReference::Table { unpivot, .. } => unpivot.as_ref().map(|b| b.as_ref()),
+            _ => None,
+        }
+    }
+}
+
 #[derive(Debug, Clone, PartialEq, Eq)]
 pub struct TableAlias {
     pub name: Identifier,
@@ -282,6 +334,28 @@ impl Display for TableAlias {
     }
 }
 
+impl Display for Pivot {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        write!(f, "PIVOT({} FOR {} IN (", self.aggregate, self.value_column)?;
+        write_comma_separated_list(f, &self.values)?;
+        write!(f, "))")?;
+        Ok(())
+    }
+}
+
+impl Display for Unpivot {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        write!(
+            f,
+            "UNPIVOT({} FOR {} IN (",
+            self.value_column, self.column_name
+        )?;
+        write_comma_separated_list(f, &self.names)?;
+        write!(f, "))")?;
+        Ok(())
+    }
+}
+
 impl Display for TableReference {
     fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
         match self {
@@ -292,6 +366,8 @@ impl Display for TableReference {
                 table,
                 alias,
                 travel_point,
+                pivot,
+                unpivot,
             } => {
                 write_period_separated_list(
                     f,
@@ -309,6 +385,13 @@ impl Display for TableReference {
                 if let Some(alias) = alias {
                     write!(f, " AS {alias}")?;
                 }
+                if let Some(pivot) = pivot {
+                    write!(f, " {pivot}")?;
+                }
+
+                if let Some(unpivot) = unpivot {
+                    write!(f, " {unpivot}")?;
+                }
             }
             TableReference::TableFunction {
                 span: _,
@@ -545,6 +628,7 @@ impl Display for CTE {
         Ok(())
     }
 }
+
 impl Display for With {
     fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
         if self.recursive {