Skip to main content

surrealdb_sql/statements/define/
namespace.rs

1use surrealdb_types::{SqlFormat, ToSql, write_sql};
2
3use super::DefineKind;
4use crate::{CoverStmts, Expr, Literal};
5
6#[derive(Clone, Debug, PartialEq, Eq)]
7#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
8pub struct DefineNamespaceStatement {
9	pub kind: DefineKind,
10	pub id: Option<u32>,
11	pub name: Expr,
12	pub comment: Expr,
13}
14
15impl Default for DefineNamespaceStatement {
16	fn default() -> Self {
17		Self {
18			kind: DefineKind::Default,
19			id: None,
20			name: Expr::Literal(Literal::None),
21			comment: Expr::Literal(Literal::None),
22		}
23	}
24}
25
26impl ToSql for DefineNamespaceStatement {
27	fn fmt_sql(&self, f: &mut String, sql_fmt: SqlFormat) {
28		write_sql!(f, sql_fmt, "DEFINE NAMESPACE");
29		match self.kind {
30			DefineKind::Default => {}
31			DefineKind::Overwrite => write_sql!(f, sql_fmt, " OVERWRITE"),
32			DefineKind::IfNotExists => write_sql!(f, sql_fmt, " IF NOT EXISTS"),
33		}
34		write_sql!(f, sql_fmt, " {}", CoverStmts(&self.name));
35		if !matches!(self.comment, Expr::Literal(Literal::None)) {
36			write_sql!(f, sql_fmt, " COMMENT {}", CoverStmts(&self.comment));
37		}
38	}
39}