smol-rs
/
async-io
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix a bug

This commit is contained in:
Stjepan Glavina 2020-09-27 23:42:14 +02:00
parent ba08b6a3ab
commit e50ce4302e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/reactor.rs
View File

@ -402,7 +402,7 @@ impl Source {
/// If a different waker is already registered, it gets replaced and woken.
fn register(&self, dir: usize, waker: &Waker) -> io::Result<()> {
let mut state = self.state.lock().unwrap();
let is_empty = (state[READ].is_empty(), state[WRITE].is_empty());
let was_empty = state[dir].is_empty();
if let Some(w) = state[dir].waker.take() {
if w.will_wake(waker) {
@ -414,7 +414,7 @@ impl Source {
state[dir].waker = Some(waker.clone());
// Update interest in this I/O handle.
if is_empty != (state[READ].is_empty(), state[WRITE].is_empty()) {
if was_empty {
Reactor::get().poller.interest(
self.raw,
Event {
@ -478,7 +478,7 @@ impl Source {
// Remember the current ticks.
if ticks.is_none() {
ticks = Some((Reactor::get().ticker(), state[READ].tick));
ticks = Some((Reactor::get().ticker(), state[dir].tick));
}
Poll::Pending