Better error handling

This commit is contained in:
Luke Steyn
2023-11-22 14:37:11 +01:00
parent 366aa8b178
commit 80a50c1eb0

View File

@@ -28,8 +28,15 @@ const wss = new WebSocketServer({ server, path: '/ws' });
const REDIS_HOST = process.env.REDIS_HOST || 'localhost'; const REDIS_HOST = process.env.REDIS_HOST || 'localhost';
const REDIS_PORT = process.env.REDIS_PORT || '6379'; const REDIS_PORT = process.env.REDIS_PORT || '6379';
const WS_PORT = process.env.WS_PORT || '3000'; const WS_PORT = process.env.WS_PORT || '3000';
console.log(`WS LISTENER PORT : ${WS_PORT}`);
const REDIS_PASSWORD = process.env.REDIS_PASSWORD; const REDIS_PASSWORD = process.env.REDIS_PASSWORD;
const safeGetRawChannelFromMessage = (message: any): string => {
return message?.channel;
};
const getRedisChannelFromMessage = (message: any): string => { const getRedisChannelFromMessage = (message: any): string => {
const channel = message.channel; const channel = message.channel;
const marketName = message.market?.toUpperCase(); const marketName = message.market?.toUpperCase();
@@ -119,7 +126,17 @@ async function main() {
try { try {
redisChannel = getRedisChannelFromMessage(parsedMessage); redisChannel = getRedisChannelFromMessage(parsedMessage);
} catch (error) { } catch (error) {
ws.send(JSON.stringify({ error: error.message })); const requestChannel = safeGetRawChannelFromMessage(parsedMessage);
if (requestChannel) {
ws.send(
JSON.stringify({
channel: requestChannel,
error: error.message,
})
);
} else {
ws.close(1003, JSON.stringify({ error: error.message }));
}
return; return;
} }
@@ -165,7 +182,17 @@ async function main() {
try { try {
redisChannel = getRedisChannelFromMessage(parsedMessage); redisChannel = getRedisChannelFromMessage(parsedMessage);
} catch (error) { } catch (error) {
ws.send(JSON.stringify({ error: error.message })); const requestChannel = safeGetRawChannelFromMessage(parsedMessage);
if (requestChannel) {
ws.send(
JSON.stringify({
channel: requestChannel,
error: error.message,
})
);
} else {
ws.close(1003, JSON.stringify({ error: error.message }));
}
return; return;
} }
const subscribers = channelSubscribers.get(redisChannel); const subscribers = channelSubscribers.get(redisChannel);