feat: 请求终止的逻辑调整
This commit is contained in:
@@ -103,7 +103,7 @@
|
|||||||
|
|
||||||
import CreateServiceOrder from '@/components/CreateServiceOrder/index.vue'
|
import CreateServiceOrder from '@/components/CreateServiceOrder/index.vue'
|
||||||
|
|
||||||
import { agentChatStream } from '@/request/api/AgentChatStream';
|
import { agentChatStream, stopAbortTask } from '@/request/api/AgentChatStream';
|
||||||
import { mainPageData } from '@/request/api/MainPageDataApi';
|
import { mainPageData } from '@/request/api/MainPageDataApi';
|
||||||
import { conversationMsgList, recentConversation } from '@/request/api/ConversationApi';
|
import { conversationMsgList, recentConversation } from '@/request/api/ConversationApi';
|
||||||
|
|
||||||
@@ -145,8 +145,6 @@
|
|||||||
|
|
||||||
// 会话进行中标志
|
// 会话进行中标志
|
||||||
const isSessionActive = ref(false);
|
const isSessionActive = ref(false);
|
||||||
// 请求任务引用
|
|
||||||
const requestTaskRef = ref(null);
|
|
||||||
/// 指令
|
/// 指令
|
||||||
let commonType = ''
|
let commonType = ''
|
||||||
|
|
||||||
@@ -357,7 +355,8 @@
|
|||||||
conversationId: conversationId.value,
|
conversationId: conversationId.value,
|
||||||
agentId: agentId.value,
|
agentId: agentId.value,
|
||||||
messageType: isInstruct ? 1 : 0,
|
messageType: isInstruct ? 1 : 0,
|
||||||
messageContent: isInstruct ? commonType : message
|
messageContent: isInstruct ? commonType : message,
|
||||||
|
messageId: 'mid' + new Date().getTime()
|
||||||
}
|
}
|
||||||
|
|
||||||
// 插入AI消息
|
// 插入AI消息
|
||||||
@@ -390,8 +389,8 @@
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 流式接收内容
|
// 流式接收内容
|
||||||
const { promise, requestTask } = agentChatStream(args, (chunk) => {
|
const promise = agentChatStream(args, (chunk) => {
|
||||||
console.log('分段内容:', chunk)
|
// console.log('分段内容:', chunk)
|
||||||
if (chunk && chunk.error) {
|
if (chunk && chunk.error) {
|
||||||
chatMsgList.value[aiMsgIndex].msg = '请求错误,请重试';
|
chatMsgList.value[aiMsgIndex].msg = '请求错误,请重试';
|
||||||
clearInterval(loadingTimer);
|
clearInterval(loadingTimer);
|
||||||
@@ -456,8 +455,6 @@
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
|
|
||||||
// 存储请求任务
|
|
||||||
requestTaskRef.value = requestTask;
|
|
||||||
|
|
||||||
// 可选:处理Promise完成/失败, 已经在回调中处理数据,此处无需再处理
|
// 可选:处理Promise完成/失败, 已经在回调中处理数据,此处无需再处理
|
||||||
promise.then(data => {
|
promise.then(data => {
|
||||||
@@ -487,11 +484,7 @@
|
|||||||
|
|
||||||
// 停止请求函数
|
// 停止请求函数
|
||||||
const stopRequest = () => {
|
const stopRequest = () => {
|
||||||
if (requestTaskRef.value && requestTaskRef.value.abort) {
|
stopAbortTask()
|
||||||
// 标记请求已中止,用于过滤后续可能到达的数据
|
|
||||||
requestTaskRef.value.isAborted = true;
|
|
||||||
// 中止请求
|
|
||||||
requestTaskRef.value.abort();
|
|
||||||
// 重置状态
|
// 重置状态
|
||||||
isSessionActive.value = false;
|
isSessionActive.value = false;
|
||||||
const msg = chatMsgList.value[currentAIMsgIndex].msg;
|
const msg = chatMsgList.value[currentAIMsgIndex].msg;
|
||||||
@@ -508,9 +501,6 @@
|
|||||||
typeWriterTimer = null;
|
typeWriterTimer = null;
|
||||||
}
|
}
|
||||||
setTimeoutScrollToBottom()
|
setTimeoutScrollToBottom()
|
||||||
// 清空请求引用
|
|
||||||
requestTaskRef.value = null;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
</script>
|
</script>
|
||||||
|
|||||||
@@ -9,13 +9,72 @@ const API = '/agent/assistant/chat';
|
|||||||
* @param {Function} onChunk 回调,每收到一段数据触发
|
* @param {Function} onChunk 回调,每收到一段数据触发
|
||||||
* @returns {Object} 包含Promise和requestTask的对象
|
* @returns {Object} 包含Promise和requestTask的对象
|
||||||
*/
|
*/
|
||||||
function agentChatStream(params, onChunk) {
|
let requestTask = null;
|
||||||
let requestTask;
|
let isAborted = false; // 添加终止状态标志
|
||||||
|
let currentPromiseReject = null; // 保存当前Promise的reject函数
|
||||||
|
let requestId = 0; // 请求ID,用于区分不同的请求
|
||||||
|
|
||||||
|
const stopAbortTask = () => {
|
||||||
|
console.log('🛑 开始强制终止请求...');
|
||||||
|
isAborted = true; // 立即设置终止标志
|
||||||
|
|
||||||
|
// 立即拒绝当前Promise(最强制的终止)
|
||||||
|
if (currentPromiseReject) {
|
||||||
|
console.log('🛑 立即拒绝Promise');
|
||||||
|
currentPromiseReject(new Error('请求已被用户终止'));
|
||||||
|
currentPromiseReject = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (requestTask) {
|
||||||
|
// 先取消所有监听器(关键:必须在abort之前)
|
||||||
|
try {
|
||||||
|
if (requestTask.offChunkReceived) {
|
||||||
|
requestTask.offChunkReceived();
|
||||||
|
console.log('🛑 已取消 ChunkReceived 监听');
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log('🛑 取消 ChunkReceived 监听失败:', e);
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
if (requestTask.offHeadersReceived) {
|
||||||
|
requestTask.offHeadersReceived();
|
||||||
|
console.log('🛑 已取消 HeadersReceived 监听');
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log('🛑 取消 HeadersReceived 监听失败:', e);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 然后终止网络请求
|
||||||
|
try {
|
||||||
|
if (requestTask.abort) {
|
||||||
|
requestTask.abort();
|
||||||
|
console.log('🛑 已终止网络请求');
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log('🛑 终止网络请求失败:', e);
|
||||||
|
}
|
||||||
|
|
||||||
|
requestTask = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 递增请求ID,使旧请求的数据无效
|
||||||
|
requestId++;
|
||||||
|
console.log('🛑 请求强制终止完成,新请求ID:', requestId);
|
||||||
|
}
|
||||||
|
|
||||||
|
const agentChatStream = (params, onChunk) => {
|
||||||
const promise = new Promise((resolve, reject) => {
|
const promise = new Promise((resolve, reject) => {
|
||||||
const token = uni.getStorageSync('token');
|
const token = uni.getStorageSync('token');
|
||||||
let hasError = false;
|
let hasError = false;
|
||||||
|
isAborted = false; // 重置终止状态
|
||||||
|
|
||||||
console.log("发送请求内容: ", params)
|
// 保存当前Promise的reject函数,用于强制终止
|
||||||
|
currentPromiseReject = reject;
|
||||||
|
|
||||||
|
// 为当前请求分配ID
|
||||||
|
const currentRequestId = ++requestId;
|
||||||
|
console.log("🚀 发送请求内容: ", params, "请求ID:", currentRequestId)
|
||||||
// #ifdef MP-WEIXIN
|
// #ifdef MP-WEIXIN
|
||||||
requestTask = uni.request({
|
requestTask = uni.request({
|
||||||
url: BASE_URL + API, // 替换为你的接口地址
|
url: BASE_URL + API, // 替换为你的接口地址
|
||||||
@@ -29,38 +88,62 @@ function agentChatStream(params, onChunk) {
|
|||||||
},
|
},
|
||||||
responseType: 'arraybuffer',
|
responseType: 'arraybuffer',
|
||||||
success(res) {
|
success(res) {
|
||||||
|
if (!isAborted && requestId === currentRequestId) {
|
||||||
|
console.log("✅ 请求成功,ID:", currentRequestId);
|
||||||
resolve(res.data);
|
resolve(res.data);
|
||||||
|
} else {
|
||||||
|
console.log("❌ 请求已过期或终止,忽略success回调,当前ID:", requestId, "请求ID:", currentRequestId);
|
||||||
|
}
|
||||||
},
|
},
|
||||||
fail(err) {
|
fail(err) {
|
||||||
console.log("====> ", JSON.stringify(err))
|
if (!isAborted && requestId === currentRequestId) {
|
||||||
|
console.log("❌ 请求失败,ID:", currentRequestId, "错误:", JSON.stringify(err));
|
||||||
reject(err);
|
reject(err);
|
||||||
|
} else {
|
||||||
|
console.log("❌ 请求已过期或终止,忽略fail回调,当前ID:", requestId, "请求ID:", currentRequestId);
|
||||||
|
}
|
||||||
},
|
},
|
||||||
complete(res) {
|
complete(res) {
|
||||||
if(res.statusCode !== 200) {
|
if (!isAborted && requestId === currentRequestId && res.statusCode !== 200) {
|
||||||
console.log("====> ", JSON.stringify(res))
|
console.log("❌ 请求完成但状态错误,ID:", currentRequestId, "状态:", res.statusCode);
|
||||||
if (onChunk) {
|
if (onChunk) {
|
||||||
onChunk({ error: true, message: '服务器错误', detail: res });
|
onChunk({ error: true, message: '服务器错误', detail: res });
|
||||||
}
|
}
|
||||||
reject(res);
|
reject(res);
|
||||||
|
} else if (requestId !== currentRequestId) {
|
||||||
|
console.log("❌ 请求已过期或终止,忽略complete回调,当前ID:", requestId, "请求ID:", currentRequestId);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
requestTask.onHeadersReceived(res => {
|
requestTask.onHeadersReceived(res => {
|
||||||
console.log('onHeadersReceived', res);
|
// 检查请求是否已终止或过期
|
||||||
|
if (isAborted || requestId !== currentRequestId) {
|
||||||
|
console.log('🚫 Headers已终止或过期,忽略,当前ID:', requestId, '请求ID:', currentRequestId);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log('📡 onHeadersReceived,ID:', currentRequestId, res);
|
||||||
const status = res.statusCode || (res.header && res.header.statusCode);
|
const status = res.statusCode || (res.header && res.header.statusCode);
|
||||||
if (status && status !== 200) {
|
if (status && status !== 200) {
|
||||||
hasError = true;
|
hasError = true;
|
||||||
if (onChunk) {
|
if (onChunk && !isAborted && requestId === currentRequestId) {
|
||||||
onChunk({ error: true, message: `服务器错误(${status})`, detail: res });
|
onChunk({ error: true, message: `服务器错误(${status})`, detail: res });
|
||||||
}
|
}
|
||||||
requestTask.abort && requestTask.abort();
|
if (requestTask && requestTask.abort) {
|
||||||
|
requestTask.abort();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
requestTask.onChunkReceived(res => {
|
requestTask.onChunkReceived(res => {
|
||||||
// 检查请求是否已被中止
|
// 第一道防线:立即检查请求ID和终止状态
|
||||||
if (hasError || requestTask.isAborted) return;
|
if (isAborted || hasError || requestTask === null || requestId !== currentRequestId) {
|
||||||
|
console.log('🚫 数据块已终止或过期,忽略 - 第一道检查,当前ID:', requestId, '请求ID:', currentRequestId);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log("📦 onChunkReceived,ID:", currentRequestId, res)
|
||||||
const base64 = uni.arrayBufferToBase64(res.data);
|
const base64 = uni.arrayBufferToBase64(res.data);
|
||||||
let data = '';
|
let data = '';
|
||||||
try {
|
try {
|
||||||
@@ -69,18 +152,28 @@ function agentChatStream(params, onChunk) {
|
|||||||
// 某些平台可能不支持 atob,可以直接用 base64
|
// 某些平台可能不支持 atob,可以直接用 base64
|
||||||
data = base64;
|
data = base64;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 第二道防线:解析前再次检查
|
||||||
|
if (isAborted || hasError || requestTask === null || requestId !== currentRequestId) {
|
||||||
|
console.log('🚫 数据块已终止或过期,忽略 - 第二道检查,当前ID:', requestId, '请求ID:', currentRequestId);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
const messages = parseSSEChunk(data);
|
const messages = parseSSEChunk(data);
|
||||||
messages.forEach(msg => {
|
messages.forEach(msg => {
|
||||||
if (onChunk) onChunk(msg);
|
// 第三道防线:每个消息处理前都检查
|
||||||
|
if (onChunk && !isAborted && !hasError && requestTask !== null && requestId === currentRequestId) {
|
||||||
|
console.log(`parseSSEChunk ${currentRequestId}:`, msg)
|
||||||
|
onChunk(msg);
|
||||||
|
} else {
|
||||||
|
console.log('🚫 消息已终止或过期,忽略处理,当前ID:', requestId, '请求ID:', currentRequestId);
|
||||||
|
}
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
// #endif
|
// #endif
|
||||||
});
|
});
|
||||||
|
|
||||||
return {
|
return promise
|
||||||
promise,
|
|
||||||
requestTask
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// window.atob兼容性处理
|
// window.atob兼容性处理
|
||||||
@@ -152,4 +245,4 @@ function parseSSEChunk(raw) {
|
|||||||
return results;
|
return results;
|
||||||
}
|
}
|
||||||
|
|
||||||
export { agentChatStream }
|
export { agentChatStream, stopAbortTask }
|
||||||
|
|||||||
Reference in New Issue
Block a user