actually add the file
This commit is contained in:
parent
c78d7a03b7
commit
305a8aff9e
58
migration/src/m20220928_015108_concurrency_limits.rs
Normal file
58
migration/src/m20220928_015108_concurrency_limits.rs
Normal file
@ -0,0 +1,58 @@
|
|||||||
|
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> {
|
||||||
|
// add a field to the UserKeys table
|
||||||
|
manager
|
||||||
|
.alter_table(
|
||||||
|
sea_query::Table::alter()
|
||||||
|
.table(UserKeys::Table)
|
||||||
|
// add column for a better version of rate limiting
|
||||||
|
.add_column(
|
||||||
|
ColumnDef::new(UserKeys::MaxConcurrentRequests)
|
||||||
|
.big_unsigned()
|
||||||
|
.null()
|
||||||
|
.default(200),
|
||||||
|
)
|
||||||
|
.to_owned(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
|
||||||
|
// put the UserKeys back to how it was before our migrations
|
||||||
|
manager
|
||||||
|
.alter_table(
|
||||||
|
sea_query::Table::alter()
|
||||||
|
.table(UserKeys::Table)
|
||||||
|
.to_owned()
|
||||||
|
.drop_column(UserKeys::MaxConcurrentRequests)
|
||||||
|
.to_owned(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// copied from *_log_reverts.rs, but added new columns
|
||||||
|
#[derive(Iden)]
|
||||||
|
pub enum UserKeys {
|
||||||
|
Table,
|
||||||
|
// we don't touch some of the columns
|
||||||
|
// Id,
|
||||||
|
// UserId,
|
||||||
|
// ApiKey,
|
||||||
|
// Description,
|
||||||
|
// PrivateTxs,
|
||||||
|
// Active,
|
||||||
|
// RequestsPerMinute,
|
||||||
|
// LogRevertChance,
|
||||||
|
// AllowedIps,
|
||||||
|
// AllowedOrigins,
|
||||||
|
// AllowedReferers,
|
||||||
|
// AllowedUserAgents,
|
||||||
|
MaxConcurrentRequests,
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user