From d5ff6e08ca8a22817fbff6b1f3bb033ffbffe98e Mon Sep 17 00:00:00 2001
From: Lei OT <lyt@hainatravel.com>
Date: Fri, 23 Feb 2024 11:34:23 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=BB=E5=8A=A8=E9=87=8D=E8=BF=9E?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/stores/ConversationStore.js | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/src/stores/ConversationStore.js b/src/stores/ConversationStore.js
index 96eb136..dd3ce0f 100644
--- a/src/stores/ConversationStore.js
+++ b/src/stores/ConversationStore.js
@@ -28,12 +28,12 @@ const initialConversationState = {
   // referenceMsg: {},
 };
 
-export const templatesSlice = (set) => ({
+const templatesSlice = (set) => ({
   templates: [],
   setTemplates: (templates) => set({ templates }),
 });
 
-export const websocketSlice = (set, get) => ({
+const websocketSlice = (set, get) => ({
   websocket: null,
   websocketOpened: null,
   websocketRetrying: null,
@@ -72,6 +72,13 @@ export const websocketSlice = (set, get) => ({
     websocket.disconnect();
     return set({ websocket: null });
   },
+  reconnectWebsocket: (userId) => {
+    const {disconnectWebsocket, connectWebsocket} = get();
+    disconnectWebsocket();
+    setTimeout(() => {
+      connectWebsocket(userId);
+    }, 500);
+  },
   handleMessage: (data) => {
     console.log('handleMessage------------------');
     console.log(data);
@@ -113,12 +120,12 @@ export const websocketSlice = (set, get) => ({
   },
 });
 
-export const referenceMsgSlice = (set) => ({
+const referenceMsgSlice = (set) => ({
   referenceMsg: {},
   setReferenceMsg: (referenceMsg) => set({ referenceMsg }),
 });
 
-export const conversationSlice = (set, get) => ({
+const conversationSlice = (set, get) => ({
   conversationsList: [],
   currentConversation: {},
 
@@ -166,7 +173,7 @@ export const conversationSlice = (set, get) => ({
   },
 });
 
-export const messageSlice = (set, get) => ({
+const messageSlice = (set, get) => ({
   msgListLoading: false,
   activeConversations: {},
   setMsgLoading: (msgListLoading) => set({ msgListLoading }),