dnim/rust-postgres
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify logic a bit

Browse Source
This commit is contained in:
Steven Fackler
2019-07-31 19:44:16 -07:00
parent beb509f3f3
commit 81bc845bcf
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tokio-postgres/src/connection.rs
View File

@@ -198,6 +198,14 @@ where
return Ok(false);
}
if let Poll::Pending = Pin::new(&mut self.stream)
.poll_ready(cx)
.map_err(Error::io)?
{
trace!("poll_write: waiting on socket");
return Ok(false);
}
let request = match self.poll_request(cx) {
Poll::Ready(Some(request)) => request,
Poll::Ready(None) if self.responses.is_empty() && self.state == State::Active => {
@@ -220,15 +228,6 @@ where
}
};
if let Poll::Pending = Pin::new(&mut self.stream)
.poll_ready(cx)
.map_err(Error::io)?
{
trace!("poll_write: waiting on socket");
self.pending_request = Some(request);
return Ok(false);
}
match request {
RequestMessages::Single(request) => {
Pin::new(&mut self.stream)