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
use crate::mysql::def::{IndexInfo, IndexOrder, IndexType};
use sea_query::{Alias, Iden, Index, IndexCreateStatement, SeaRc};
impl IndexInfo {
pub fn write(&self) -> IndexCreateStatement {
let mut index = Index::create();
if self.name == "PRIMARY" {
index.primary();
} else {
index.name(&self.name);
if self.unique {
index.unique();
}
}
for part in self.parts.iter() {
let pre = if let Some(pre) = part.sub_part {
Some(pre)
} else {
None
};
let ord = if self.parts.len() == 1 {
match part.order {
IndexOrder::Ascending => None,
IndexOrder::Descending => Some(sea_query::IndexOrder::Desc),
IndexOrder::Unordered => None,
}
} else {
None
};
if pre.is_none() && ord.is_none() {
index.col(Alias::new(&part.column));
} else if pre.is_none() && ord.is_some() {
index.col((Alias::new(&part.column), ord.unwrap()));
} else if pre.is_some() && ord.is_none() {
index.col((Alias::new(&part.column), pre.unwrap()));
} else {
index.col((Alias::new(&part.column), pre.unwrap(), ord.unwrap()));
}
}
match self.idx_type {
IndexType::BTree => {}
IndexType::FullText => {
index.index_type(sea_query::IndexType::FullText);
}
IndexType::Hash => {
index.index_type(sea_query::IndexType::Hash);
}
IndexType::RTree => {
index.index_type(sea_query::IndexType::Custom(SeaRc::new(Alias::new(
&self.idx_type.to_string(),
))));
}
IndexType::Spatial => {
index.index_type(sea_query::IndexType::Custom(SeaRc::new(Alias::new(
&self.idx_type.to_string(),
))));
}
}
index
}
}