Zynh0722/nyazoom
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat: upload sql error "handling" and reorder

Browse source
This commit is contained in:
Zynh Ludwig 2024-11-16 05:24:38 -08:00
parent 99adbd4ea4
commit 5d577df56a
1 changed files with 10 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/router/upload.rs
View file

@ -13,7 +13,6 @@ use sanitize_filename_reader_friendly::sanitize;
use tokio::io; use tokio::io;
use tokio_util::{compat::FuturesAsyncWriteCompatExt, io::StreamReader}; use tokio_util::{compat::FuturesAsyncWriteCompatExt, io::StreamReader};
use tower_http::limit::RequestBodyLimitLayer; use tower_http::limit::RequestBodyLimitLayer;
use tracing::debug;
use crate::{askama::DownloadLinkFragment, cache, util, AppState, UploadRecord}; use crate::{askama::DownloadLinkFragment, cache, util, AppState, UploadRecord};
@ -77,26 +76,22 @@ async fn upload_to_zip(
let record = UploadRecord::new(archive_path); let record = UploadRecord::new(archive_path);
let mut records = state.records.lock().await; let mut records = state.records.lock().await;
{
let mut conn = state.pool.acquire().await.unwrap();
let path = record.file.clone().into_os_string().into_string().unwrap();
let id = sqlx::query_file!("queries/records/new.sql", path, 5)
.execute(&mut *conn)
.await
.unwrap()
.last_insert_rowid();
// TODO: Looks like I actually gotta store cache_name lmfao
debug!("Saved Record {id}");
}
records.insert(cache_name.clone(), record.clone()); records.insert(cache_name.clone(), record.clone());
let records_cache = records.clone(); let records_cache = records.clone();
// Manually drop the records mutex guard // Manually drop the records mutex guard
drop(records); drop(records);
{
let mut conn = state.pool.acquire().await.unwrap();
let cache_name = cache_name.clone();
sqlx::query_file!("queries/records/new.sql", cache_name, 5)
.execute(&mut *conn)
.await
.map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?;
}
cache::write_to_cache(&records_cache) cache::write_to_cache(&records_cache)
.await .await
.map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?; .map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?;