feat: 请求终止的逻辑调整

This commit is contained in:
2025-08-08 09:08:07 +08:00
parent a432c32fb5
commit 1af6fd2e5f
2 changed files with 141 additions and 58 deletions

View File

@@ -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>

View File

@@ -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('📡 onHeadersReceivedID:', 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("📦 onChunkReceivedID:", 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 }