mirror of
https://git.asonix.dog/asonix/background-jobs.git
synced 2024-11-24 13:01:00 +00:00
Clean a bit
This commit is contained in:
parent
17fefd0224
commit
60569fc384
1 changed files with 52 additions and 30 deletions
56
src/lib.rs
56
src/lib.rs
|
@ -77,12 +77,24 @@ impl Jobs {
|
|||
}
|
||||
}
|
||||
|
||||
impl Default for Jobs {
|
||||
fn default() -> Self {
|
||||
Jobs {
|
||||
inner: Default::default(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Processors {
|
||||
inner: HashMap<String, Box<Processor + Send + Sync + 'static>>,
|
||||
jobs: Option<Jobs>,
|
||||
}
|
||||
|
||||
impl Processors {
|
||||
pub fn new() -> Self {
|
||||
Default::default()
|
||||
}
|
||||
|
||||
pub fn register_processor<P>(&mut self, processor: P)
|
||||
where
|
||||
P: Processor + Send + Sync + 'static,
|
||||
|
@ -107,41 +119,51 @@ impl Processors {
|
|||
name: String,
|
||||
job: JobInfo,
|
||||
) -> impl Future<Item = Self, Error = JobError> {
|
||||
let mut jobs = self.jobs.take().unwrap();
|
||||
let args = job.args.clone();
|
||||
let jobs = self.jobs.take().unwrap();
|
||||
|
||||
let processor = self.inner.remove(&name);
|
||||
|
||||
processor
|
||||
.ok_or(JobError::MissingProcessor)
|
||||
.into_future()
|
||||
.and_then(move |processor| {
|
||||
.and_then(move |processor| process(self, jobs, processor, job, name))
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for Processors {
|
||||
fn default() -> Self {
|
||||
Processors {
|
||||
inner: Default::default(),
|
||||
jobs: Default::default(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn process(
|
||||
mut processors: Processors,
|
||||
mut jobs: Jobs,
|
||||
processor: Box<Processor + Send + Sync + 'static>,
|
||||
job: JobInfo,
|
||||
name: String,
|
||||
) -> impl Future<Item = Processors, Error = JobError> {
|
||||
let args = job.args.clone();
|
||||
let local_name = name.clone();
|
||||
let local_name_2 = name.clone();
|
||||
|
||||
let fut = processor.process(args).then(move |res| match res {
|
||||
Ok(_) => Ok(info!("Job completed, {}", name)),
|
||||
Err(e) => {
|
||||
error!("Job errored, {}, {}", name, e);
|
||||
|
||||
Err(JobError::Processing(e))
|
||||
}
|
||||
Err(e) => Err(error!("Job errored, {}, {}", name, e)),
|
||||
});
|
||||
|
||||
self.inner.insert(local_name, processor);
|
||||
processors.inner.insert(local_name, processor);
|
||||
|
||||
fut.then(move |res| {
|
||||
match res {
|
||||
Ok(_) => (),
|
||||
Err(_) => {
|
||||
if res.is_err() {
|
||||
jobs.requeue(&local_name_2, job);
|
||||
}
|
||||
}
|
||||
|
||||
self.jobs = Some(jobs);
|
||||
processors.jobs = Some(jobs);
|
||||
|
||||
Ok(self)
|
||||
Ok(processors)
|
||||
})
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue