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

feat: db downloads

Browse source
This commit is contained in:
Zynh Ludwig 2024-11-21 14:08:51 -08:00
parent ec6b1b6477
commit 07273fad14
3 changed files with 27 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
queries/records/get_and_update.sql
View file

@ -1,8 +1,6 @@
UPDATE record
UPDATE records
SET downloads = downloads + 1
WHERE cache_name = ?
RETURNING
cache_name,
uploaded,
downloads,
max_downloads;
WHERE
cache_name = ?
AND downloads < max_downloads
AND julianday('now') - julianday(uploaded) > 5;

4
src/lib.rs
View file

@ -8,3 +8,7 @@ pub mod views;
pub use router::*;
pub use state::*;
pub use views::*;
use std::{path::PathBuf, sync::LazyLock};
pub static CACHE_DIR: LazyLock<PathBuf> = LazyLock::new(|| PathBuf::from("./.cache/serve"));

34
src/router/download.rs
View file

@ -8,7 +8,7 @@ use axum::{
use reqwest::StatusCode;
use tokio_util::io::ReaderStream;
use crate::{AppState, AsyncRemoveRecord};
use crate::{AppState, CACHE_DIR};
pub fn get_download_router() -> Router<AppState> {
Router::new().route("/:id", get(download))
@ -18,23 +18,25 @@ async fn download(
axum::extract::Path(id): axum::extract::Path<String>,
State(state): State<AppState>,
) -> Result<axum::response::Response, (StatusCode, String)> {
{
let mut records = state.records.lock().await;
if let Some(record) = records
.get_mut(&id)
.filter(|record| record.can_be_downloaded())
{
record.downloads += 1;
let mut conn = state.pool.acquire().await.unwrap();
let file = tokio::fs::File::open(&record.file).await.unwrap();
let rows_affected = sqlx::query_file!("queries/records/get_and_update.sql", id)
.execute(&mut *conn)
.await
.map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?
.rows_affected();
return Ok(axum::response::Response::builder()
.header("Content-Type", "application/zip")
.body(Body::from_stream(ReaderStream::new(file)))
.unwrap());
} else {
records.remove_record(&id).await.unwrap()
}
drop(conn);
if rows_affected > 0 {
let file = tokio::fs::File::open(CACHE_DIR.join(id))
.await
.map_err(|err| (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()))?;
return Ok(axum::response::Response::builder()
.header("Content-Type", "application/zip")
.body(Body::from_stream(ReaderStream::new(file)))
.unwrap());
}
Ok(Redirect::to("/404.html").into_response())