readyset-sql: Convert TRUNCATE

Change-Id: I9fe4f8887b7a9ded62170ad0b451543ac0130359

Author: mvzink

readyset-sql/src/ast/query.rs

@@ -187,6 +187,7 @@ impl TryFromDialect<sqlparser::ast::Statement> for SqlQuery {
             Rollback { .. } => Ok(Self::Rollback(RollbackStatement {})),
             Commit { .. } => Ok(Self::Commit(CommitStatement {})),
             alter @ AlterTable { .. } => Ok(Self::AlterTable(alter.try_into_dialect(dialect)?)),
+            truncate @ Truncate { .. } => Ok(Self::Truncate(truncate.try_into_dialect(dialect)?)),
             CreateDatabase { .. } => skipped!("CREATE DATABASE"),
             _ => not_yet_implemented!("other query: {value:?}"),
         }

readyset-sql/src/ast/truncate.rs

@@ -5,7 +5,7 @@ use readyset_util::fmt::fmt_with;
 use serde::{Deserialize, Serialize};
 use test_strategy::Arbitrary;
 
-use crate::{ast::*, Dialect, DialectDisplay};
+use crate::{ast::*, AstConversionError, Dialect, DialectDisplay, IntoDialect, TryFromDialect};
 
 #[derive(Clone, Debug, Eq, Hash, PartialEq, Serialize, Deserialize, Arbitrary)]
 
@@ -21,6 +21,42 @@ pub struct TruncateStatement {
     pub cascade: bool,
 }
 
+impl TryFromDialect<sqlparser::ast::Statement> for TruncateStatement {
+    fn try_from_dialect(
+        value: sqlparser::ast::Statement,
+        dialect: Dialect,
+    ) -> Result<Self, AstConversionError> {
+        if let sqlparser::ast::Statement::Truncate {
+            table_names,
+            partitions: _,
+            table: _,
+            // XXX(mvzink): I believe sqlparser is incorrect here; ONLY should be on each table
+            // target, not the whole statement. We interpret it as applying to all tables, but
+            // should probably upstream a fix (or verify that sqlparser is correct).
+            only,
+            identity,
+            cascade,
+            on_cluster: _,
+        } = value
+        {
+            let tables = table_names
+                .into_iter()
+                .map(|table| TruncateTable {
+                    relation: table.name.into_dialect(dialect),
+                    only,
+                })
+                .collect();
+            Ok(Self {
+                tables,
+                restart_identity: identity == Some(sqlparser::ast::TruncateIdentityOption::Restart),
+                cascade: cascade == Some(sqlparser::ast::CascadeOption::Cascade),
+            })
+        } else {
+            failed!("Should only be called on TRUNCATE statement, got {value:?}")
+        }
+    }
+}
+
 impl DialectDisplay for TruncateStatement {
     fn display(&self, dialect: Dialect) -> impl fmt::Display + '_ {
         fmt_with(move |f| {