berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

use recursive poll

This commit is contained in:
fakeshadow 2021-01-14 03:03:03 +08:00
parent 2394890bf1
commit e7ec216842
1 changed files with 14 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/middleware/err_handlers.rs
View File

@ -152,26 +152,22 @@ where
type Output = Fut::Output; type Output = Fut::Output;
fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> { fn poll(mut self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
loop { match self.as_mut().project() {
match self.as_mut().project() { ErrorHandlersProj::ServiceFuture { fut, handlers } => {
ErrorHandlersProj::ServiceFuture { fut, handlers } => { let res = ready!(fut.poll(cx))?;
let res = ready!(fut.poll(cx))?; match handlers.get(&res.status()) {
match handlers.get(&res.status()) { Some(handler) => match handler(res)? {
Some(handler) => match handler(res)? { ErrorHandlerResponse::Response(res) => Poll::Ready(Ok(res)),
ErrorHandlerResponse::Response(res) => { ErrorHandlerResponse::Future(fut) => {
return Poll::Ready(Ok(res)) self.as_mut()
} .set(ErrorHandlersFuture::HandlerFuture { fut });
ErrorHandlerResponse::Future(fut) => self self.poll(cx)
.as_mut() }
.set(ErrorHandlersFuture::HandlerFuture { fut }), },
}, None => Poll::Ready(Ok(res)),
None => return Poll::Ready(Ok(res)),
}
}
ErrorHandlersProj::HandlerFuture { fut } => {
return fut.as_mut().poll(cx)
} }
} }
ErrorHandlersProj::HandlerFuture { fut } => fut.as_mut().poll(cx),
} }
} }
} }