92 lines
3.3 KiB
Rust
92 lines
3.3 KiB
Rust
|
// setup
|
||
|
|
use sea_orm::sea_query::{ForeignKeyCreateStatement, ColumnDef, TableCreateStatement};
|
||
|
|
use sea_orm::{Set, EntityTrait, error::*, sea_query, ConnectionTrait, DbConn, ExecResult};
|
||
|
|
pub mod comment;
|
||
|
|
pub mod post;
|
||
|
|
pub use comment::Entity as Comment;
|
||
|
|
pub use post::Entity as Post;
|
||
|
|
use chrono::{Local, Duration, DurationRound};
|
||
|
|
use sea_orm::prelude::Decimal;
|
||
|
|
|
||
|
|
// setup
|
||
|
|
async fn create_table(db: &DbConn, stmt: &TableCreateStatement) -> Result<ExecResult, DbErr> {
|
||
|
|
let builder = db.get_database_backend();
|
||
|
|
db.execute(builder.build(stmt)).await
|
||
|
|
}
|
||
|
|
|
||
|
|
pub async fn create_post_table(db: &DbConn) -> Result<ExecResult, DbErr> {
|
||
|
|
let stmt = sea_query::Table::create()
|
||
|
|
.table(post::Entity)
|
||
|
|
.if_not_exists()
|
||
|
|
.col(
|
||
|
|
ColumnDef::new(post::Column::Id)
|
||
|
|
.integer()
|
||
|
|
.not_null()
|
||
|
|
.auto_increment()
|
||
|
|
.primary_key(),
|
||
|
|
)
|
||
|
|
.col(ColumnDef::new(post::Column::Title).string().not_null())
|
||
|
|
.col(ColumnDef::new(post::Column::Text).string().not_null())
|
||
|
|
.col(ColumnDef::new(post::Column::TeaMandatory).string().not_null())
|
||
|
|
.col(ColumnDef::new(post::Column::TeaOptional).string())
|
||
|
|
.col(ColumnDef::new(post::Column::InsertDate).date().not_null())
|
||
|
|
.col(ColumnDef::new(post::Column::Attachment).string())
|
||
|
|
.to_owned();
|
||
|
|
|
||
|
|
let _result = create_table(db, &stmt).await;
|
||
|
|
|
||
|
|
let stmt = sea_query::Table::create()
|
||
|
|
.table(comment::Entity)
|
||
|
|
.if_not_exists()
|
||
|
|
.col(
|
||
|
|
ColumnDef::new(post::Column::Id)
|
||
|
|
.integer()
|
||
|
|
.not_null()
|
||
|
|
.auto_increment()
|
||
|
|
.primary_key(),
|
||
|
|
)
|
||
|
|
.col(ColumnDef::new(comment::Column::Comment).string().not_null())
|
||
|
|
.col(ColumnDef::new(comment::Column::User).string().not_null())
|
||
|
|
.col(ColumnDef::new(comment::Column::InsertDate).date_time().not_null())
|
||
|
|
.col(ColumnDef::new(comment::Column::IsVisible).boolean().not_null())
|
||
|
|
.col(ColumnDef::new(comment::Column::MyDecimal).decimal().not_null())
|
||
|
|
.col(ColumnDef::new(comment::Column::PostId).integer())
|
||
|
|
.foreign_key(
|
||
|
|
ForeignKeyCreateStatement::new()
|
||
|
|
.name("fk-comment-post")
|
||
|
|
.from_tbl(Comment)
|
||
|
|
.from_col(comment::Column::PostId)
|
||
|
|
.to_tbl(Post)
|
||
|
|
.to_col(post::Column::Id),
|
||
|
|
)
|
||
|
|
.to_owned();
|
||
|
|
|
||
|
|
let res = create_table(db, &stmt).await;
|
||
|
|
|
||
|
|
for i in 1..1000 {
|
||
|
|
let row = post::ActiveModel {
|
||
|
|
title: Set(format!("Test {}", i)),
|
||
|
|
text: Set("some content".to_string()),
|
||
|
|
tea_mandatory: Set(post::Tea::EverydayTea),
|
||
|
|
tea_optional: Set(None),
|
||
|
|
insert_date: Set(Local::now().date_naive()),
|
||
|
|
..Default::default()
|
||
|
|
};
|
||
|
|
let _res = Post::insert(row).exec(db).await;
|
||
|
|
}
|
||
|
|
|
||
|
|
for i in 1..1000 {
|
||
|
|
let row = comment::ActiveModel {
|
||
|
|
comment: Set(format!("Test {}", i)),
|
||
|
|
user: Set("me@home.com".to_string()),
|
||
|
|
my_decimal: Set(Decimal::new(105, 0)),
|
||
|
|
insert_date: Set(Local::now().naive_utc().duration_round(Duration::minutes(1)).unwrap()),
|
||
|
|
is_visible: Set(i%2 == 0),
|
||
|
|
..Default::default()
|
||
|
|
};
|
||
|
|
let _res = Comment::insert(row).exec(db).await;
|
||
|
|
}
|
||
|
|
|
||
|
|
res
|
||
|
|
}
|