use crate::{ForeignKeyAction, TableForeignKey, backend::ForeignKeyBuilder, types::*, prepare::*};
#[derive(Debug, Clone)]
pub struct ForeignKeyCreateStatement {
pub(crate) foreign_key: TableForeignKey,
pub(crate) inside_table_creation: bool,
}
impl Default for ForeignKeyCreateStatement {
fn default() -> Self {
Self::new()
}
}
impl ForeignKeyCreateStatement {
pub fn new() -> Self {
Self {
foreign_key: Default::default(),
inside_table_creation: false,
}
}
pub fn name(mut self, name: &str) -> Self {
self.foreign_key.name(name);
self
}
pub fn table<T: 'static, R: 'static>(mut self, table: T, ref_table: R) -> Self
where T: Iden, R: Iden {
self.foreign_key.table(table, ref_table);
self
}
pub fn col<T: 'static, R: 'static>(mut self, column: T, ref_column: R) -> Self
where T: Iden, R: Iden {
self.foreign_key.col(column, ref_column);
self
}
pub fn on_delete(mut self, action: ForeignKeyAction) -> Self {
self.foreign_key.on_delete(action);
self
}
pub fn on_update(mut self, action: ForeignKeyAction) -> Self {
self.foreign_key.on_update(action);
self
}
pub fn build<T: ForeignKeyBuilder>(&self, foreign_key_builder: T) -> String {
let mut sql = SqlWriter::new();
foreign_key_builder.prepare_foreign_key_create_statement(self, &mut sql);
sql.result()
}
pub fn build_any(&self, foreign_key_builder: &dyn ForeignKeyBuilder) -> String {
let mut sql = SqlWriter::new();
foreign_key_builder.prepare_foreign_key_create_statement(self, &mut sql);
sql.result()
}
pub fn to_string<T: ForeignKeyBuilder>(&self, foreign_key_builder: T) -> String {
self.build(foreign_key_builder)
}
}