trade/trade-drift-dlob
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' into mainnet-beta

Browse Source 
memory management
This commit is contained in:
Nour Alharithi
2023-12-12 09:11:47 -08:00
parent 5d68938d6b 436e63dbd5
commit f76f1dcb5d
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/wsConnectionManager.ts
View File

@@ -103,7 +103,7 @@ async function main() {
const subscribers = channelSubscribers.get(subscribedChannel); const subscribers = channelSubscribers.get(subscribedChannel);
if (subscribers) { if (subscribers) {
subscribers.forEach((ws) => { subscribers.forEach((ws) => {
if (ws.readyState === WebSocket.OPEN) if (ws.readyState === WebSocket.OPEN && ws.bufferedAmount < 20)
ws.send( ws.send(
JSON.stringify({ channel: subscribedChannel, data: message }) JSON.stringify({ channel: subscribedChannel, data: message })
); );
@@ -271,6 +271,19 @@ async function main() {
server.on('error', (error) => { server.on('error', (error) => {
console.error('Server error:', error); console.error('Server error:', error);
}); });
// Periodic check for bad clients and disconnect them to alleviate backpressure
setInterval(() => {
let set = new Set<WebSocket>();
for (const [_channel, wsSet] of channelSubscribers) {
set = new Set([...set, ...wsSet]);
}
for (const ws of set) {
if (ws.bufferedAmount > 500) {
ws.close();
}
}
}, 5000);
} }
async function recursiveTryCatch(f: () => void) { async function recursiveTryCatch(f: () => void) {