forked/hyper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor(client): replace futures oneshot with tokio in dispatcher

Browse Source
This commit is contained in:
Sean McArthur
2020-03-25 13:32:06 -07:00
parent e6a6ddef7c
commit 97ed0478a8
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/client/dispatch.rs
View File

@@ -1,6 +1,5 @@
use futures_channel::oneshot;
use futures_util::future; use futures_util::future;
use tokio::sync::mpsc; use tokio::sync::{mpsc, oneshot};
use crate::common::{task, Future, Pin, Poll}; use crate::common::{task, Future, Pin, Poll};
@@ -195,15 +194,15 @@ pub enum Callback<T, U> {
impl<T, U> Callback<T, U> { impl<T, U> Callback<T, U> {
pub(crate) fn is_canceled(&self) -> bool { pub(crate) fn is_canceled(&self) -> bool {
match *self { match *self {
Callback::Retry(ref tx) => tx.is_canceled(), Callback::Retry(ref tx) => tx.is_closed(),
Callback::NoRetry(ref tx) => tx.is_canceled(), Callback::NoRetry(ref tx) => tx.is_closed(),
} }
} }
pub(crate) fn poll_canceled(&mut self, cx: &mut task::Context<'_>) -> Poll<()> { pub(crate) fn poll_canceled(&mut self, cx: &mut task::Context<'_>) -> Poll<()> {
match *self { match *self {
Callback::Retry(ref mut tx) => tx.poll_canceled(cx), Callback::Retry(ref mut tx) => tx.poll_closed(cx),
Callback::NoRetry(ref mut tx) => tx.poll_canceled(cx), Callback::NoRetry(ref mut tx) => tx.poll_closed(cx),
} }
} }