From 16c4693c91afc4c39514b0950c42b17eb4d674ba Mon Sep 17 00:00:00 2001
From: Daniel Abramov <dabramov@snapview.de>
Date: Fri, 20 Oct 2017 12:00:51 +0200
Subject: [PATCH] Implement `Stream::close` for WebSocket

---
 src/lib.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index c42ac82..3402c05 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -133,6 +133,10 @@ impl<T> Sink for WebSocketStream<T> where T: AsyncRead + AsyncWrite {
     fn poll_complete(&mut self) -> Poll<(), WsError> {
         self.inner.write_pending().to_async()
     }
+
+    fn close(&mut self) -> Poll<(), WsError> {
+        self.inner.close(None).to_async()
+    }
 }
 
 /// Future returned from client_async() which will resolve