HoovesSocial/hippofish
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity Actions

add environment variable to skip copying antenna

Browse source
This commit is contained in:
Namekuji 2023-07-10 02:31:50 -04:00
parent 94014e029d
commit 04ac723a5f
No known key found for this signature in database
GPG key ID: 1D62332C07FBA532
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
packages/backend/native-utils/migration/src/m20230709_000510_move_antenna_to_cache.rs
View file

@ -10,13 +10,16 @@ pub struct Migration;
impl MigrationTrait for Migration { impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> { async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
let cache_url = env::var("CACHE_URL").unwrap(); let cache_url = env::var("CACHE_URL").unwrap();
let skip_copy = env::var("ANTENNA_MIGRATION_SKIP").unwrap_or_default();
let copy_limit = env::var("ANTENNA_MIGRATION_LIMIT").unwrap_or_default(); let copy_limit = env::var("ANTENNA_MIGRATION_LIMIT").unwrap_or_default();
let copy_limit: i64 = match copy_limit.parse() { let copy_limit: i64 = match copy_limit.parse() {
Ok(limit) => limit, Ok(limit) => limit,
Err(_) => 0, Err(_) => 0,
}; };
if cache_url != "no" { if skip_copy == "true" {
println!("Skipped antenna migration");
} else {
let prefix = env::var("CACHE_PREFIX").unwrap(); let prefix = env::var("CACHE_PREFIX").unwrap();
let db = manager.get_connection(); let db = manager.get_connection();
@ -102,8 +105,9 @@ impl MigrationTrait for Migration {
break; break;
} }
} }
println!("Migrating antenna [100.00%]");
} }
println!("Migrating antenna [100.00%]");
manager manager
.drop_table( .drop_table(