diff --git a/aiowebsocket/freams.py b/aiowebsocket/freams.py index fadacbd..22e97d0 100644 --- a/aiowebsocket/freams.py +++ b/aiowebsocket/freams.py @@ -289,6 +289,8 @@ async def write(self, fin, code, message, mask=True, rsv1=0, rsv2=0, rsv3=0): output.write(message) self.writer.write(output.getvalue()) + # it would see an error when the socket is closed. + await self.writer.drain() async def receive_close(self): """ When you receive a message that