This commit is contained in:
Steven Fackler
2017-07-19 21:27:12 -07:00
parent fed246e9fd
commit 4c6e36e03a
2 changed files with 2 additions and 4 deletions

View File

@@ -13,7 +13,7 @@ use std::os::windows::io::{AsRawSocket, RawSocket};
use postgres_protocol::message::frontend;
use postgres_protocol::message::backend;
use {Error, Result, TlsMode};
use {Result, TlsMode};
use error;
use tls::TlsStream;
use params::{ConnectParams, Host};

View File

@@ -147,15 +147,13 @@ where
}
Ok(Async::NotReady) => {
match sink.poll_complete() {
Ok(Async::Ready(())) => return Ok(Async::Ready((sink, stream.into_inner()))),
Ok(Async::NotReady) => {
Ok(Async::Ready(())) | Ok(Async::NotReady) => {
self.sink = Some(sink);
self.stream = Some(stream);
return Ok(Async::NotReady);
}
Err(e) => return Err((e, sink, stream.into_inner())),
}
return Ok(Async::NotReady);
}
Err(e) => return Err((e.into(), sink, stream.into_inner())),
}