如何在react中切换聊天时订阅新的websocket会话

constructor(props) {
  super(props);
  this.state = {
    websocket: null,
    chatSessionId: null,
  };
}

componentDidMount() {
  const websocket = new WebSocket('wss://your-websocket-url');
  websocket.onopen = () => {
    console.log('WebSocket连接已建立');
  };
  websocket.onmessage = (event) => {
    console.log('收到WebSocket消息:', event.data);
    // 处理收到的消息
  };
  websocket.onclose = () => {
    console.log('WebSocket连接已关闭');
  };
  this.setState({ websocket });
}

switchChatSession(newChatSessionId) {
  const { websocket, chatSessionId } = this.state;
  if (websocket && chatSessionId !== newChatSessionId) {
    // 取消订阅旧的WebSocket会话
    websocket.send(JSON.stringify({ action: 'unsubscribe', chatSessionId }));
    // 订阅新的WebSocket会话
    websocket.send(JSON.stringify({ action: 'subscribe', chatSessionId: newChatSessionId }));
    this.setState({ chatSessionId: newChatSessionId });
  }
}

render() {
  return (
    <div>
      <button onClick={() => this.switchChatSession('new-session-id')}>
        切换聊天会话
      </button>
    </div>
  );
}