oxana 2.0.1

A simple & fast job queue system.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53

use crate::{QueueConfig, RuntimeBuilder, worker_registry::WorkerConfig};

pub struct ComponentRegistry<DT> {
    /// `module_path!()`
    pub module_path: &'static str,
    /// `stringify!(MyStruct)`
    pub type_name: &'static str,
    pub definition: fn() -> ComponentDefinition<DT>,
}

pub enum ComponentDefinition<DT> {
    Queue(QueueConfig),
    Worker(WorkerConfig<DT>),
}

pub trait RegisterComponents {
    type Context: Clone + Send + Sync + 'static;

    fn register_components(runtime: RuntimeBuilder<Self::Context>)
    -> RuntimeBuilder<Self::Context>;
}

/// Macro to create a component registry
pub use inventory::collect as create_component_registry;

/// Macro to register a Queue or Worker
pub use inventory::submit as register_component;

/// Helper type to iterate components
pub use inventory::iter as iterate_components;

impl<DT> ComponentRegistry<DT>
where
    DT: Clone + Send + Sync + 'static,
{
    pub fn register_components(
        mut runtime: RuntimeBuilder<DT>,
        items: impl Iterator<Item = &'static Self>,
    ) -> RuntimeBuilder<DT> {
        for component in items {
            tracing::info!(
                "Registering {}::{}",
                component.module_path,
                component.type_name
            );
            match (component.definition)() {
                ComponentDefinition::Queue(q) => runtime = runtime.queue_with(q),
                ComponentDefinition::Worker(w) => runtime = runtime.worker_with(w),
            }
        }
        runtime
    }
}