refactor(lib_core): task manager structure

refs: #5
This commit is contained in:
Alexander Navarro 2025-05-16 20:00:25 -04:00
parent 4199a97a19
commit 45a3bf291b
7 changed files with 176 additions and 160 deletions

View file

@ -0,0 +1,85 @@
use chrono::Utc;
use futures::StreamExt;
use serde::de::DeserializeOwned;
use serde::Serialize;
use std::fmt::Display;
use tabled::Tabled;
mod manager;
#[derive(sqlx::Type, Debug, Clone)]
#[repr(u8)]
pub enum TaskStatus {
Pending = 1,
InProgress = 2,
Completed = 3,
Failed = 4,
}
impl Display for TaskStatus {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
TaskStatus::Pending => {
write!(f, "Pending")
}
TaskStatus::InProgress => {
write!(f, "In Progress")
}
TaskStatus::Completed => {
write!(f, "Completed")
}
TaskStatus::Failed => {
write!(f, "Failed")
}
}
}
}
pub trait TaskPayloadKey {
fn get_key(&self) -> String;
}
pub trait TaskPayload:
Serialize + DeserializeOwned + Send + Unpin + 'static + Display + TaskPayloadKey
{
}
impl<T: Serialize + DeserializeOwned + Send + Unpin + 'static + Display + TaskPayloadKey>
TaskPayload for T
{
}
pub type TaskJob<T> = fn(&Task<T>) -> TaskStatus;
#[derive(sqlx::FromRow, Tabled, Debug)]
pub struct Task<T: TaskPayload> {
id: u32,
payload_key: String,
#[sqlx(json)]
#[tabled(skip)]
payload: T,
#[sqlx(rename = "status_id")]
status: TaskStatus,
created_at: chrono::DateTime<Utc>,
#[tabled(display = "display_option_date")]
updated_at: Option<chrono::DateTime<Utc>>,
}
impl<T: TaskPayload> Task<T> {
pub fn payload(&self) -> &T {
&self.payload
}
}
impl<T: TaskPayload> Task<T> {
pub fn get_key(&self) -> String {
self.payload_key.clone()
}
}
fn display_option_date(o: &Option<chrono::DateTime<Utc>>) -> String {
match o {
Some(s) => format!("{}", s),
None => String::from(""),
}
}