flibusta-apps/batch_downloader
1
0
Fork 0
mirror of https://github.com/flibusta-apps/batch_downloader.git synced 2025-12-06 14:25:36 +01:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-08-06 22:53:09 +02:00
parent e52cf6a976
commit baf0605ca7
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/structures.rs
View File

@@ -3,7 +3,7 @@ use smallvec::SmallVec;
use smartstring::alias::String as SmartString; use smartstring::alias::String as SmartString;
#[derive(Serialize, Clone)] #[derive(Serialize, Clone, PartialEq)]
#[serde(rename_all = "snake_case")] #[serde(rename_all = "snake_case")]
pub enum TaskStatus { pub enum TaskStatus {
InProgress, InProgress,

10
src/views.rs
View File

@@ -8,7 +8,7 @@ use tower_http::trace::{TraceLayer, self};
use tracing::Level; use tracing::Level;
use crate::{config::CONFIG, structures::{Task, CreateTask}, services::{task_creator::create_task, utils::get_key}}; use crate::{config::CONFIG, structures::{Task, CreateTask, TaskStatus}, services::{task_creator::create_task, utils::get_key}};
pub static TASK_RESULTS: Lazy<Cache<String, Task>> = Lazy::new(|| { pub static TASK_RESULTS: Lazy<Cache<String, Task>> = Lazy::new(|| {
@@ -25,7 +25,13 @@ async fn create_archive_task(
let key = get_key(data.clone()); let key = get_key(data.clone());
let result = match TASK_RESULTS.get(&key) { let result = match TASK_RESULTS.get(&key) {
Some(result) => result, Some(result) => {
if result.status == TaskStatus::Failled {
create_task(data).await
} else {
result
}
},
None => create_task(data).await, None => create_task(data).await,
}; };