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

use graphile_worker_database::Schema;

pub fn get_flag_clause(flags_to_skip: &[String], param_ord: u8) -> String {
    if !flags_to_skip.is_empty() {
        return format!("and ((flags ?| ${param_ord}::text[]) is not true)");
    }
    String::new()
}

pub fn get_queue_clause(schema: &Schema) -> String {
    let job_queues = schema.private_table("job_queues");
    format!(
        r#"
            and (
                jobs.job_queue_id is null
                or
                jobs.job_queue_id in (
                    select id
                    from {job_queues} as job_queues
                    where job_queues.is_available = true
                    for update
                    skip locked
                )
            )
        "#
    )
}

pub fn get_update_queue_clause(
    schema: &Schema,
    worker_id_param: u8,
    now_param: Option<u8>,
) -> String {
    let locked_at = get_now_clause(now_param);
    let job_queues = schema.private_table("job_queues");
    format!(
        r#",
            q as (
                update {job_queues} as job_queues
                    set
                        locked_by = ${worker_id_param},
                        locked_at = {locked_at}
                from j
                where job_queues.id = j.job_queue_id
            )
        "#
    )
}

pub fn get_now_clause(now_param: Option<u8>) -> String {
    now_param
        .map(|p| format!("${p}::timestamptz"))
        .unwrap_or_else(|| "now()".to_string())
}