Mirrors/chatmail-core
1
0
Fork 0
mirror of https://github.com/chatmail/core.git synced 2026-04-17 21:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat(scheduler): Allow overlapping IoPauseGuards (#4246)

Browse Source 
This enables using multiple pause guards at the same time.
This commit is contained in:
Floris Bruynooghe
2023-03-30 12:40:13 +02:00
committed by GitHub
parent 82ace72527
commit 91c10b3ac6
1 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/scheduler.rs
View File

@@ -50,7 +50,7 @@ impl SchedulerState {
pub(crate) async fn start(&self, context: Context) {
let mut inner = self.inner.write().await;
inner.started = true;
if inner.scheduler.is_none() && !inner.paused {
if inner.scheduler.is_none() && inner.paused == 0 {
Self::do_start(inner, context).await;
}
}
@@ -99,15 +99,15 @@ impl SchedulerState {
pub(crate) async fn pause<'a>(&'_ self, context: Context) -> IoPausedGuard {
{
let mut inner = self.inner.write().await;
inner.paused = true;
inner.paused += 1;
Self::do_stop(inner, &context).await;
}
let (tx, rx) = oneshot::channel();
tokio::spawn(async move {
rx.await.ok();
let mut inner = context.scheduler.inner.write().await;
inner.paused = false;
if inner.started && inner.scheduler.is_none() {
inner.paused -= 1;
if inner.paused == 0 && inner.started && inner.scheduler.is_none() {
SchedulerState::do_start(inner, context.clone()).await;
}
});
@@ -199,8 +199,10 @@ impl SchedulerState {
#[derive(Debug, Default)]
struct InnerSchedulerState {
scheduler: Option<Scheduler>,
/// Whether IO should be started if there is no [`IoPausedGuard`] active.
started: bool,
paused: bool,
/// The number of [`IoPausedGuard`]s that are outstanding.
paused: u32,
}
/// Guard to make sure the IO Scheduler is resumed.