mayfly-go/mayfly_go_web/src/common/SocketBuilder.ts

class SocketBuilder {
    websocket: WebSocket;

    constructor(url: string) {
        if (typeof WebSocket === 'undefined') {
            throw new Error('不支持websocket');
        }
        if (!url) {
            throw new Error('websocket url不能为空');
        }
        this.websocket = new WebSocket(url);
    }

    static builder(url: string) {
        return new SocketBuilder(url);
    }

    open(onopen: any) {
        this.websocket.onopen = onopen;
        return this;
    }

    error(onerror: any) {
        this.websocket.onerror = onerror;
        return this;
    }

    message(onmessage: any) {
        this.websocket.onmessage = onmessage;
        return this;
    }

    close(onclose: any) {
        this.websocket.onclose = onclose;
        return this;
    }

    build() {
        return this.websocket;
    }
}

export default SocketBuilder;