apalis-sqlite 1.0.0-rc.9

Background task processing for rust using apalis and sqlite
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

use apalis_core::{backend::WorkerFilter, task::task_id::coalesce_ids};
use sqlx::Executor;

/// Re-enqueue tasks that were being processed by dead workers
///
/// A worker that has not sent a keep-alive signal within the heartbeat duration is considered dead
pub async fn reenqueue_orphaned<'a, E>(
    executor: E,
    dead_for: i64,
    queue: &str,
    filter: &WorkerFilter,
) -> Result<u64, sqlx::Error>
where
    E: Executor<'a, Database = sqlx::Sqlite>,
{
    let (exclude_id, only_id) = match filter {
        WorkerFilter::AllExcept(id) => (Some(id), None),
        WorkerFilter::Only(id) => (None, Some(id)),
        WorkerFilter::None => (None, None),
        _ => unreachable!(),
    };
    match sqlx::query_file!(
        "queries/backend/reenqueue_orphaned.sql",
        dead_for,
        queue,
        exclude_id,
        only_id,
    )
    .execute(executor)
    .await
    {
        Ok(res) => Ok(res.rows_affected()),
        Err(e) => Err(e),
    }
}

/// Re-enqueue tasks that were being processed by dying a worker
///
/// This will be invoked during `Backend::poll_close`
pub async fn reenqueue_abandoned<'a, E>(
    executor: E,
    queue: &str,
    worker: &str,
    task_ids: &Vec<String>,
) -> Result<u64, sqlx::Error>
where
    E: Executor<'a, Database = sqlx::Sqlite>,
{
    let task_ids = coalesce_ids(task_ids);
    match sqlx::query_file!(
        "queries/backend/reenqueue_abandoned.sql",
        queue,
        worker,
        task_ids
    )
    .execute(executor)
    .await
    {
        Ok(res) => Ok(res.rows_affected()),
        Err(e) => Err(e),
    }
}