Merge pull request #590 from benesch/copy-out-short-reads

Don't accidentally return early from CopyOutReader

Author: sfackler

postgres/src/copy_out_reader.rs

@@ -34,15 +34,15 @@ impl Read for CopyOutReader<'_> {
 
 impl BufRead for CopyOutReader<'_> {
     fn fill_buf(&mut self) -> io::Result<&[u8]> {
-        if !self.cur.has_remaining() {
+        while !self.cur.has_remaining() {
             let mut stream = self.stream.pinned();
             match self
                 .connection
                 .block_on({ async { stream.next().await.transpose() } })
             {
                 Ok(Some(cur)) => self.cur = cur,
                 Err(e) => return Err(io::Error::new(io::ErrorKind::Other, e)),
-                Ok(None) => {}
+                Ok(None) => break,
             };
         }