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

resub on redis reconnect

Browse Source
This commit is contained in:
Nour Alharithi
2023-11-10 08:54:29 -08:00
parent 65e575ac67
commit 384e785371
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/wsConnectionManager.ts
View File

@@ -36,6 +36,16 @@ async function main() {
const channelSubscribers = new Map<string, Set<WebSocket>>(); const channelSubscribers = new Map<string, Set<WebSocket>>();
const subscribedChannels = new Set<string>(); const subscribedChannels = new Set<string>();
redisClient.client.on('connect', () => {
subscribedChannels.forEach(async (channel) => {
try {
await redisClient.client.subscribe(channel);
} catch (error) {
console.error(`Error subscribing to ${channel}:`, error);
}
});
});
redisClient.client.on('message', (subscribedChannel, message) => { redisClient.client.on('message', (subscribedChannel, message) => {
const subscribers = channelSubscribers.get(subscribedChannel); const subscribers = channelSubscribers.get(subscribedChannel);
subscribers.forEach((ws) => { subscribers.forEach((ws) => {
@@ -132,7 +142,6 @@ async function main() {
// Listen for pong messages // Listen for pong messages
ws.on('pong', () => { ws.on('pong', () => {
console.log('poooooooong');
isAlive = true; isAlive = true;
clearTimeout(pongTimeoutId); clearTimeout(pongTimeoutId);
}); });
@@ -146,6 +155,7 @@ async function main() {
if (subscribers.delete(ws) && subscribers.size === 0) { if (subscribers.delete(ws) && subscribers.size === 0) {
redisClient.client.unsubscribe(channel); redisClient.client.unsubscribe(channel);
channelSubscribers.delete(channel); channelSubscribers.delete(channel);
subscribedChannels.delete(channel);
} }
}); });
}); });