use super::common::*;
use crate::{backend::SchemaBuilder, prepare::*, types::*, SchemaStatementBuilder};
#[derive(Debug, Clone)]
pub struct IndexCreateStatement {
pub(crate) table: Option<DynIden>,
pub(crate) index: TableIndex,
pub(crate) primary: bool,
pub(crate) unique: bool,
pub(crate) index_type: Option<IndexType>,
}
#[derive(Debug, Clone)]
pub enum IndexType {
BTree,
FullText,
Hash,
Custom(DynIden),
}
impl Default for IndexCreateStatement {
fn default() -> Self {
Self::new()
}
}
impl IndexCreateStatement {
pub fn new() -> Self {
Self {
table: None,
index: Default::default(),
primary: false,
unique: false,
index_type: None,
}
}
pub fn name(&mut self, name: &str) -> &mut Self {
self.index.name(name);
self
}
pub fn table<T: 'static>(&mut self, table: T) -> &mut Self
where
T: Iden,
{
self.table = Some(SeaRc::new(table));
self
}
pub fn col<C: 'static>(&mut self, col: C) -> &mut Self
where
C: IntoIndexColumn,
{
self.index.col(col.into_index_column());
self
}
pub fn primary(&mut self) -> &mut Self {
self.primary = true;
self
}
pub fn unique(&mut self) -> &mut Self {
self.unique = true;
self
}
pub fn full_text(&mut self) -> &mut Self {
self.index_type(IndexType::FullText)
}
pub fn index_type(&mut self, index_type: IndexType) -> &mut Self {
self.index_type = Some(index_type);
self
}
pub fn is_primary_key(&self) -> bool {
self.primary
}
pub fn is_unique_key(&self) -> bool {
self.unique
}
pub fn get_index_spec(&self) -> &TableIndex {
&self.index
}
pub fn take(&mut self) -> Self {
Self {
table: self.table.take(),
index: self.index.take(),
primary: self.primary,
unique: self.unique,
index_type: self.index_type.take(),
}
}
}
impl SchemaStatementBuilder for IndexCreateStatement {
fn build<T: SchemaBuilder>(&self, schema_builder: T) -> String {
let mut sql = SqlWriter::new();
schema_builder.prepare_index_create_statement(self, &mut sql);
sql.result()
}
fn build_any(&self, schema_builder: &dyn SchemaBuilder) -> String {
let mut sql = SqlWriter::new();
schema_builder.prepare_index_create_statement(self, &mut sql);
sql.result()
}
}