graphile_worker 0.13.2

High performance Rust/PostgreSQL job queue (also suitable for getting jobs generated by PostgreSQL triggers/functions out into a different work queue)
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

use graphile_worker_database::{Database, Schema};

use super::CompletionRequest;

mod queued;
mod unqueued;

use queued::{complete_jobs_with_queues, complete_one_job_with_queue};
use unqueued::{complete_jobs_without_queues, complete_one_job_without_queue};

pub(super) struct CompletionBatchResult {
    with_queue_succeeded: bool,
    without_queue_succeeded: bool,
}

impl CompletionBatchResult {
    pub(super) fn persisted(&self, req: &CompletionRequest) -> bool {
        (req.has_queue && self.with_queue_succeeded)
            || (!req.has_queue && self.without_queue_succeeded)
    }
}

pub(super) async fn complete_batch(
    batch: &[CompletionRequest],
    database: &Database,
    schema: &Schema,
    worker_id: &str,
) -> CompletionBatchResult {
    let mut with_queue_ids = Vec::new();
    let mut without_queue_ids = Vec::with_capacity(batch.len());

    for req in batch {
        if req.has_queue {
            with_queue_ids.push(req.job_id);
        } else {
            without_queue_ids.push(req.job_id);
        }
    }

    CompletionBatchResult {
        with_queue_succeeded: complete_jobs_with_queues(
            database,
            schema,
            worker_id,
            with_queue_ids,
        )
        .await,
        without_queue_succeeded: complete_jobs_without_queues(database, schema, without_queue_ids)
            .await,
    }
}

pub(super) async fn complete_job_direct(
    req: &CompletionRequest,
    database: &Database,
    schema: &Schema,
    worker_id: &str,
) -> bool {
    if req.has_queue {
        return complete_one_job_with_queue(req, database, schema, worker_id).await;
    }

    complete_one_job_without_queue(req, database, schema).await
}