itsscb/tokio
1
0
Fork 0
mirror of https://github.com/tokio-rs/tokio.git synced 2025-09-25 12:00:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

sync: add AtomicTask::take_task()

Browse Source
This commit is contained in:
Sean McArthur 2019-02-14 14:55:13 -08:00
parent 860ca79d62
commit d7a556fe8b
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tokio-sync/src/task/atomic_task.rs
View File

@ -242,6 +242,15 @@ impl AtomicTask {
/// If `register` has not been called yet, then this does nothing.
pub fn notify(&self) {
debug!(" + notify");
if let Some(task) = self.take_task() {
task.notify();
}
}
/// Attempts to take the `Task` value out of the `AtomicTask` with the
/// intention that the caller will notify the task later.
pub fn take_task(&self) -> Option<Task> {
debug!(" + take_task");
// AcqRel ordering is used in order to acquire the value of the `task`
// cell as well as to establish a `release` ordering with whatever
// memory the `AtomicTask` is associated with.
@ -253,11 +262,9 @@ impl AtomicTask {
// Release the lock
self.state.fetch_and(!NOTIFYING, Release);
debug!(" + Done notifying");
debug!(" + Done taking");
if let Some(task) = task {
task.notify();
}
task
}
state => {
debug!(" + state = {:?}", state);
@ -272,6 +279,7 @@ impl AtomicTask {
state == REGISTERING ||
state == REGISTERING | NOTIFYING ||
state == NOTIFYING);
None
}
}
}