mirrors/background-jobs
1
0
Fork 0
mirror of https://git.asonix.dog/asonix/background-jobs.git synced 2024-11-22 03:51:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

Port warn on drop from 0.9.x

Browse source
This commit is contained in:
Aode (Lion) 2021-10-06 21:29:34 -05:00
parent c0ce6f303e
commit 461c9e5ed2
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
jobs-actix/src/worker.rs
View file

@ -56,6 +56,19 @@ impl Worker for LocalWorkerHandle {
} }
} }
struct LogOnDrop<F>(F)
where
F: Fn() -> Span;
impl<F> Drop for LogOnDrop<F>
where
F: Fn() -> Span,
{
fn drop(&mut self) {
(self.0)().in_scope(|| info!("Worker closing"));
}
}
pub(crate) async fn local_worker<State>( pub(crate) async fn local_worker<State>(
queue: String, queue: String,
processors: CachedProcessorMap<State>, processors: CachedProcessorMap<State>,
@ -68,6 +81,8 @@ pub(crate) async fn local_worker<State>(
let handle = LocalWorkerHandle { tx, id, queue }; let handle = LocalWorkerHandle { tx, id, queue };
let log_on_drop = LogOnDrop(|| handle.span("closing"));
loop { loop {
let span = handle.span("request"); let span = handle.span("request");
if let Err(e) = server if let Err(e) = server
@ -105,5 +120,5 @@ pub(crate) async fn local_worker<State>(
break; break;
} }
handle.span("closing").in_scope(|| info!("Worker closing")); drop(log_on_drop);
} }