99 lines
2.7 KiB
TypeScript
99 lines
2.7 KiB
TypeScript
import { message } from 'antd';
|
|
import i18n from '@/i18n'
|
|
import { cookieUtils } from './request'
|
|
const API_PREFIX = '/api'
|
|
|
|
export interface SSEMessage {
|
|
event?: string
|
|
data?: string | object
|
|
}
|
|
export function parseSSEToJSON(sseString: string) {
|
|
const events: SSEMessage[] = []
|
|
const lines = sseString.trim().split('\n')
|
|
|
|
let currentEvent: SSEMessage = {}
|
|
|
|
try {
|
|
for (const line of lines) {
|
|
if (line.startsWith('event:')) {
|
|
if (Object.keys(currentEvent).length > 0) {
|
|
events.push(currentEvent)
|
|
currentEvent = {}
|
|
}
|
|
currentEvent.event = line.substring(6).trim()
|
|
} else if (line.startsWith('data:')) {
|
|
const dataStr = line.substring(5).trim()
|
|
if (dataStr) {
|
|
try {
|
|
// 尝试解析为 JSON
|
|
currentEvent.data = JSON.parse(dataStr)
|
|
} catch {
|
|
// JSON 解析失败时，检查是否是被转义的 JSON 字符串
|
|
try {
|
|
const unescaped = dataStr.replace(/"/g, '"').replace(/&/g, '&')
|
|
currentEvent.data = JSON.parse(unescaped)
|
|
} catch {
|
|
// 如果仍然失败，保存为原始字符串
|
|
currentEvent.data = dataStr
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (Object.keys(currentEvent).length > 0) {
|
|
events.push(currentEvent)
|
|
}
|
|
|
|
return events
|
|
} catch (error) {
|
|
console.error('Parse stream error:', error)
|
|
return []
|
|
}
|
|
}
|
|
|
|
|
|
export const handleSSE = async (url: string, data: any, onMessage?: (data: SSEMessage[]) => void, config = { headers: {} }) => {
|
|
try {
|
|
const token = cookieUtils.get('authToken');
|
|
const response = await fetch(`${API_PREFIX}${url}`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
'Authorization': `Bearer ${token}`,
|
|
...config.headers,
|
|
},
|
|
body: JSON.stringify(data)
|
|
});
|
|
|
|
const { status } = response
|
|
|
|
switch(status) {
|
|
case 401:
|
|
if (url?.includes('/public')) {
|
|
return message.warning(i18n.t('common.publicApiCannotRefreshToken'));
|
|
}
|
|
window.location.href = `/#/login`;
|
|
break;
|
|
default:
|
|
if (!response.body) throw new Error('No response body');
|
|
|
|
const reader = response.body.getReader();
|
|
const decoder = new TextDecoder();
|
|
|
|
while (true) {
|
|
const { done, value } = await reader.read();
|
|
if (done) break;
|
|
|
|
const chunk = decoder.decode(value, { stream: true });
|
|
if (onMessage) {
|
|
onMessage(parseSSEToJSON(chunk) ?? {});
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
} catch (error) {
|
|
console.error('Request failed:', error);
|
|
throw error;
|
|
}
|
|
} |