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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

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

use crate::errors::Result;
use graphile_worker_job::Job;

use super::shared::{single_job_params, FailJobTables};

#[tracing::instrument(skip_all, err, fields(otel.kind="client", db.system="postgresql"))]
pub async fn fail_job(
    mut executor: impl DbExecutorArg,
    job: &Job,
    schema: impl Into<Schema>,
    worker_id: &str,
    message: &str,
    replacement_payload: Option<serde_json::Value>,
) -> Result<()> {
    let schema = schema.into();
    let tables = FailJobTables::new(&schema);
    let params = single_job_params(job, worker_id, message, replacement_payload);

    if job.job_queue_id().is_some() {
        return fail_queued_job(&mut executor, &tables, params).await;
    }

    fail_unqueued_job(&mut executor, &tables, params).await
}

async fn fail_unqueued_job(
    executor: &mut impl DbExecutorArg,
    tables: &FailJobTables,
    params: Vec<DbValue>,
) -> Result<()> {
    let jobs = &tables.jobs;
    let sql = formatdoc!(
        r#"
            update {jobs} as jobs
            set
                last_error = $2::text,
                run_at = greatest(now(), run_at) + (exp(least(attempts, 10)) * interval '1 second'),
                locked_by = null,
                locked_at = null,
                payload = coalesce($4::json, jobs.payload)
            where id = $1::bigint and locked_by = $3::text;
        "#
    );

    executor.execute(&sql, params.into()).await?;
    Ok(())
}

async fn fail_queued_job(
    executor: &mut impl DbExecutorArg,
    tables: &FailJobTables,
    params: Vec<DbValue>,
) -> Result<()> {
    let jobs = &tables.jobs;
    let job_queues = &tables.job_queues;
    let sql = formatdoc!(
        r#"
            with j as (
                update {jobs} as jobs
                set
                    last_error = $2::text,
                    run_at = greatest(now(), run_at) + (exp(least(attempts, 10)) * interval '1 second'),
                    locked_by = null,
                    locked_at = null,
                    payload = coalesce($4::json, jobs.payload)
                where id = $1::bigint and locked_by = $3::text
                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 = $3::text;
        "#
    );

    executor.execute(&sql, params.into()).await?;
    Ok(())
}