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

opt: add heartbeat message for websocket connects

Browse Source
This commit is contained in:
RockYang
2024-01-22 18:42:51 +08:00
parent b3d0c1ef9c
commit 0f05970141
4 changed files with 38 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
web/src/views/ChatPlus.vue
View File

@@ -543,6 +543,7 @@ const lineBuffer = ref(''); // 输出缓冲行
const socket = ref(null);
const activelyClose = ref(false); // 主动关闭
const canSend = ref(true);
const heartbeatHandle = ref(null)
const connect = function (chat_id, role_id) {
let isNewChat = false;
if (!chat_id) {
@@ -589,6 +590,14 @@ const connect = function (chat_id, role_id) {
loadChatHistory(chat_id);
}
// 发送心跳消息
clearInterval(heartbeatHandle.value)
heartbeatHandle.value = setInterval(() => {
if (socket.value !== null) {
socket.value.send(JSON.stringify({type: "heartbeat", content: "ping"}))
}
}, 5000);
});
_socket.addEventListener('message', event => {
@@ -718,7 +727,7 @@ const sendMessage = function () {
showHello.value = false
disableInput(false)
socket.value.send(prompt.value);
socket.value.send(JSON.stringify({type: "chat", content: prompt.value}));
previousText.value = prompt.value;
prompt.value = '';
return true;

11
web/src/views/mobile/ChatSession.vue
View File

@@ -200,6 +200,7 @@ const lineBuffer = ref(''); // 输出缓冲行
const socket = ref(null);
const activelyClose = ref(false); // 主动关闭
const canSend = ref(true);
const heartbeatHandle = ref(null)
const connect = function (chat_id, role_id) {
let isNewChat = false;
if (!chat_id) {
@@ -238,6 +239,14 @@ const connect = function (chat_id, role_id) {
orgContent: role.helloMsg,
})
}
// 发送心跳消息
clearInterval(heartbeatHandle.value)
heartbeatHandle.value = setInterval(() => {
if (socket.value !== null) {
socket.value.send(JSON.stringify({type: "heartbeat", content: "ping"}))
}
}, 5000);
});
_socket.addEventListener('message', event => {
@@ -370,7 +379,7 @@ const sendMessage = () => {
})
disableInput(false)
socket.value.send(prompt.value);
socket.value.send(JSON.stringify({type: "chat", content: prompt.value}));
previousText.value = prompt.value;
prompt.value = '';
return true;