websocket封装

最新推荐文章于 2024-08-21 16:48:02 发布

原创最新推荐文章于 2024-08-21 16:48:02 发布 · 1.4k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#websocket

js 专栏收录该内容

37 篇文章

订阅专栏

该博客介绍了WebSocketPlugin类的实现，用于管理和重连WebSocket连接。类中包含连接、断开、重连及消息发送的方法。在连接关闭或发生异常时，会自动尝试重新连接。使用时，可以通过配置参数启动连接、发送消息和关闭连接，所有回调在指定的回调函数中处理。

//封装

class SocketPlugin {
    constructor(param) {
        this.websocket = null
        this.isConnect = false
        this.timeoutNum = null
        this.isActivelyClose = false
        this.param = param
    }


    connect() {
        this.websocket = new WebSocket(this.param.url)
        this.initSocket(this.param)
    }


    initSocket(param) {
        this.isActivelyClose = false


        this.websocket.onclose = e => {
            console.log('websocket连接关闭~' + this.param.url)
            this.isConnect = false
            // 如果手动关闭则不进行重连
            if (!this.isActivelyClose) {
                this.reconnectSocket(param)
            }
        }


        this.websocket.onerror = e => {
            console.log('websocket发生异常~' + this.param.url + e)
            this.reconnectSocket(param)
        }


        this.websocket.onopen = () => {
            console.log('websocket已连接~ ' + this.param.url)
            this.isConnect = true
            if (param.hasOwnProperty('msg')) {
                this.send(param.msg || '')
            }
        }


        this.websocket.onmessage = e => {
            param.callback(JSON.parse(e.data))
        }
    }


    reconnectSocket(param) {
        if (this.isConnect === true) {
            return false
        }
        console.log('websocket 重新连接~ ')
        this.isConnect = true
        this.timeoutNum && clearTimeout(this.timeoutNum)
        this.timeoutNum = setTimeout(() => {
            this.connect(param)
            this.isConnect = false
        }, 1000)
    }


    /**
     * // websocket连接状态下才能进行send
     * @param {*} msg
     * 向服务send的消息
     */
    send(msg) {
        this.websocket.send(JSON.stringify(msg))
    }


    close() {
        this.isActivelyClose = true
        if (this.websocket) {
            this.websocket.close()
        }
    }
}


export default SocketPlugin

//启动与调用/关闭

let  socketConfig = {
        url: '/ints/websocket/test',
        callback: this.getSocketMsg,
        msg: {
          fanId: '01'
        }
     }
let  testSocket = new SocketPlugin(socketConfig)
// 初始化 启动连接，一般在登录时
testSocket.connect()
ps:如果没有重新new socketplugin，所有回调都在getSocketMsg中


//首次启动连接之后，后续如果再次调用，掉用send方法
// 发送消息
testSocket.send(socketConfig.msg)


// 关闭 一般在退出登录时
testSocket.close()