Add websocket handling for selected flights.

This commit is contained in:
Dan Albert
2022-03-01 20:42:59 -08:00
parent 6d29bfdf65
commit 8e8bbe84f3
13 changed files with 190 additions and 13 deletions

View File

@@ -0,0 +1,36 @@
import { ReactChild, createContext, useEffect, useState } from "react";
import { WEBSOCKET_URL } from "../../api/backend";
const socket = new WebSocket(WEBSOCKET_URL);
export const SocketContext = createContext(socket);
interface SocketProviderProps {
children: ReactChild;
}
export const SocketProvider = (props: SocketProviderProps) => {
const [ws, setWs] = useState<WebSocket>(socket);
useEffect(() => {
const onClose = () => {
setWs(new WebSocket(WEBSOCKET_URL));
};
const onError = (error: Event) => {
console.log(`Websocket error: ${error}`);
};
ws.addEventListener("close", onClose);
ws.addEventListener("error", onError);
return () => {
ws.removeEventListener("close", onClose);
ws.removeEventListener("error", onError);
};
});
return (
<SocketContext.Provider value={ws}>{props.children}</SocketContext.Provider>
);
};