Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

fix(wk websocket): only send messages to the open websocket

Browse source
This commit is contained in:
Dmitry Gozman 2020-01-23 09:17:27 -08:00
parent 5a67d780e5
commit 8b9d36c21c
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/server/wkPlaywright.ts
View file

@ -283,7 +283,12 @@ function wrapTransportWithWebSocket(transport: ConnectionTransport) {
}
socket = s;
s.on('message', message => transport.send(Buffer.from(message).toString()));
transport.onmessage = message => s.send(message);
transport.onmessage = message => {
// We are not notified when socket starts closing, and sending messages to a closing
// socket throws an error.
if (s.readyState !== ws.CLOSING)
s.send(message);
};
s.on('close', () => {
socket = undefined;
transport.onmessage = undefined;