1 function longSock(url, fn, intro = ‘‘) { 2 let lockReconnect = false //避免重复连接 3 let timeoutFlag = true 4 let timeoutSet = null 5 let reconectNum = 0 6 const timeout = 30000 //超时重连间隔 7 let ws 8 function reconnect() { 9 if (lockReconnect) return10 lockReconnect = true11 //没连接上会一直重连,设置延迟避免请求过多12 if (reconectNum < 3) {13 setTimeout(function() {14 timeoutFlag = true15 createWebSocket()16 console.info(`${intro}正在重连第${reconectNum + 1}次`)17 reconectNum++18 lockReconnect = false19 }, 5000) //这里设置重连间隔(ms)20 }21 }22 //心跳检测23 const heartCheck = {24 timeout: 5000, //毫秒25 timeoutObj: null,26 serverTimeoutObj: null,27 reset: function() {28 clearInterval(this.timeoutObj)29 clearTimeout(this.serverTimeoutObj)30 return this31 },32 start: function() {33 const self = this34 let count = 035 this.timeoutObj = setInterval(() => {36 if (count < 3) {37 if (ws.readyState === 1) {38 ws.send(‘HeartBeat‘)39 console.info(`${intro}HeartBeat第${count + 1}次`)40 }41 count++42 } else {43 clearInterval(this.timeoutObj)44 count = 045 if (ws.readyState === 0 && ws.readyState === 1) {46 ws.close()47 }48 }49 }, self.timeout)50 }51 }52 const createWebSocket = () => {53 console.info(`${intro}创建11`)54 timeoutSet = setTimeout(() => {55 if (timeoutFlag && reconectNum < 3) {56 console.info(`${intro}重连22`)57 reconectNum++58 createWebSocket()59 }60 }, timeout)61 ws = new WebSocket(url)62 63 ws.onopen = () => {64 reconectNum = 065 timeoutFlag = false66 clearTimeout(timeoutSet)67 heartCheck.reset().start()68 }69 ws.onmessage = evt => {70 heartCheck.reset().start()71 // console.info(evt);72 if(evt.data === ‘HeartBeat‘) return73 fn(evt, ws)74 }75 ws.onclose = e => {76 console.info(`${intro}关闭11`, e.code)77 if (e.code !== 1000) {78 timeoutFlag = false79 clearTimeout(timeoutSet)80 reconnect()81 } else {82 clearInterval(heartCheck.timeoutObj)83 clearTimeout(heartCheck.serverTimeoutObj)84 }85 }86 ws.onerror = function() {87 console.info(`${intro}错误11`)88 reconnect() //重连89 }90 }91 createWebSocket()92 return ws93 }94 export default longSock
1 //方法调用 2 const handler = (evt, ws) => { 3 //evt 是 websockett数据 4 //ws 是请求名称,方便关闭websocket 5 } 6 7 const wssCenter = createSocket(`wss://url`, handler, ‘接待中心-小卡片‘) 8 9 10 11 wssCenter.close(); //断开连接