const my = { socket: undefined, reconnecting: undefined, listeners: {}, } export default { initSocket (url, reconnect = false) { if (!my.socket || (my.socket.readyState !== my.socket.OPEN && typeof url === 'string')) { console.log(`WebSocket is connecting to ${url}...`) my.socket = uni.connectSocket({ url: url.replace(/^http/, 'ws'), complete: () => {}, }) my.socket.onOpen((res) => { console.log('WebSocket onOpen: ', res) clearInterval(my.reconnecting) delete my.reconnecting if (reconnect && uni.getStorageSync('_passtoken')) { console.log('Reporting owner for reconnecting socket') my.socket.send({ data: JSON.stringify({ skevent: 'SOCKET_OWNER_RECONNECT', _passtoken: uni.getStorageSync('_passtoken') }) }) } }) my.socket.onClose((res) => { console.log('Websocket onClose: ', res) if (!my.reconnecting) my.reconnecting = setInterval(() => { console.log(new Date().toJSON(), 'WebSocket reconnecting...') this.initSocket(url, true) }, 5000) // 每5秒尝试重连 }) my.socket.onError((err) => { console.log('Websocket onError: ', err) }) my.socket.onMessage(({ data }) => { // 在这里统一分发消息(用户端通常不需要返回结果给服务器,因此不用 rpc 模式,而用 event 模式。 try { let { skevent, ...apiWhat } = JSON.parse(data) console.log('WebSocket onMessage for skevent: ', skevent) let listeners = my.listeners[skevent] || [] for (let listener of listeners) { listener(apiWhat) } } catch (exception) { console.log(new Date().toJSON(), 'unknown message', data, exception) return } }) } return this }, closeSocket () { if (my.socket) my.socket.close() setTimeout(() => { clearInterval(my.reconnecting) delete my.reconnecting }, 2000) }, initListener (skevent, listener) { // 当该 skevent 尚不具有任何 listener 时,添加本 listener my.listeners[skevent]?.length > 0 || this.addListener(skevent, listener) return this }, addListener (skevent, listener) { if (Array.isArray(my.listeners[skevent]) && typeof listener === 'function') { my.listeners[skevent].push(listener) } else { my.listeners[skevent] = [listener] } return this }, countListener (skevent) { if (Array.isArray(my.listeners[skevent])) { return my.listeners[skevent].length } return 0 }, sendObject (dataObj) { if (my.socket && my.socket.readyState === my.socket.OPEN) { my.socket.send({ data: typeof dataObj !== 'string' ? JSON.stringify(dataObj) : dataObj, }) } }, }