Use flush combinator

2019-08-02 20:30:36 -07:00
parent 26a17ac4ed
commit 048e91a57d
1 changed files with 1 additions and 3 deletions
--- a/tokio-postgres/src/cancel_query_raw.rs
+++ b/tokio-postgres/src/cancel_query_raw.rs
@@ -23,9 +23,7 @@ where
    frontend::cancel_request(process_id, secret_key, &mut buf);

    stream.write_all(&buf).await.map_err(Error::io)?;
-    future::poll_fn(|cx| Pin::new(&mut stream).poll_flush(cx))
-        .await
-        .map_err(Error::io)?;
+    stream.flush().await.map_err(Error::io)?;
    future::poll_fn(|cx| Pin::new(&mut stream).poll_shutdown(cx))
        .await
        .map_err(Error::io)?;