bonsaimq 0.3.0

Message/job queue based on bonsaidb, similar to sqlxmq.
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

//! Functions for the library users.

use std::time::Duration;

use bonsaidb::core::{connection::AsyncConnection, schema::SerializedCollection};

use crate::{
	queue::{Id, Message},
	Error,
};

/// Check whether the job with the given ID exists. Can also be used to check if
/// a job has finished already.
pub async fn job_exists<DB: AsyncConnection>(id: Id, db: &DB) -> Result<bool, Error> {
	Ok(Message::get_async(&id, db).await?.is_some())
}

/// Wait for a job to finish using a fixed interval to check if it exists.
pub async fn await_job<DB: AsyncConnection>(
	id: Id,
	interval_ms: u64,
	db: &DB,
) -> Result<(), Error> {
	let mut int = tokio::time::interval(Duration::from_millis(interval_ms));
	loop {
		if !job_exists(id, db).await? {
			return Ok(());
		}
		int.tick().await;
	}
}