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

use apalis_core::worker::context::WorkerContext;
use sqlx::Executor;

use crate::config::Config;

/// Register a new worker in the database
pub async fn register_worker<'a, E>(
    executor: E,
    config: &Config,
    worker: &WorkerContext,
    storage_type: &str,
) -> Result<(), sqlx::Error>
where
    E: Executor<'a, Database = sqlx::Sqlite>,
{
    let worker_id = worker.name().to_owned();
    let queue = config.queue.to_string();
    let layers = worker.get_service().to_owned();
    let keep_alive = config.heartbeat_interval.as_secs() as i64;
    let res = sqlx::query_file!(
        "queries/backend/register_worker.sql",
        worker_id,
        queue,
        storage_type,
        layers,
        keep_alive,
    )
    .execute(executor)
    .await?;
    if res.rows_affected() == 0 {
        return Err(sqlx::Error::Io(std::io::Error::new(
            std::io::ErrorKind::AlreadyExists,
            "WORKER_ALREADY_EXISTS",
        )));
    }
    Ok(())
}