rrdb 0.0.2

ready
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76

use crate::ast::{
    types::{Column, ForeignKey, TableName, TableOptions, UniqueKey},
    DDLStatement, SQLStatement,
};

/*
CREATE TABLE [IF NOT EXISTS] [database_name.]table_name (
    column_name data_type [NOT NULL | NULL] [PRIMARY KEY] [COMMENT 'comment'],
    column_name data_type [NOT NULL | NULL] [PRIMARY KEY] [COMMENT 'comment'],
    ...
    PRIMARY KEY (column_name),
    UNIQUE (column_name),
    FOREIGN KEY (column_name) REFERENCES table_name (column_name),
    FOREIGN KEY (column_name) REFERENCES table_name (column_name),
    ...
);
*/

#[derive(Clone, Debug, PartialEq)]
pub struct CreateTableQuery {
    pub table: Option<TableName>,
    pub columns: Vec<Column>,
    pub primary_key: Vec<String>,
    pub foreign_keys: Vec<ForeignKey>,
    pub unique_keys: Vec<UniqueKey>,
    pub table_options: Option<TableOptions>,
    pub if_not_exists: bool,
}

impl CreateTableQuery {
    pub fn builder() -> Self {
        CreateTableQuery {
            table: None,
            columns: vec![],
            primary_key: vec![],
            foreign_keys: vec![],
            unique_keys: vec![],
            table_options: None,
            if_not_exists: false,
        }
    }

    pub fn set_table(mut self, table: TableName) -> Self {
        self.table = Some(table);
        self
    }

    pub fn set_table_option(mut self, option: TableOptions) -> Self {
        self.table_options = Some(option);
        self
    }

    pub fn add_column(mut self, column: Column) -> Self {
        self.columns.push(column);
        self
    }

    pub fn set_primary_key(mut self, columns: Vec<String>) -> Self {
        self.primary_key = columns;
        self
    }

    pub fn add_unique_key(mut self, unique_key: UniqueKey) -> Self {
        self.unique_keys.push(unique_key);
        self
    }

    pub fn set_if_not_exists(mut self, if_not_exists: bool) -> Self {
        self.if_not_exists = if_not_exists;
        self
    }

    pub fn build(self) -> SQLStatement {
        SQLStatement::DDL(DDLStatement::CreateTableQuery(self))
    }
}