Skip to content

Commit ae6cf88

Browse files
committed
fixed force close error
When a client force close(such as CTRL+C), websocket server will show error because can't get next message.
1 parent 4dae1c1 commit ae6cf88

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

websocket_server/websocket_server.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,8 +157,10 @@ def read_bytes(self, num):
157157
return bytes
158158

159159
def read_next_message(self):
160-
161-
b1, b2 = self.read_bytes(2)
160+
try:
161+
b1, b2 = self.read_bytes(2)
162+
except ValueError as e:
163+
b1, b2 = 0, 0
162164

163165
fin = b1 & FIN
164166
opcode = b1 & OPCODE

0 commit comments

Comments
 (0)