broccoli_queue 0.4.6

Broccoli is a simple, fast, and reliable job queue 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

use broccoli_queue::queue::BroccoliQueue;

pub async fn setup_queue() -> BroccoliQueue {
    let queue_url = std::env::var("BROCCOLI_QUEUE_URL").unwrap();

    BroccoliQueue::builder(queue_url)
        .pool_connections(5)
        .enable_scheduling(true)
        .build()
        .await
        .expect("Queue setup failed. Are you sure Redis/RabbitMQ/SurrealDB is running?")
}

pub async fn setup_queue_with_url(
    url: &str,
) -> Result<BroccoliQueue, broccoli_queue::error::BroccoliError> {
    BroccoliQueue::builder(url)
        .pool_connections(5)
        .build()
        .await
}

#[cfg(feature = "redis")]
pub async fn get_redis_client() -> redis::aio::MultiplexedConnection {
    let queue_url = std::env::var("BROCCOLI_QUEUE_URL").unwrap();
    let client = redis::Client::open(queue_url).unwrap();
    client.get_multiplexed_async_connection().await.unwrap()
}

#[cfg(all(feature = "surrealdb", test))]
pub async fn get_surrealdb_client() -> surrealdb::Surreal<surrealdb::engine::any::Any> {
    let queue_url = std::env::var("BROCCOLI_QUEUE_URL").unwrap();
    let db = broccoli_queue::brokers::surrealdb::utils::client_from_url(&queue_url)
        .await
        .unwrap()
        .unwrap();
    db
}