feat: add netty WebSocket client

Signed-off-by: MiaoWoo <admin@yumc.pw>
2020-12-17 17:00:27 +08:00
parent f42d5d12ed
commit 48cc6325b8
6 changed files with 400 additions and 0 deletions
--- a/packages/websocket/src/client/transport.ts
+++ b/packages/websocket/src/client/transport.ts
@@ -0,0 +1,100 @@
+import { EventEmitter } from 'events'
+import { WebSocket } from './index'
+import { CloseEvent, ErrorEvent, Event, MessageEvent, WebSocketHeader } from './interface'
+
+export abstract class Transport extends EventEmitter {
+    protected _url: string
+    protected _state: number = WebSocket.CONNECTING
+    protected _protocol: string
+    protected _headers: WebSocketHeader = {}
+
+    constructor(uri: string, subProtocol: string = '', headers: WebSocketHeader = {}) {
+        super()
+        this._url = uri
+        this._protocol = subProtocol
+        this._headers = headers
+    }
+
+    get id() {
+        return this.getId()
+    }
+
+    get protocol() {
+        return this._protocol
+    }
+
+    get readyStatus() {
+        return this._state
+    }
+
+    set readyStatus(state: number) {
+        this._state = state
+    }
+
+    connect() {
+        try {
+            this.doConnect()
+        } catch (error) {
+            this.onerror({ error })
+        }
+    }
+
+    send(text: string) {
+        try {
+            this.doSend(text)
+        } catch (error) {
+            this.onerror({ error })
+        }
+    }
+
+    close(code: number = 0, reason: string = '') {
+        if (this.readyStatus != WebSocket.CLOSING && this.readyStatus != WebSocket.CLOSED) {
+            this.readyStatus = WebSocket.CLOSING
+            try {
+                this.onclose({ code, reason })
+                this.doClose(code, reason)
+            } catch (error) {
+                this.onerror({ error })
+            }
+        } else {
+            console.debug(`${this.id} call close but state is ${this.readyStatus}`)
+        }
+    }
+
+    onconnection(event: Event) {
+        this._state == WebSocket.CONNECTING
+        this.emit('connecting', event)
+    }
+
+    onconnect(event: Event) {
+        console.debug(`${this.id} call onconnect`)
+        if (this.readyStatus != WebSocket.OPEN) {
+            this.readyStatus = WebSocket.OPEN
+            this.emit('open', event)
+        } else {
+            console.debug(`${this.id} call onconnect but state is ${this.readyStatus}`)
+        }
+    }
+
+    onmessage(event: MessageEvent) {
+        this.emit('message', event)
+    }
+
+    onerror(event: ErrorEvent) {
+        this.emit('error', event)
+    }
+
+    onclose(event: CloseEvent) {
+        if (this.readyStatus != WebSocket.CLOSED) {
+            this.readyStatus = WebSocket.CLOSED
+            this.emit('close', event)
+            this.removeAllListeners()
+        } else {
+            console.debug(`${this.id} call onclose but state is ${this.readyStatus} CloseEvent[code: ${event.code}, reason: ${event.reason}]`)
+        }
+    }
+    abstract getId()
+    abstract doConnect()
+    abstract doSend(text: string)
+    abstract doClose(code: number, reason: string)
+}