discours.io/quoter
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handle-accurater
All checks were successful
deploy / deploy (push) Successful in 1m0s
Details

Browse Source
This commit is contained in:
Untone 2024-10-23 17:46:30 +03:00
parent b0d7162460
commit 0d4f21c79b
1 changed files with 44 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
src/handlers/proxy.rs
View File

@ -1,3 +1,4 @@
use actix_web::error::ErrorNotFound;
use actix_web::{error::ErrorInternalServerError, web, HttpRequest, HttpResponse, Result}; use actix_web::{error::ErrorInternalServerError, web, HttpRequest, HttpResponse, Result};
use log::{error, warn}; use log::{error, warn};
@ -142,46 +143,52 @@ pub async fn proxy_handler(
let ct_parts = content_type.split("/").collect::<Vec<&str>>(); let ct_parts = content_type.split("/").collect::<Vec<&str>>();
let filepath = format!("production/{}/{}.{}", ct_parts[0], base_filename, extension); // NOTE: original ext let filepath = format!("production/{}/{}.{}", ct_parts[0], base_filename, extension); // NOTE: original ext
match check_file_exists(&state.storj_client, &state.bucket, &filepath).await? { let exists_in_storj = check_file_exists(&state.storj_client, &state.bucket, &filepath).await?;
true => {
warn!("file {} exists in storj", filepath);
}
false => {
warn!("file {} does not exist in storj", filepath);
}
}
match check_file_exists(&state.aws_client, &state.bucket, &filepath).await? { if exists_in_storj {
true => { warn!("file {} exists in storj, try to generate thumbnails", filepath);
warn!("file {} exists in aws", filepath);
}
false => {
warn!("file {} does not exist in aws", filepath);
}
}
match load_file_from_s3(&state.aws_client, &state.bucket, &filepath).await { match load_file_from_s3(&state.aws_client, &state.bucket, &filepath).await {
Ok(filedata) => { Ok(filedata) => {
let _ = thumbdata_save(filedata.clone(), &state, &filekey, content_type.to_string()) let _ = thumbdata_save(filedata.clone(), &state, &filekey, content_type.to_string()).await;
.await;
if let Err(e) = upload_to_s3(
&state.storj_client,
&state.bucket,
&filekey,
filedata.clone(),
content_type,
)
.await {
warn!("cannot upload to storj: {}", e);
} else {
warn!("file {} uploaded to storj", filekey);
} }
Ok(HttpResponse::Ok().content_type(content_type).body(filedata)) Err(e) => {
}, error!("cannot download {} from storj: {}", filekey, e);
Err(e) => { return Err(ErrorInternalServerError(e));
error!("cannot download {} from aws: {}", filepath, e); }
Err(ErrorInternalServerError(e)) }
}, } else {
warn!("file {} does not exist in storj", filepath);
}
let exists_in_aws = check_file_exists(&state.aws_client, &state.bucket, &filepath).await?;
if exists_in_aws {
match load_file_from_s3(&state.aws_client, &state.bucket, &filepath).await {
Ok(filedata) => {
let _ = thumbdata_save(filedata.clone(), &state, &filekey, content_type.to_string())
.await;
if let Err(e) = upload_to_s3(
&state.storj_client,
&state.bucket,
&filekey,
filedata.clone(),
content_type,
)
.await {
warn!("cannot upload to storj: {}", e);
} else {
warn!("file {} uploaded to storj", filekey);
}
Ok(HttpResponse::Ok().content_type(content_type).body(filedata))
},
Err(e) => {
error!("cannot download {} from aws: {}", filepath, e);
Err(ErrorInternalServerError(e))
},
}
} else {
error!("file {} does not exist in aws", filepath);
Err(ErrorNotFound("file does not exist"))
} }
}, },
Err(e) => { Err(e) => {