155 lines
3.6 KiB
JavaScript
155 lines
3.6 KiB
JavaScript
import config from '@/config'
|
||
|
||
/**
|
||
* 测量数据 WebSocket（与 l2/socket/index.vue 类似）
|
||
* 默认订阅 type=track_measure
|
||
*/
|
||
export function createMeasureSocket({
|
||
type = 'track_measure',
|
||
onOpen,
|
||
onClose,
|
||
onError,
|
||
onMessage
|
||
} = {}) {
|
||
let socket = null
|
||
let manualClose = false
|
||
// 保存回调函数的引用，允许清空
|
||
let callbacks = {
|
||
onOpen,
|
||
onClose,
|
||
onError,
|
||
onMessage
|
||
}
|
||
|
||
const wsBase = (config.wsUrl || config.baseUrl || '').replace(/^http/, 'ws')
|
||
const url = `${wsBase}/websocket?type=${type}`
|
||
console.log(url)
|
||
|
||
function connect() {
|
||
manualClose = false
|
||
|
||
// 使用 uni.connectSocket 建立连接（部分运行环境可能不支持 WebSocket，会直接抛异常）
|
||
try {
|
||
socket = uni.connectSocket({
|
||
url,
|
||
success() {
|
||
console.log('connectSocket 调用成功，等待 onOpen 回调')
|
||
},
|
||
fail(err) {
|
||
console.error('connectSocket 调用失败:', err)
|
||
callbacks.onError && callbacks.onError(err)
|
||
}
|
||
})
|
||
} catch (err) {
|
||
console.error('connectSocket 执行异常（可能环境不支持 WebSocket）:', err)
|
||
callbacks.onError && callbacks.onError(err)
|
||
return
|
||
}
|
||
|
||
if (!socket) {
|
||
console.error('connectSocket 未返回有效 socketTask')
|
||
return
|
||
}
|
||
|
||
// 正确的事件注册方式：socketTask.onOpen / onMessage / onError / onClose
|
||
socket.onOpen((res) => {
|
||
console.log('WebSocket 已打开', res)
|
||
// 检查回调是否已被清空
|
||
if (callbacks && callbacks.onOpen) {
|
||
try {
|
||
callbacks.onOpen(res)
|
||
} catch (e) {
|
||
// 忽略回调错误
|
||
}
|
||
}
|
||
})
|
||
|
||
socket.onMessage((evt) => {
|
||
// H5 为 evt.data，小程序为 evt.data，统一兼容
|
||
const data = evt && (evt.data || evt)
|
||
// 检查回调是否已被清空
|
||
if (callbacks && callbacks.onMessage) {
|
||
try {
|
||
callbacks.onMessage(data)
|
||
} catch (e) {
|
||
// 忽略回调错误，可能是组件已销毁
|
||
}
|
||
}
|
||
})
|
||
|
||
socket.onError((err) => {
|
||
console.error('WebSocket 发生错误:', err)
|
||
// 检查回调是否已被清空
|
||
if (callbacks && callbacks.onError) {
|
||
try {
|
||
callbacks.onError(err)
|
||
} catch (e) {
|
||
// 忽略回调错误
|
||
}
|
||
}
|
||
})
|
||
|
||
socket.onClose((evt) => {
|
||
console.log('WebSocket 已关闭', evt)
|
||
// 检查回调是否已被清空
|
||
if (callbacks && callbacks.onClose) {
|
||
try {
|
||
callbacks.onClose(evt)
|
||
} catch (e) {
|
||
// 忽略回调错误
|
||
}
|
||
}
|
||
if (!manualClose && callbacks) {
|
||
setTimeout(connect, 3000)
|
||
}
|
||
})
|
||
}
|
||
|
||
function close() {
|
||
manualClose = true
|
||
// 清空所有回调，防止在关闭过程中或关闭后触发回调
|
||
callbacks = {
|
||
onOpen: null,
|
||
onClose: null,
|
||
onError: null,
|
||
onMessage: null
|
||
}
|
||
if (socket) {
|
||
try {
|
||
socket.close(1000, 'client close')
|
||
} catch (e) {
|
||
// ignore
|
||
}
|
||
socket = null
|
||
}
|
||
}
|
||
|
||
function send(data) {
|
||
if (!socket) {
|
||
console.error('WebSocket 未初始化，无法发送消息')
|
||
return
|
||
}
|
||
|
||
// uni-app 的 socketTask.send 使用对象形式
|
||
socket.send({
|
||
data: JSON.stringify(data),
|
||
fail(err) {
|
||
console.error('发送 WebSocket 消息失败:', err)
|
||
}
|
||
})
|
||
}
|
||
|
||
// 清空回调函数的方法
|
||
function clearCallbacks() {
|
||
callbacks = {
|
||
onOpen: null,
|
||
onClose: null,
|
||
onError: null,
|
||
onMessage: null
|
||
}
|
||
}
|
||
|
||
return { connect, close, send, clearCallbacks }
|
||
}
|
||
|