40 lines
1.1 KiB
Rust
40 lines
1.1 KiB
Rust
use background_jobs_core::{JobInfo, NewJobInfo, ReturnJobInfo, Stats, Storage};
|
|
use failure::{Error, Fail};
|
|
|
|
pub(crate) trait ActixStorage {
|
|
fn new_job(&mut self, job: NewJobInfo) -> Result<u64, Error>;
|
|
|
|
fn request_job(&mut self, queue: &str, runner_id: u64) -> Result<Option<JobInfo>, Error>;
|
|
|
|
fn return_job(&mut self, ret: ReturnJobInfo) -> Result<(), Error>;
|
|
|
|
fn get_stats(&self) -> Result<Stats, Error>;
|
|
}
|
|
|
|
pub(crate) struct StorageWrapper<S>(pub(crate) S)
|
|
where
|
|
S: Storage,
|
|
S::Error: Fail;
|
|
|
|
impl<S> ActixStorage for StorageWrapper<S>
|
|
where
|
|
S: Storage,
|
|
S::Error: Fail,
|
|
{
|
|
fn new_job(&mut self, job: NewJobInfo) -> Result<u64, Error> {
|
|
self.0.new_job(job).map_err(Error::from)
|
|
}
|
|
|
|
fn request_job(&mut self, queue: &str, runner_id: u64) -> Result<Option<JobInfo>, Error> {
|
|
self.0.request_job(queue, runner_id).map_err(Error::from)
|
|
}
|
|
|
|
fn return_job(&mut self, ret: ReturnJobInfo) -> Result<(), Error> {
|
|
self.0.return_job(ret).map_err(Error::from)
|
|
}
|
|
|
|
fn get_stats(&self) -> Result<Stats, Error> {
|
|
self.0.get_stats().map_err(Error::from)
|
|
}
|
|
}
|