qml_rs/processing/
mod.rs

1//! Job Processing Engine
2//!
3//! This module contains the job processing engine that handles job execution,
4//! worker management, and background job processing.
5
6use async_trait::async_trait;
7use std::collections::HashMap;
8
9use crate::core::Job;
10use crate::error::Result;
11
12pub mod activator;
13pub mod processor;
14pub mod retry;
15pub mod scheduler;
16pub mod server;
17pub mod worker;
18
19pub use activator::JobActivator;
20pub use processor::JobProcessor;
21pub use retry::{RetryPolicy, RetryStrategy};
22pub use scheduler::JobScheduler;
23pub use server::{BackgroundJobServer, ServerConfig};
24pub use worker::{WorkerConfig, WorkerContext, WorkerResult};
25
26/// Trait for executing jobs
27///
28/// Implementations of this trait define how specific job types are executed.
29/// The worker receives the job and its arguments, and returns a result.
30#[async_trait]
31pub trait Worker: Send + Sync {
32    /// Execute a job with the given arguments
33    ///
34    /// # Arguments
35    /// * `job` - The job to execute
36    /// * `context` - Worker context with execution information
37    ///
38    /// # Returns
39    /// * `Ok(WorkerResult)` if the job was executed successfully
40    /// * `Err(QmlError)` if there was an error executing the job
41    async fn execute(&self, job: &Job, context: &WorkerContext) -> Result<WorkerResult>;
42
43    /// Get the method name this worker handles
44    fn method_name(&self) -> &str;
45
46    /// Check if this worker can handle the given job method
47    fn can_handle(&self, method: &str) -> bool {
48        self.method_name() == method
49    }
50}
51
52/// Registry for job workers
53///
54/// This registry maps job method names to their corresponding worker implementations.
55/// It's used by the job processor to find the appropriate worker for each job.
56#[derive(Default)]
57pub struct WorkerRegistry {
58    workers: HashMap<String, Box<dyn Worker>>,
59}
60
61impl WorkerRegistry {
62    /// Create a new worker registry
63    pub fn new() -> Self {
64        Self {
65            workers: HashMap::new(),
66        }
67    }
68
69    /// Register a worker for a specific method
70    pub fn register<W>(&mut self, worker: W)
71    where
72        W: Worker + 'static,
73    {
74        let method_name = worker.method_name().to_string();
75        self.workers.insert(method_name, Box::new(worker));
76    }
77
78    /// Get a worker for the given method name
79    pub fn get_worker(&self, method: &str) -> Option<&dyn Worker> {
80        self.workers.get(method).map(|w| w.as_ref())
81    }
82
83    /// Get all registered method names
84    pub fn get_methods(&self) -> Vec<&str> {
85        self.workers.keys().map(|s| s.as_str()).collect()
86    }
87
88    /// Check if a method is registered
89    pub fn has_worker(&self, method: &str) -> bool {
90        self.workers.contains_key(method)
91    }
92
93    /// Get the number of registered workers
94    pub fn len(&self) -> usize {
95        self.workers.len()
96    }
97
98    /// Check if the registry is empty
99    pub fn is_empty(&self) -> bool {
100        self.workers.is_empty()
101    }
102}
qml_rs/processing/mod.rs

qml_rs/processing/
mod.rs
