sql/schema/
table.rs

1use crate::{Dialect, ToSql};
2use crate::schema::column::Column;
3use crate::util::SqlExtension;
4
5#[derive(Debug, Clone, Eq, PartialEq)]
6#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
7pub struct Table {
8    pub schema: Option<String>,
9    pub name: String,
10    pub columns: Vec<Column>,
11}
12
13impl Table {
14    pub fn primary_key(&self) -> Option<&Column> {
15        self.columns.iter().find(|c| c.primary_key)
16    }
17
18    pub fn new(name: &str) -> Table {
19        Table {
20            schema: None,
21            name: name.to_string(),
22            columns: vec![],
23        }
24    }
25
26    pub fn column(mut self, column: Column) -> Self {
27        self.columns.push(column);
28        self
29    }
30
31    pub fn schema(mut self, schema: &str) -> Self {
32        self.schema = Some(schema.to_string());
33        self
34    }
35}
36
37impl ToSql for Table {
38    fn write_sql(&self, buf: &mut String, dialect: Dialect) {
39        buf.push_str("CREATE TABLE ");
40        buf.push_table_name(&self.schema, &self.name);
41        buf.push_str(" (\n");
42        buf.push_sql_sequence(&self.columns, ",\n", dialect);
43        buf.push_str("\n)");
44    }
45}
sql/schema/table.rs

sql/schema/
table.rs
