A Handy React Hook for creating auto-closing WebSockets.

useWebSocket.ts

const useWebSocket = (url, protocols, onMessage, onError) => {
  const ws = useRef(null);
  useEffect(() => {
    const socket = new WebSocket(url, protocols);
    socket.onmessage = onMessage;
    socket.onerror = onError;
    ws.current = socket;
    return () => {
      ws.current.close();
    };
  }, []);

  return ws;
};