Reactjs 用钩关闭时重新打开腹板箱

我正在React应用程序中使用WebSocket。 任务之一是在连接关闭时重新打开连接

不知道怎么做。看起来很有效，但我不知道这是否足够：

export const useWebSocket = () => {
    const webSocket = useRef();

    useEffect(() => {
        webSocket.current = new WebSocket("ws://url");

        webSocket.current.onclose = event => {
            if (!event.wasClean) {
                webSocket.current = new WebSocket("ws://url");
            }
        }

        return () => webSocket.current.close();
    }, []);
}

---

我上面的解决方案中存在一个问题：重新连接后，消息无法到达

webSocket.current.onmessage