socketTask.js

let timer = null; //心跳计时器
let debug = false; //debug输出log
let retimer = null; //重连计时器
let pingCount = 0; //心跳次数
let pingCountMax = 5; //最大心跳次数超过重连
let pingTimeout = 5000; //心跳间隔时间(毫秒)
let connectTimeout = 2000; //重连时间(毫秒)
let socketTask = null; //ws
// 连接WebSocket
export function connectSocket() {
    debug && console.log(socketTask);
    var token = uni.getStorageSync('Authorization');
    if (!token || socketTask) {
        console.log('开启socketTask失败')
        return
    }
    socketTask = uni.connectSocket({
        url: 'wss://im-api.q3z3.com/ws?Authorization=' + uni.getStorageSync('Authorization'),//WebSocket地址
        // url: 'ws://192.168.0.200:8080/ws?Authorization=' + uni.getStorageSync('Authorization'),//WebSocket地址
        complete: () => {}
    });
    socketTask.onOpen(res => {
        console.log('WebSocket连接已打开！');
        debug && console.log(socketTask);
        // 发送心跳
        socketTask.send({
            data: 'isConnact',
            success: res => {
                if (res.errMsg == 'sendSocketMessage:ok') {
                    debug && console.log('WebSocket心跳ping');
                }
            }
        })
        // 定时心跳
        timer = setInterval(() => {
            // ping超过5次未响应则重连
            pingCount++
            debug && console.log('ping次数'+pingCount+'/'+pingCountMax);
            if (pingCount >= pingCountMax) {
                clearSocketTask()
                reConnectSocket()
                return
            }
            socketTask.send({
                data: 'isConnact',
                success: res => {
                    if (res.errMsg == 'sendSocketMessage:ok') {
                        debug && console.log('WebSocket心跳ping');
                    }
                }
            })
        }, pingTimeout)
    })
    // 监听接收
    socketTask.onMessage(res => {
        if (socketTask && token) {
            if(res.data=='ok'){
                pingCount = 0;
                return
            }
            var data = JSON.parse(res.data);
       
            console.log('WebSocket接收消息！');
        }
    })
    // 监听关闭
    socketTask.onClose((res) => {
        debug && console.log(socketTask);
        console.log('WebSocket连接已关闭！');
        if (!socketTask) {
            console.log('无需操作')
            return
        }
        if (socketTask.readyState !== 1) {
            console.log('需要重新连接')
            clearSocketTask()
            reConnectSocket()
        }
    })
    // 监听异常
    socketTask.onError(res => {
        debug && console.log(socketTask);
        console.log('WebSocket连接异常！');
        if (!socketTask) {
            console.log('无需操作')
            return
        }
        if (socketTask.readyState !== 1) {
            console.log('需要重新连接')
            clearSocketTask()
            reConnectSocket()
        }
    });
}

// 清理WebSocket
export function clearSocketTask() {
    clearInterval(timer)
    clearTimeout(retimer)
    pingCount = 0;
    if (socketTask) {
        socketTask.close()
        socketTask = null
        console.log('主动关闭WebSocket！');
    }
}

// 重新连接WebSocket
export function reConnectSocket() {
    retimer = setTimeout(() => {
        connectSocket()
    }, connectTimeout)
}

main.js 全局引入

import * as socketTask from "@/common/socketTask.js";

Vue.prototype.$socketTask = socketTask;

使用 onLaunch

this.$socketTask.connectSocket()

401登录过期时候

import {clearSocketTask} from "@/common/socketTask.js";

使用

clearSocketTask()