Mhbcoin_PC/src/api/custom/Socket.js

class Socket {

    constructor(ws, ...args) {
        // 初始化socket
        if (ws.constructor == WebSocket) {
            this.socket = ws;
        } else {
            this.socket = new WebSocket(ws);
        }
        // this.socket.binaryType = 'arraybuffer';

        this.doOpen();


        this.readyState = this.socket.readyState;

        // 订阅/发布模型
        this._events = {
            // 订阅的事件 ： 发布的方法

        };

        // 定时验证的标识符
        this.heartBeatTimer = null;

    }

    // 执行socket并发布事件
    doOpen() {

        this.afterOpenEmit = [];

        // 执行socket连接 并初始化验证请求
        this.socket.addEventListener("open", evt => {
            // 连接状态的标识符
            this.readyState = this.socket.readyState;
            this.onOpen(evt)
        });

        // 接收socket数据
        this.socket.addEventListener("message", evt => {
            // 连接状态的标识符
            this.readyState = this.socket.readyState;
            this.onMessage(evt)
        });

        // 关闭socket连接
        this.socket.addEventListener("close", evt => {
            // 连接状态的标识符
            this.readyState = this.socket.readyState;
            this.onClose(evt)
        });

        // 请求发生错误
        this.socket.addEventListener("error", err => {
            // 连接状态的标识符
            this.readyState = this.socket.readyState;
            this.onError(err)
        });

    }

    // 发布后通知订阅者
    Notify(entry) {
        // 检查是否有订阅者 返回队列
        const cbQueue = this._events[entry.Event];
        if (cbQueue && cbQueue.length) {
            for (let callback of cbQueue) {
                 
                if (callback instanceof Function) callback(entry.Data);
            }
        }
    }

    // 请求数据的方法
    onOpen(evt) {

        // 每隔20s检查连接
        // this.heartBeatTimer = setInterval(() => this.send({
        //     'cmd': 'ping',
        //     'args': ''
        // }), 20000);

        // 通知订阅
        this.Notify({ Event: 'open', Data: evt });
    }

    /**
     * 订阅所有的数据
     * @param {array|object} datas 订阅参数集合
     */
    send(datas) {
        if (datas.constructor != Array) {
            datas = [datas];
        }

        for (let item of datas) {
            this.socket.send(JSON.stringify(item));
        }
    }


    onMessage(evt) {

        try {

            // 解析推送的数据
            const data = JSON.parse(evt.data);

            // 通知订阅者
            this.Notify({
                Event: 'message',
                Data: data
            });

        } catch (err) {
            console.error(' >> Data parsing error:', err.message);

            // 通知订阅者
            this.Notify({
                Event: 'error',
                Data: err
            });
        }
    }

    // 添加事件监听
    on(name, handler) {
        this.subscribe(name, handler);
    }

    // 取消订阅事件
    off(name, handler) {
        this.unsubscribe(name, handler);
    }

    // 订阅事件的方法
    subscribe(name, handler) {
        // debugger;
        if (this._events.hasOwnProperty(name)) {
            this._events[name].push(handler); // 追加事件
        } else {
            this._events[name] = [handler]; // 添加事件
        }
    }

    // 取消订阅事件
    unsubscribe(name, handler) {

        let start = this._events[name].findIndex(item => item === handler);

        // 删除该事件
        this._events[name].splice(start, 1);

    }

    checkOpen() {

        return this.readyState >= 1;
    }

    onClose(evt) {
        this.Notify({ Event: 'close', Data: evt });
    }


    onError(err) {
        this.Notify({ Event: 'error', Data: err });
    }

    emit(data) {
        return new Promise((resolve) => {
            this.send(JSON.stringify(data));
            this.on('message', function (data) {
                resolve(data);
            });
        });
    }

    doClose() {
        this.socket.close();
    }

    destroy() {
        if (this.heartBeatTimer) {
            clearInterval(this.heartBeatTimer);
            this.heartBeatTimer = null;
        }
        this.doClose();
        this._events = {};
        this.readyState = 0;
        this.socket = null;
    }
}

export default Socket