machankura/mempool
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix acceleration websocket ping error

Browse Source
This commit is contained in:
Mononaut 2024-11-19 23:00:28 +00:00
parent abe9aa1fdc
commit 9a81db8e6c
No known key found for this signature in database
GPG Key ID: A3F058E41374C04E
1 changed files with 19 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
backend/src/api/services/acceleration.ts
View File

@ -246,11 +246,11 @@ class AccelerationApi {
this.startedWebsocketLoop = true; this.startedWebsocketLoop = true;
if (!this.ws) { if (!this.ws) {
this.ws = new WebSocket(this.websocketPath); this.ws = new WebSocket(this.websocketPath);
this.websocketConnected = true;
this.lastPing = 0; this.lastPing = 0;
this.ws.on('open', () => { this.ws.on('open', () => {
logger.info(`Acceleration websocket opened to ${this.websocketPath}`); logger.info(`Acceleration websocket opened to ${this.websocketPath}`);
this.websocketConnected = true;
this.ws?.send(JSON.stringify({ this.ws?.send(JSON.stringify({
'watch-accelerations': true 'watch-accelerations': true
})); }));
@ -286,17 +286,28 @@ class AccelerationApi {
logger.debug('received pong from acceleration websocket server'); logger.debug('received pong from acceleration websocket server');
this.lastPong = Date.now(); this.lastPong = Date.now();
}); });
} else { } else if (this.websocketConnected) {
if (this.lastPing && this.lastPing > this.lastPong && (Date.now() - this.lastPing > 10000)) { if (this.lastPing && this.lastPing > this.lastPong && (Date.now() - this.lastPing > 10000)) {
logger.warn('No pong received within 10 seconds, terminating connection'); logger.warn('No pong received within 10 seconds, terminating connection');
this.ws.terminate(); try {
this.ws?.terminate();
} catch (e) {
logger.warn('failed to terminate acceleration websocket connection: ' + (e instanceof Error ? e.message : e));
} finally {
this.ws = null; this.ws = null;
this.websocketConnected = false; this.websocketConnected = false;
this.lastPing = 0; this.lastPing = 0;
}
} else if (!this.lastPing || (Date.now() - this.lastPing > 30000)) { } else if (!this.lastPing || (Date.now() - this.lastPing > 30000)) {
logger.debug('sending ping to acceleration websocket server'); logger.debug('sending ping to acceleration websocket server');
this.ws.ping(); if (this.ws?.readyState === WebSocket.OPEN) {
try {
this.ws?.ping();
this.lastPing = Date.now(); this.lastPing = Date.now();
} catch (e) {
logger.warn('failed to send ping to acceleration websocket server: ' + (e instanceof Error ? e.message : e));
}
}
} }
} }
await new Promise(resolve => setTimeout(resolve, 5000)); await new Promise(resolve => setTimeout(resolve, 5000));