114 lines
3.0 KiB
JavaScript
114 lines
3.0 KiB
JavaScript
|
import axios from 'axios'
|
|||
|
|
import { fetchEventSource } from '@microsoft/fetch-event-source'
|
|||
|
|
|
|||
|
|
const service = axios.create({
|
|||
|
|
baseURL: process.env.VITE_APP_BASE_API,
|
|||
|
|
timeout: 30000
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
// 请求拦截器
|
|||
|
|
service.interceptors.request.use(
|
|||
|
|
config => {
|
|||
|
|
// 如果配置中已经包含了 Authorization header，则使用配置中的值
|
|||
|
|
if (!config.headers['Authorization']) {
|
|||
|
|
const token = localStorage.getItem('token')
|
|||
|
|
if (token) {
|
|||
|
|
config.headers['Authorization'] = `Bearer ${token}`
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
config.headers['Content-Type'] = 'application/json'
|
|||
|
|
return config
|
|||
|
|
},
|
|||
|
|
error => {
|
|||
|
|
return Promise.reject(error)
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 响应拦截器
|
|||
|
|
service.interceptors.response.use(
|
|||
|
|
response => {
|
|||
|
|
return response.data
|
|||
|
|
},
|
|||
|
|
error => {
|
|||
|
|
return Promise.reject(error)
|
|||
|
|
}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// SSE 请求方法
|
|||
|
|
export const sseRequest = (config) => {
|
|||
|
|
return new Promise((resolve) => {
|
|||
|
|
try {
|
|||
|
|
const url = new URL(`${config.url}`, window.location.origin)
|
|||
|
|
const ctrl = new AbortController()
|
|||
|
|
let messageQueue = []
|
|||
|
|
let resolveQueue = null
|
|||
|
|
|
|||
|
|
const messageGenerator = async function* () {
|
|||
|
|
const source = new fetchEventSource(url.toString(), {
|
|||
|
|
method: 'POST',
|
|||
|
|
headers: {
|
|||
|
|
'Authorization': config.headers?.Authorization || `Bearer ${localStorage.getItem('token')}`,
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Accept': 'text/event-stream'
|
|||
|
|
},
|
|||
|
|
body: JSON.stringify(config.params),
|
|||
|
|
signal: ctrl.signal,
|
|||
|
|
openWhenHidden: true,
|
|||
|
|
onmessage: (event) => {
|
|||
|
|
try {
|
|||
|
|
const data = JSON.parse(event.data)
|
|||
|
|
if (resolveQueue) {
|
|||
|
|
resolveQueue(data)
|
|||
|
|
resolveQueue = null
|
|||
|
|
} else {
|
|||
|
|
messageQueue.push(data)
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('Error parsing SSE message:', error)
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
onerror: (error) => {
|
|||
|
|
console.error('EventSource failed:', error)
|
|||
|
|
ctrl.abort()
|
|||
|
|
if (resolveQueue) {
|
|||
|
|
resolveQueue(null)
|
|||
|
|
resolveQueue = null
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
onclose: () => {
|
|||
|
|
console.log('Connection closed')
|
|||
|
|
if (resolveQueue) {
|
|||
|
|
resolveQueue(null)
|
|||
|
|
resolveQueue = null
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
while (true) {
|
|||
|
|
if (messageQueue.length > 0) {
|
|||
|
|
yield messageQueue.shift()
|
|||
|
|
} else {
|
|||
|
|
const message = await new Promise(resolve => {
|
|||
|
|
resolveQueue = resolve
|
|||
|
|
})
|
|||
|
|
if (message === null) {
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
yield message
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
} finally {
|
|||
|
|
ctrl.abort()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
resolve(messageGenerator())
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('SSE request setup failed:', error)
|
|||
|
|
resolve((async function* () {})())
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export default service
|