jamesf/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(chat): prevent ws not open errors

Browse source
This commit is contained in:
ItzCrazyKns 2024-10-15 18:04:50 +05:30
parent 7532c436db
commit 3a01eebc04
6 changed files with 59 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

31
ui/components/ChatWindow.tsx
View file

@ -171,11 +171,22 @@ const useSocket = (
}
}, 10000);
ws.onopen = () => {
console.log('[DEBUG] open');
clearTimeout(timeoutId);
setIsWSReady(true);
};
ws.addEventListener('message', (e) => {
const data = JSON.parse(e.data);
if (data.type === 'signal' && data.data === 'open') {
const interval = setInterval(() => {
if (ws.readyState === 1) {
setIsWSReady(true);
clearInterval(interval);
}
}, 5);
clearTimeout(timeoutId);
console.log('[DEBUG] opened');
}
if (data.type === 'error') {
toast.error(data.data);
}
});
ws.onerror = () => {
clearTimeout(timeoutId);
@ -189,13 +200,6 @@ const useSocket = (
console.log('[DEBUG] closed');
};
ws.addEventListener('message', (e) => {
const data = JSON.parse(e.data);
if (data.type === 'error') {
toast.error(data.data);
}
});
setWs(ws);
};
@ -325,6 +329,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
useEffect(() => {
if (isMessagesLoaded && isWSReady) {
setIsReady(true);
console.log('[DEBUG] ready');
}
}, [isMessagesLoaded, isWSReady]);
@ -477,7 +482,7 @@ const ChatWindow = ({ id }: { id?: string }) => {
sendMessage(initialMessage);
}
// eslint-disable-next-line react-hooks/exhaustive-deps
}, [isReady, initialMessage, ws?.readyState]);
}, [ws?.readyState, isReady, initialMessage, isWSReady]);
if (hasError) {
return (