graphile_worker_queries 0.1.1

Database query helpers for graphile_worker
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

use graphile_worker_database::{DbExecutorArg, DbValue, Schema};
use indoc::formatdoc;

use crate::errors::GraphileWorkerError;

use graphile_worker_job::Job;

#[tracing::instrument(skip_all, err, fields(otel.kind="client", db.system="postgresql"))]
pub async fn complete_job(
    mut executor: impl DbExecutorArg,
    job: &Job,
    worker_id: &str,
    schema: impl Into<Schema>,
) -> Result<(), GraphileWorkerError> {
    let schema = schema.into();
    let jobs = schema.private_table("jobs");
    let job_queues = schema.private_table("job_queues");

    if job.job_queue_id().is_some() {
        let sql = formatdoc!(
            r#"
                with j as (
                    delete from {jobs} as jobs
                    where id = $1::bigint
                    returning *
                )
                update {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;
            "#
        );

        executor
            .execute(
                &sql,
                vec![
                    DbValue::I64(*job.id()),
                    DbValue::Text(worker_id.to_string()),
                ]
                .into(),
            )
            .await?;
    } else {
        let sql = format!(
            r#"
                delete from {jobs}
                where id = $1::bigint;
            "#
        );

        executor
            .execute(&sql, vec![DbValue::I64(*job.id())].into())
            .await?;
    }

    Ok(())
}