oxanus 1.1.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94

use crate::shared::{WorkerError, WorkerState as WorkerContext, random_string, setup};
use deadpool_redis::redis::AsyncCommands;
use serde::{Deserialize, Serialize};
use testresult::TestResult;

#[derive(oxanus::Registry)]
struct ComponentRegistry(oxanus::ComponentRegistry<WorkerContext, WorkerError>);

#[derive(Serialize, oxanus::Queue)]
#[oxanus(key = "two")]
struct QueueTwo;

#[derive(Debug, Serialize, Deserialize)]
pub struct WorkerCounterJob {
    pub key: String,
}

#[derive(oxanus::Worker)]
pub struct WorkerCounter {
    ctx: WorkerContext,
}

impl WorkerCounter {
    async fn process(
        &self,
        job: &WorkerCounterJob,
        _ctx: &oxanus::JobContext,
    ) -> Result<(), WorkerError> {
        let mut redis = self.ctx.redis.get().await?;
        let _: () = redis.incr(&job.key, 1).await?;
        Ok(())
    }
}

#[derive(Debug, Serialize, Deserialize)]
pub struct CronWorkerCounterJob {}

#[derive(oxanus::Worker)]
#[oxanus(cron(schedule = "* * * * * *", queue = QueueTwo))]
pub struct CronWorkerCounter {
    ctx: WorkerContext,
}

impl CronWorkerCounter {
    async fn process(
        &self,
        _job: &CronWorkerCounterJob,
        _ctx: &oxanus::JobContext,
    ) -> Result<(), WorkerError> {
        let mut redis = self.ctx.redis.get().await?;
        let _: () = redis.incr("test_worker:counter", 1).await?;
        Ok(())
    }
}

#[tokio::test]
pub async fn test_registry() -> TestResult {
    let redis_pool = setup();
    let mut redis_conn = redis_pool.get().await?;
    let _: i64 = redis_conn.del("test_worker:counter").await?;

    let ctx = oxanus::ContextValue::new(WorkerContext {
        redis: redis_pool.clone(),
    });

    let storage = oxanus::Storage::builder()
        .namespace(random_string())
        .build_from_pool(redis_pool.clone())?;

    let config = ComponentRegistry::build_config(&storage).exit_when_processed(2);

    // no need to manually register, here we verify they were registered
    assert!(config.has_registered_queue::<QueueTwo>());
    assert!(config.has_registered_worker_type::<WorkerCounter>());
    assert!(config.has_registered_cron_worker_type::<CronWorkerCounter>());

    storage
        .enqueue(
            QueueTwo,
            WorkerCounterJob {
                key: "test_worker:counter".to_owned(),
            },
        )
        .await?;

    oxanus::run(config, ctx).await?;

    let mut redis_conn = redis_pool.get().await?;
    let value: Option<i64> = redis_conn.get("test_worker:counter").await?;

    assert_eq!(value, Some(2));

    Ok(())
}