surql_parser/upstream/sql/statements/define/
bucket.rs1use super::DefineKind;
2use crate::upstream::fmt::CoverStmts;
3use crate::upstream::sql::{Expr, Literal, Permission};
4use surrealdb_types::{SqlFormat, ToSql, write_sql};
5#[derive(Clone, Debug, Eq, PartialEq)]
6#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
7pub struct DefineBucketStatement {
8 pub kind: DefineKind,
9 pub name: Expr,
10 pub backend: Option<Expr>,
11 pub permissions: Permission,
12 pub readonly: bool,
13 pub comment: Expr,
14}
15impl Default for DefineBucketStatement {
16 fn default() -> Self {
17 Self {
18 kind: DefineKind::Default,
19 name: Expr::Literal(Literal::None),
20 backend: None,
21 permissions: Permission::default(),
22 readonly: false,
23 comment: Expr::Literal(Literal::None),
24 }
25 }
26}
27impl ToSql for DefineBucketStatement {
28 fn fmt_sql(&self, f: &mut String, sql_fmt: SqlFormat) {
29 write_sql!(f, sql_fmt, "DEFINE BUCKET");
30 match self.kind {
31 DefineKind::Default => {}
32 DefineKind::Overwrite => write_sql!(f, sql_fmt, " OVERWRITE"),
33 DefineKind::IfNotExists => write_sql!(f, sql_fmt, " IF NOT EXISTS"),
34 }
35 write_sql!(f, sql_fmt, " {}", CoverStmts(&self.name));
36 if self.readonly {
37 write_sql!(f, sql_fmt, " READONLY");
38 }
39 if let Some(ref backend) = self.backend {
40 write_sql!(f, sql_fmt, " BACKEND {}", CoverStmts(backend));
41 }
42 write_sql!(f, sql_fmt, " PERMISSIONS {}", self.permissions);
43 if !matches!(self.comment, Expr::Literal(Literal::None)) {
44 write_sql!(f, sql_fmt, " COMMENT {}", CoverStmts(&self.comment));
45 }
46 }
47}