Forma correcta de usar ganchos React + WebSockets

Necesito conectarme al servidor WebSockets y registrar sus mensajes. Con el componente de clase React, pondría esta lógica en componentDidMount gancho de ciclo de vida y seguir adelante felizmente, pero no estoy seguro de cómo implementarlo correctamente con ganchos.

Aquí está mi primer intento.

import React, {useEffect} from 'react';

export default function AppWs() {
  useEffect(() => {
    let ws = new WebSocket('wss://ws.kraken.com/');
    ws.onopen = () => console.log('ws opened');
    ws.onclose = () => console.log('ws closed');

    ws.onmessage = e => {
      const message = JSON.parse(e.data);
      console.log('e', message);
    };

    return () => {
      ws.close();
    }
  }, []);

  return (
    <div>hooks + ws</div>
  )
}

Agregué conexión y lógica de registro a useEffect, proporcionó una matriz vacía con dependencias, y todo funcionó muy bien. Hasta que tuve que agregar pause estado para pausar el registro.

export default function AppWs() {
  const [isPaused, setPause] = useState(false);

  useEffect(() => {
    let ws = new WebSocket('wss://ws.kraken.com/');
    ws.onopen = () => console.log('ws opened');
    ws.onclose = () => console.log('ws closed');

    ws.onmessage = e => {
      if (isPaused) return;
      const message = JSON.parse(e.data);
      console.log('e', message);
    };

    return () => {
      ws.close();
    }
  }, []);

  return (
    <div>
      <button onClick={() => setPause(!isPaused)}>{isPaused ? 'Resume' : 'Pause'}</button>
    </div>
  )
}

ESLint comenzó a gritarme que debería agregar isPaused estado como una dependencia de useEffect.
Bueno, está bien, hecho.
Pero noté la reconexión al servidor WS cada vez que hago clic en el botón. Esto claramente no es lo que quiero.

Mi siguiente iteración fue usar dos useEffects: uno para conexión y otro para procesamiento de mensajes.

export default function AppWs() {
  const [isPaused, setPause] = useState(false);
  const [ws, setWs] = useState(null);

  useEffect(() => {
    const wsClient = new WebSocket('wss://ws.kraken.com/');
    wsClient.onopen = () => {
      console.log('ws opened');
      setWs(wsClient);
    };
    wsClient.onclose = () => console.log('ws closed');

    return () => {
      wsClient.close();
    }
  }, []);

  useEffect(() => {
    if (!ws) return;

    ws.onmessage = e => {
      if (isPaused) return;
      const message = JSON.parse(e.data);
      console.log('e', message);
    };
  }, [isPaused, ws]);

  return (
    <div>
      <button onClick={() => setPause(!isPaused)}>{isPaused ? 'Resume' : 'Pause'}</button>
    </div>
  )
}

Esto funciona como se esperaba, pero tengo la sensación de que me falta algo y esta tarea se puede resolver más fácilmente, con uno useEffect. Ayúdeme a refactorizar el código para convencerme de que estoy usando los ganchos React de manera adecuada. ¡Gracias!

Como solo está configurando el socket web una vez, creo que un mejor enfoque es usar una referencia en lugar de un estado:

El orden de useEffect es importante.

Como sugiere George en los comentarios, en la primera useEffect ws.current se guarda en una variable para asegurarse de que cuando close se llama se refiere a la misma instancia.

export default function AppWs() {
    const [isPaused, setPause] = useState(false);
    const ws = useRef(null);

    useEffect(() => {
        ws.current = new WebSocket("wss://ws.kraken.com/");
        ws.current.onopen = () => console.log("ws opened");
        ws.current.onclose = () => console.log("ws closed");

        const wsCurrent = ws.current;

        return () => {
            wsCurrent.close();
        };
    }, []);

    useEffect(() => {
        if (!ws.current) return;

        ws.current.onmessage = e => {
            if (isPaused) return;
            const message = JSON.parse(e.data);
            console.log("e", message);
        };
    }, [isPaused]);

    return (
        <div>
            <button onClick={() => setPause(!isPaused)}>
                {isPaused ? "Resume" : "Pause"}
            </button>
        </div>
    );
}

 useEffect(() => {
const socket = new WebSocket('wss://ws.kraken.com');
socket.addEventListener('message', function (event) {
  const a = JSON.parse(event.data);
  setPriceWebSocket(a);

  const amin = socket.send(
    JSON.stringify({
      event: 'subscribe',
      pair: ['XBT/USD', 'XBT/EUR', 'ADA/USD'],
      subscription: { name: 'ticker' },
    }),
  );
  

});