使用 socket.io 事件更新状态

use socket.io event to update state

我有一个 React 功能组件:

function Chat(props) {
  const [messages, setMessages] = useState([]);

  const inputRef = useRef();

  //The socket is a module that exports the actual socket.io socket
  socket.socket.on("chatMessage", (msg) => {
    setMessages([...messages, msg]);
  });

  const inputChanged = (e) => {
    if (e.key === "Enter") {
      socket.sendMessage(inputRef.current.value)
        .then(() => {
          //do something
        })
        .catch(console.log);
      inputRef.current.value = "";
    }
  };

  return (
      <div>
        {messages.map((msg, i) => <ChatMessage key={i}>{msg}</ChatMessage>)}
        <input ref={inputRef} onKeyPress={inputChanged} />
     </div>
  );
}

但是当我从 socket.socket.on("chatMessage" 更新状态时,出现错误

Can't perform a React state update on an unmounted component

套接字告诉我响应时间很长,并且开始发生一些递归。

我应该如何从套接字事件更新我的组件状态?

您需要在 useEffect 函数中设置套接字监听器,否则在每次重新渲染时都会创建一个新实例,而旧实例将继续监听并导致内存溢出和意外的状态更新错误。还要清除你的套接字监听器

function Chat(props) {
  const [messages, setMessages] = useState([]);

  const inputRef = useRef();

  useEffect(() => {
      //The socket is a module that exports the actual socket.io socket
      const addMessage = (msg) => setMessages(prevMessages => [...prevMessages, msg]);
      socket.socket.on("chatMessage", addMessage)
      () => {
            // turning of socket listner on unmount
          socket.off('chatMessage', addMessage);
       }
  }, [])

  const inputChanged = (e) => {
    if (e.key === "Enter") {
      socket.sendMessage(inputRef.current.value)
        .then(() => {
          //do something
        })
        .catch(console.log);
      inputRef.current.value = "";
    }
  };

  return (
      <div>
        {messages.map((msg, i) => <ChatMessage key={i}>{msg}</ChatMessage>)}
        <input ref={inputRef} onKeyPress={inputChanged} />
     </div>
  );
}

// P.S。确保您正在使用回调方法更新状态

Socket 是副作用。

useEffecf(() => {
  const callback =  (msg) => {
    setMessages([...messages, msg]);
  };

  socket.socket.on("chatMessage", callback);

  return () => {
    socket.socket.off("chatMessage", callback);
  };
}, []);