1use rumtk_core::core::RUMResult;
22use rumtk_core::id::id_to_uuid;
23use rumtk_core::strings::rumtk_format;
24use rumtk_core::strings::RUMString;
25use rumtk_core::threading::threading_manager::{Task, TaskID, TaskManager};
26use rumtk_core::types::RUMBuffer;
27
28pub type JobID = TaskID;
29pub type JobBuffer = RUMBuffer;
30
31#[derive(Default, Debug, Clone)]
32pub enum JobResultType<T = RUMString> {
33 File(JobBuffer),
34 JSON(RUMString),
35 TEXT(RUMString),
36 RustType(T),
37 #[default]
38 NONE,
39}
40
41pub type JobResult = RUMResult<JobResultType>;
42pub type Job = Task<JobResult>;
43type JobManager = TaskManager<JobResult>;
44
45static mut TASK_MANAGER: Option<JobManager> = None;
46
47pub fn job_str_id_to_id(id: &str) -> JobID {
48 id_to_uuid(id)
49}
50
51pub fn init_job_manager(workers: &usize) -> RUMResult<()> {
52 let manager = TaskManager::<JobResult>::new(workers)?;
53 unsafe {
54 TASK_MANAGER = Some(manager);
55 }
56 Ok(())
57}
58
59pub fn get_manager() -> RUMResult<&'static mut JobManager> {
60 unsafe {
61 match TASK_MANAGER.as_mut() {
62 Some(m) => Ok(m),
63 None => return Err(rumtk_format!("TaskManager is not initialized")),
64 }
65 }
66}
67
68#[macro_export]
69macro_rules! rumtk_web_init_job_manager {
70 ( $workers:expr ) => {{
71 use $crate::jobs::init_job_manager;
72 init_job_manager($workers)
73 }};
74}
75
76#[macro_export]
77macro_rules! rumtk_web_get_job_manager {
78 ( ) => {{
79 use $crate::jobs::get_manager;
80 get_manager()
81 }};
82}
83
84#[macro_export]
85macro_rules! rumtk_web_generate_job_id {
86 ( $id:expr ) => {{
87 use $crate::jobs::job_str_id_to_id;
88 job_str_id_to_id($id)
89 }};
90}