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 Sink for WebSocketStream 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