graphile_worker 0.11.4

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

use indoc::formatdoc;
use sqlx::{query, PgExecutor};

use crate::errors::GraphileWorkerError;

use crate::Job;

#[tracing::instrument(skip_all, err, fields(otel.kind="client", db.system="postgresql"))]
pub async fn complete_job(
    executor: impl for<'e> PgExecutor<'e>,
    job: &Job,
    worker_id: &str,
    escaped_schema: &str,
) -> Result<(), GraphileWorkerError> {
    if job.job_queue_id().is_some() {
        let sql = formatdoc!(
            r#"
                with j as (
                    delete from {escaped_schema}._private_jobs as jobs
                    where id = $1::bigint
                    returning *
                )
                update {escaped_schema}._private_job_queues as job_queues
                    set locked_by = null, locked_at = null
                    from j
                    where job_queues.id = j.job_queue_id and job_queues.locked_by = $2::text;
            "#
        );

        query(&sql)
            .bind(job.id())
            .bind(worker_id)
            .execute(executor)
            .await?;
    } else {
        let sql = format!(
            r#"
                delete from {escaped_schema}._private_jobs
                where id = $1::bigint;
            "#
        );

        query(&sql).bind(job.id()).execute(executor).await?;
    }

    Ok(())
}