2023-03-11 15:38:32 +00:00
|
|
|
use crate::errors::BackieError;
|
2023-03-10 22:41:34 +00:00
|
|
|
use crate::runnable::BackgroundTask;
|
2023-03-11 16:49:23 +00:00
|
|
|
use crate::store::TaskStore;
|
|
|
|
use crate::task::NewTask;
|
2023-03-10 22:41:34 +00:00
|
|
|
use std::sync::Arc;
|
|
|
|
use std::time::Duration;
|
2023-03-04 18:07:17 +00:00
|
|
|
|
2023-03-10 22:41:34 +00:00
|
|
|
#[derive(Clone)]
|
2023-03-11 16:49:23 +00:00
|
|
|
pub struct Queue<S>
|
|
|
|
where
|
2023-03-13 12:08:54 +00:00
|
|
|
S: TaskStore + Clone,
|
2023-03-11 16:49:23 +00:00
|
|
|
{
|
|
|
|
task_store: Arc<S>,
|
2023-03-10 22:41:34 +00:00
|
|
|
}
|
2023-03-04 18:07:17 +00:00
|
|
|
|
2023-03-11 16:49:23 +00:00
|
|
|
impl<S> Queue<S>
|
|
|
|
where
|
2023-03-13 12:08:54 +00:00
|
|
|
S: TaskStore + Clone,
|
2023-03-11 16:49:23 +00:00
|
|
|
{
|
2023-03-13 13:11:19 +00:00
|
|
|
pub fn new(task_store: S) -> Self {
|
|
|
|
Queue {
|
|
|
|
task_store: Arc::new(task_store),
|
|
|
|
}
|
2023-03-10 22:41:34 +00:00
|
|
|
}
|
2023-03-04 18:07:17 +00:00
|
|
|
|
2023-03-11 15:38:32 +00:00
|
|
|
pub async fn enqueue<BT>(&self, background_task: BT) -> Result<(), BackieError>
|
2023-03-10 22:41:34 +00:00
|
|
|
where
|
|
|
|
BT: BackgroundTask,
|
|
|
|
{
|
|
|
|
self.task_store
|
|
|
|
.create_task(NewTask::new(background_task, Duration::from_secs(10))?)
|
|
|
|
.await?;
|
|
|
|
Ok(())
|
|
|
|
}
|
2023-03-04 18:07:17 +00:00
|
|
|
}
|