xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-14 13:13:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed bug for websocket message handler rebind

Browse Source
This commit is contained in:
RockYang
2024-10-08 16:41:19 +08:00
parent c91a38a882
commit 430a7b2297
4 changed files with 35 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
web/src/store/sharedata.js
View File

@@ -6,8 +6,7 @@ export const useSharedStore = defineStore('shared', {
showLoginDialog: false,
chatListStyle: Storage.get("chat_list_style","chat"),
chatStream: Storage.get("chat_stream",true),
socket: WebSocket,
messageHandlers:{},
socket: {conn:null, handlers:{}},
mobileTheme: Storage.get("mobile_theme", "light"),
adminTheme: Storage.get("admin_theme", "light"),
isLogin: false
@@ -26,41 +25,42 @@ export const useSharedStore = defineStore('shared', {
Storage.set("chat_stream", value);
},
setSocket(value) {
this.socket = value;
for (const key in this.socket.handlers) {
this.setMessageHandler(value, this.socket.handlers[key])
}
this.socket.conn = value
},
addMessageHandler(key, callback) {
if (!this.messageHandlers[key]) {
this.setMessageHandler(callback)
if (!this.socket.handlers[key]) {
this.socket.handlers[key] = callback;
}
this.messageHandlers[key] = callback;
this.setMessageHandler(this.socket.conn, callback)
},
setMessageHandler(callback) {
if (this.socket instanceof WebSocket && this.socket.readyState === WebSocket.OPEN) {
console.log(callback)
this.socket.addEventListener('message', (event) => {
try {
if (event.data instanceof Blob) {
const reader = new FileReader();
reader.readAsText(event.data, "UTF-8");
reader.onload = () => {
callback(JSON.parse(String(reader.result)))
}
setMessageHandler(conn, callback) {
if (!conn) {
return
}
conn.addEventListener('message', (event) => {
try {
if (event.data instanceof Blob) {
const reader = new FileReader();
reader.readAsText(event.data, "UTF-8");
reader.onload = () => {
callback(JSON.parse(String(reader.result)))
}
} catch (e) {
console.warn(e)
}
})
} else {
setTimeout(() => {
this.setMessageHandler(callback)
}, 1000)
}
} catch (e) {
console.warn(e)
}
})
},
removeMessageHandler(key) {
if (this.socket.readyState === WebSocket.OPEN) {
this.socket.removeEventListener('message', this.messageHandlers[key])
if (this.socket.conn && this.socket.conn.readyState === WebSocket.OPEN) {
this.socket.conn.removeEventListener('message', this.socket.handlers[key])
}
delete this.messageHandlers[key]
delete this.socket.handlers[key]
},
setMobileTheme(theme) {
this.mobileTheme = theme