Updated the dashboard page
This commit is contained in:
22
src/tasks.rs
22
src/tasks.rs
@@ -4,11 +4,19 @@ use std::sync::Mutex;
|
||||
use chrono::{NaiveDateTime, Utc};
|
||||
use serde::Serialize;
|
||||
|
||||
#[derive(Debug, Clone, Serialize)]
|
||||
pub struct TaskProgress {
|
||||
pub current: u64,
|
||||
pub total: u64,
|
||||
pub message: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone, Serialize)]
|
||||
pub struct TaskInfo {
|
||||
pub id: String,
|
||||
pub task_type: String,
|
||||
pub status: TaskStatus,
|
||||
pub progress: Option<TaskProgress>,
|
||||
pub started_at: NaiveDateTime,
|
||||
pub completed_at: Option<NaiveDateTime>,
|
||||
pub result: Option<String>,
|
||||
@@ -39,6 +47,7 @@ impl TaskManager {
|
||||
id: id.clone(),
|
||||
task_type: task_type.to_string(),
|
||||
status: TaskStatus::Running,
|
||||
progress: None,
|
||||
started_at: Utc::now().naive_utc(),
|
||||
completed_at: None,
|
||||
result: None,
|
||||
@@ -47,10 +56,22 @@ impl TaskManager {
|
||||
id
|
||||
}
|
||||
|
||||
/// Update progress on a running task.
|
||||
pub fn update_progress(&self, id: &str, current: u64, total: u64, message: &str) {
|
||||
if let Some(task) = self.tasks.lock().unwrap().get_mut(id) {
|
||||
task.progress = Some(TaskProgress {
|
||||
current,
|
||||
total,
|
||||
message: message.to_string(),
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
/// Mark a task as completed with a result string.
|
||||
pub fn complete(&self, id: &str, result: String) {
|
||||
if let Some(task) = self.tasks.lock().unwrap().get_mut(id) {
|
||||
task.status = TaskStatus::Completed;
|
||||
task.progress = None;
|
||||
task.completed_at = Some(Utc::now().naive_utc());
|
||||
task.result = Some(result);
|
||||
}
|
||||
@@ -60,6 +81,7 @@ impl TaskManager {
|
||||
pub fn fail(&self, id: &str, error: String) {
|
||||
if let Some(task) = self.tasks.lock().unwrap().get_mut(id) {
|
||||
task.status = TaskStatus::Failed;
|
||||
task.progress = None;
|
||||
task.completed_at = Some(Utc::now().naive_utc());
|
||||
task.result = Some(error);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user