adrift_core 0.0.11

Adrift a web 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

pub use job_queue::*;
use dotenv::dotenv;
use std::env;
use std::sync::OnceLock;

static QUEUE: OnceLock<Client> = OnceLock::new();

pub struct Queue;

impl Queue {
    pub async fn dispatch(task: &impl Job) -> anyhow::Result<()> {
        let queue = get_queue().await?;

        queue.dispatch(task).await?;

        Ok(())
    }
}

async fn get_queue() -> anyhow::Result<Client> {
    match QUEUE.get() {
        Some(client) => Ok(client.clone()),
        None => {
            dotenv().ok();

            let database_url = env::var("DATABASE_URL")
                .map_err(|_| anyhow::anyhow!("Failed to get DATABASE_URL"))?;

            let queue = Client::builder()
                .connect(&database_url)
                .await
                .map_err(|_| anyhow::anyhow!("Failed to connect to database"))?;

            let _ = QUEUE.set(queue);

            match QUEUE.get() {
                Some(client) => Ok(client.clone()),
                None => Err(anyhow::anyhow!("Failed to get queue")),
            }
        }
    }
}