loco-rs 0.16.1

The one-person framework for Rust
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
82
83
84
85

#[cfg(any(feature = "bg_pg", feature = "bg_sqlt"))]
use crate::bgworker;
use std::path::PathBuf;

#[cfg(feature = "bg_pg")]
/// # Panics
///
/// This function will panic if it fails to prepare or insert the seed data, causing the tests to fail quickly
/// and preventing further test execution with incomplete setup.
pub async fn postgres_seed_data(pool: &sqlx::PgPool) {
    let yaml_tasks = std::fs::read_to_string(
        PathBuf::from("tests")
            .join("fixtures")
            .join("queue")
            .join("jobs.yaml"),
    )
    .expect("Failed to read YAML file");

    let tasks: Vec<bgworker::pg::Job> =
        serde_yaml::from_str(&yaml_tasks).expect("Failed to parse YAML");
    for task in tasks {
        sqlx::query(
            r"
            INSERT INTO pg_loco_queue (id, name, task_data, status, run_at, interval, created_at, updated_at)
            VALUES ($1, $2, $3, $4, $5, NULL, $6, $7)
            ",
        )
        .bind(task.id)
        .bind(task.name)
        .bind(task.data)
        .bind(task.status.to_string())
        .bind(task.run_at)
        .bind(task.created_at)
        .bind(task.updated_at)
        .execute(pool)
        .await.expect("execute insert query");
    }
}

#[cfg(feature = "bg_sqlt")]
/// # Panics
///
/// This function will panic if it fails to prepare or insert the seed data, causing the tests to fail quickly
/// and preventing further test execution with incomplete setup.
pub async fn sqlite_seed_data(pool: &sqlx::Pool<sqlx::Sqlite>) {
    let yaml_tasks = std::fs::read_to_string(
        PathBuf::from("tests")
            .join("fixtures")
            .join("queue")
            .join("jobs.yaml"),
    )
    .expect("Failed to read YAML file");

    let tasks: Vec<bgworker::sqlt::Job> =
        serde_yaml::from_str(&yaml_tasks).expect("Failed to parse YAML");
    for task in tasks {
        sqlx::query(
            r"
            INSERT INTO sqlt_loco_queue (id, name, task_data, status, run_at, interval, created_at, updated_at)
            VALUES (?, ?, ?, ?, ?, NULL, ?, ?)
            "
        )
        .bind(task.id)
        .bind(task.name)
        .bind(task.data.to_string())
        .bind(task.status.to_string())
        .bind(task.run_at)
        .bind(task.created_at)
        .bind(task.updated_at)
        .execute(pool)
        .await.expect("create row");
    }

    sqlx::query(
        r"
                INSERT INTO sqlt_loco_queue_lock (id, is_locked, locked_at)
    VALUES (1, FALSE, NULL)
    ON CONFLICT (id) DO NOTHING;

            ",
    )
    .execute(pool)
    .await
    .expect("execute insert query");
}