ComputerSurge/komodo
3
0
Fork 0
forked from github-starred/komodo
Code Pull Requests Packages Releases Activity

clean up ws correctly

Browse Source
This commit is contained in:
mbecker20
2023-08-11 02:09:47 -04:00
parent bec8b3ce0f
commit 5e572ce5ef
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
frontend/src/util/socket.tsx
View File

@@ -64,12 +64,15 @@ export const WebsocketProvider = ({ children }: { children: ReactNode }) => {
const invalidate = useInvalidate();
const [ws] = useWebsocket();
const on_open_fn = () => on_open(ws);
const on_message_fn = (e: MessageEvent) => on_message(e, invalidate);
useEffect(() => {
ws.addEventListener("open", () => on_open(ws));
ws.addEventListener("message", (e) => on_message(e, invalidate));
ws.addEventListener("open", on_open_fn);
ws.addEventListener("message", on_message_fn);
return () => {
ws.removeEventListener("open", () => on_open(ws));
ws.removeEventListener("message", (e) => on_message(e, invalidate));
ws.removeEventListener("open", on_open_fn);
ws.removeEventListener("message", on_message_fn);
};
});