60 lines
1.6 KiB
Rust
60 lines
1.6 KiB
Rust
|
use sea_orm_migration::prelude::*;
|
||
|
|
||
|
#[derive(DeriveMigrationName)]
|
||
|
pub struct Migration;
|
||
|
|
||
|
#[async_trait::async_trait]
|
||
|
impl MigrationTrait for Migration {
|
||
|
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
||
|
let db_conn = manager.get_connection();
|
||
|
let db_backend = manager.get_database_backend();
|
||
|
|
||
|
let select_free_id = Query::select()
|
||
|
.column(UserTier::Id)
|
||
|
.column(UserTier::Title)
|
||
|
.from(UserTier::Table)
|
||
|
.and_having(Expr::col(UserTier::Title).eq("Free"))
|
||
|
.to_owned();
|
||
|
|
||
|
let free_id: u64 = db_conn
|
||
|
.query_one(db_backend.build(&select_free_id))
|
||
|
.await?
|
||
|
.expect("Free tier should always exist")
|
||
|
.try_get("", &UserTier::Id.to_string())?;
|
||
|
|
||
|
let user_tiers = Query::insert()
|
||
|
.into_table(UserTier::Table)
|
||
|
.columns([
|
||
|
UserTier::Title,
|
||
|
UserTier::MaxRequestsPerPeriod,
|
||
|
UserTier::MaxConcurrentRequests,
|
||
|
UserTier::DowngradeTierId,
|
||
|
])
|
||
|
.values_panic([
|
||
|
"High Concurrency".into(),
|
||
|
None::<u64>.into(),
|
||
|
Some(300).into(),
|
||
|
Some(free_id).into(),
|
||
|
])
|
||
|
.to_owned();
|
||
|
|
||
|
manager.exec_stmt(user_tiers).await?;
|
||
|
|
||
|
Ok(())
|
||
|
}
|
||
|
|
||
|
async fn down(&self, _manager: &SchemaManager) -> Result<(), DbErr> {
|
||
|
Ok(())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[derive(Iden)]
|
||
|
enum UserTier {
|
||
|
Table,
|
||
|
Id,
|
||
|
Title,
|
||
|
MaxRequestsPerPeriod,
|
||
|
MaxConcurrentRequests,
|
||
|
DowngradeTierId,
|
||
|
}
|