sql-parser: update statement ast

Author: Almaz-KG

src/ast/select.rs

@@ -22,7 +22,7 @@ pub struct Select {
     pub columns: Vec<SelectItem>,
 
     /// The FROM clause
-    pub from: Option<SelectFrom>,
+    pub from: Option<FromClause>,
 
     /// The WHERE clause
     pub where_clause: Option<Box<Expression>>,
@@ -67,7 +67,7 @@ pub enum SelectItem {
 /// An AST for representing a FROM clause in the
 /// [SELECT](https://www.sqlite.org/lang_select.html) statement
 #[derive(Debug, PartialEq, Clone)]
-pub enum SelectFrom {
+pub enum FromClause {
     /// Select from a table
     Table(QualifiedTableName),
 
@@ -78,7 +78,7 @@ pub enum SelectFrom {
     Subquery(SelectFromSubquery),
 
     /// Select from cartesian product of tables
-    Froms(Vec<SelectFrom>),
+    Froms(Vec<FromClause>),
 
     /// Select from a JOIN clause
     Join(JoinClause),
@@ -134,7 +134,7 @@ pub struct SelectFromSubquery {
 /// A clause for a JOIN statement
 #[derive(Debug, PartialEq, Clone)]
 pub struct JoinClause {
-    pub lhs_table: Box<SelectFrom>,
+    pub lhs_table: Box<FromClause>,
 
     pub join_tables: Vec<JoinTable>,
 }
@@ -143,7 +143,7 @@ pub struct JoinClause {
 #[derive(Debug, PartialEq, Clone)]
 pub struct JoinTable {
     pub join_type: JoinType,
-    pub table: Box<SelectFrom>,
+    pub table: Box<FromClause>,
     pub constraints: Option<JoinConstraint>,
 }
 

src/ast/update.rs

@@ -1,3 +1,36 @@
+use super::{
+    ConflictClause, Expression, FromClause, Identifier, QualifiedTableName, ReturningClause,
+};
+
 /// An AST for [UPDATE SQL](https://www.sqlite.org/lang_update.html) statement.
 #[derive(Debug, PartialEq)]
-pub struct UpdateStatement {}
+pub struct UpdateStatement {
+    // The conflict clause defined after the OR keyword is used to specify the
+    // behavior of the UPDATE statement when a constraint violation occurs.
+    pub conflict_clause: Option<ConflictClause>,
+
+    // The table name to update.
+    pub table_name: QualifiedTableName,
+
+    // The set clause to update the table with.
+    pub set_clause: Vec<SetClause>,
+
+    // The from clause to select the rows to update.
+    pub from_clause: Option<FromClause>,
+
+    // The where clause to filter the rows to update.
+    pub where_clause: Option<Expression>,
+
+    // The returning clause to return the updated rows.
+    pub returning_clause: Option<ReturningClause>,
+}
+
+/// A [SetClause](https://www.sqlite.org/syntax/set-clause.html)
+#[derive(Debug, PartialEq)]
+pub enum SetClause {
+    /// A single column assignment, e.g. `column = expression`.
+    ColumnAssignment(Identifier, Expression),
+
+    /// A multiple column assignment, e.g. `(column1, column2) = expression`.
+    MultipleColumnAssignment((Vec<Identifier>, Expression)),
+}

src/parser/delete/mod.rs

@@ -385,7 +385,7 @@ mod test_delete_statements_with_cte {
     use super::test_utils::delete_statement_with_cte_clause;
     use crate::parser::select::test_utils::select_from;
     use crate::parser::test_utils::*;
-    use crate::{Identifier, QualifiedTableName, SelectFrom};
+    use crate::{FromClause, Identifier, QualifiedTableName};
 
     #[test]
     fn test_delete_with_cte() {
@@ -395,7 +395,7 @@ mod test_delete_statements_with_cte {
                 Identifier::Single("cte_1".to_string()),
                 vec![],
                 None,
-                select_from(SelectFrom::Table(QualifiedTableName::from(
+                select_from(FromClause::Table(QualifiedTableName::from(
                     Identifier::from("cte_table"),
                 ))),
             )],
@@ -417,15 +417,15 @@ mod test_delete_statements_with_cte {
                     Identifier::Single("cte_1".to_string()),
                     vec![],
                     None,
-                    select_from(SelectFrom::Table(QualifiedTableName::from(
+                    select_from(FromClause::Table(QualifiedTableName::from(
                         Identifier::Single("cte_table1".to_string()),
                     ))),
                 ),
                 cte_expression(
                     Identifier::Single("cte_2".to_string()),
                     vec![],
                     None,
-                    select_from(SelectFrom::Table(QualifiedTableName::from(
+                    select_from(FromClause::Table(QualifiedTableName::from(
                         Identifier::Single("cte_table2".to_string()),
                     ))),
                 ),

src/parser/mod.rs

@@ -10,6 +10,7 @@ pub(crate) mod expression;
 mod select;
 mod sqlite;
 mod trx;
+mod update;
 mod window_definition;
 
 #[cfg(test)]
@@ -28,6 +29,7 @@ use expression::ExpressionParser;
 use select::{SelectStatementParser, ValuesStatementParser};
 use sqlite::SQLite3StatementParser;
 use trx::TransactionStatementParser;
+use update::UpdateStatementParser;
 use window_definition::WindowDefinitionParser;
 /// A parser for SQLite SQL statements
 pub struct Parser<'a> {
@@ -316,6 +318,9 @@ impl<'a> Parser<'a> {
             Keyword::Delete => {
                 DeleteStatementParser::parse_delete_statement(self).map(Statement::Delete)
             }
+            Keyword::Update => {
+                UpdateStatementParser::parse_update_statement(self).map(Statement::Update)
+            }
             Keyword::Alter => AlterTableStatementParser::parse_alter_table_statement(self)
                 .map(Statement::AlterTable),
             keyword => Err(ParsingError::UnexpectedKeyword(keyword)),