circlecloud/ms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

feat: optimize websocket client

Browse Source
This commit is contained in:
MiaoWoo 2023-08-08 18:06:34 +08:00
parent 9126ec8035
commit 9cfac1672a
9 changed files with 170 additions and 124 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
README.MD
View File

@ -8,27 +8,28 @@
```txt ```txt
└─packages └─packages
├─api 全平台兼容的接口 ├─api 全平台兼容的接口
├─core 核心代码 用于引导加载 ├─core 核心代码 用于引导加载
├─common 公共类库代码 例如 http reflect 模块 ├─common 公共类库代码 例如 http reflect 模块
├─client NodeJS的Minecraft客户端 用于调试插件 ├─compile 编译器相关功能
├─container IOC容器 用于注入具体实现 ├─client NodeJS 的 Minecraft 客户端 已迁移至 ms-client
├─ployfill Nashorn 的一些自定义增强 ├─container IOC容器 用于注入具体实现
├─nashorn Nashorn 的类型定义 ├─database 数据库相关功能
├─bungee BungeeCordAPI内部实现 ├─protocol 协议处理相关功能
├─bukkit BukkitAPI内部实现 ├─service 服务相关功能
├─sponge SpongeAPI内部实现 ├─i18n 多语言环境相关支持
├─nukkit NukkitAPI内部实现 ├─polyfill Nashorn 的一些自定义增强
├─plugin 插件管理器 ├─nashorn Nashorn 的类型定义
├─websocket Netty的WebSocket注入 ├─nodejs NodeJS 的部分 Java 实现
├─type Java的类型定义 ├─bungee BungeeCordAPI 内部实现
| ├─bungee BungeeCord类型定义 ├─bukkit BukkitAPI 内部实现
| ├─bukkit Bukkit类型定义 ├─sponge SpongeAPI 内部实现
| ├─sponge Sponge类型定义 ├─nukkit NukkitAPI 内部实现
| └─nukkit Nukkit类型定义 ├─molang MoLang 解析库
└─plugins 这里当然是插件啦 ├─qrcode 二维码相关类库
├─bungee 只兼容BungeeCord的插件 ├─plugin 插件管理器
├─bukkit 只兼容Bukkit的插件 ├─websocket WebSocket 相关实现
├─sponge 只兼容Sponge的插件 | ├─client 基于 Netty 的 WebSocket 客户端
└─nukkit 只兼容Nukkit的插件 | └─server 基于 Netty 的 WebSocket 服务端
└─type 类型定义 已迁移到 @javatypes
``` ```

37
packages/api/src/command.ts
View File

@ -5,6 +5,13 @@ import { plugin } from './plugin'
export namespace command { export namespace command {
@injectable() @injectable()
export abstract class Command { export abstract class Command {
/**
* first time script engine need optimize jit code
* so ignore first slow exec notify
*/
private cacheSlowCommandKey = {};
private cacheSlowCompleteKey = {};
/** /**
* 注册插件命令 * 注册插件命令
* @param plugin 插件 * @param plugin 插件
@ -48,6 +55,8 @@ export namespace command {
let result = executor(sender, command, Java.from(args)) let result = executor(sender, command, Java.from(args))
let cost = Date.now() - time let cost = Date.now() - time
if (cost > global.ScriptSlowExecuteTime) { if (cost > global.ScriptSlowExecuteTime) {
let commandKey = `${plugin.description.name}-${command}-${sender.name}`
if (!this.cacheSlowCommandKey[commandKey]) { return this.cacheSlowCommandKey[commandKey] = cost }
console.i18n("ms.api.command.execute.slow", { console.i18n("ms.api.command.execute.slow", {
player: sender.name, player: sender.name,
plugin: plugin.description.name, plugin: plugin.description.name,
@ -58,23 +67,17 @@ export namespace command {
} }
return result return result
} catch (ex: any) { } catch (ex: any) {
console.i18n("ms.api.command.execute.error", { let message = i18n.translate("ms.api.command.execute.error", {
player: sender.name, player: sender.name,
plugin: plugin.description.name, plugin: plugin.description.name,
command, command,
args: Java.from(args).join(' '), args: Java.from(args).join(' '),
ex ex
}) })
console.console(message)
console.ex(ex) console.ex(ex)
if (sender.name != 'CONSOLE') { if (sender.name != 'CONSOLE') {
console.sender(sender, [i18n.translate("ms.api.command.execute.error", { console.sender(sender, [message, ...console.stack(ex)])
player: sender.name,
plugin: plugin.description.name,
command,
args: Java.from(args).join(' '),
ex
}),
...console.stack(ex)])
} }
return true return true
} }
@ -89,6 +92,8 @@ export namespace command {
let result = this.copyPartialMatches(complete, token) let result = this.copyPartialMatches(complete, token)
let cost = Date.now() - time let cost = Date.now() - time
if (cost > global.ScriptSlowExecuteTime) { if (cost > global.ScriptSlowExecuteTime) {
let completerKey = `${plugin.description.name}-${command}-${sender.name}`
if (!this.cacheSlowCompleteKey[completerKey]) { return this.cacheSlowCompleteKey[completerKey] = cost }
console.i18n("ms.api.command.tab.completer.slow", { console.i18n("ms.api.command.tab.completer.slow", {
player: sender.name, player: sender.name,
plugin: plugin.description.name, plugin: plugin.description.name,
@ -99,25 +104,17 @@ export namespace command {
} }
return result return result
} catch (ex: any) { } catch (ex: any) {
console.i18n("ms.api.command.tab.completer.error", { let message = i18n.translate("ms.api.command.tab.completer.error", {
player: sender.name, player: sender.name,
plugin: plugin.description.name, plugin: plugin.description.name,
command, command,
args: Java.from(args).join(' '), args: Java.from(args).join(' '),
ex ex
}) })
console.console(message)
console.ex(ex) console.ex(ex)
if (sender.name != 'CONSOLE') { if (sender.name != 'CONSOLE') {
console.sender(sender, [ console.sender(sender, [message, ...console.stack(ex)])
i18n.translate("ms.api.command.tab.completer.error", {
player: sender.name,
plugin: plugin.description.name,
command,
args: Java.from(args).join(' '),
ex
}),
...console.stack(ex)
])
} }
return [] return []
} }

11
packages/nodejs/src/events/index.ts
View File

@ -20,6 +20,7 @@
// USE OR OTHER DEALINGS IN THE SOFTWARE. // USE OR OTHER DEALINGS IN THE SOFTWARE.
'use strict'; 'use strict';
var Throwable = Java.type('java.lang.Throwable')
var R = typeof Reflect === 'object' ? Reflect : null var R = typeof Reflect === 'object' ? Reflect : null
var ReflectApply = R && typeof R.apply === 'function' var ReflectApply = R && typeof R.apply === 'function'
? R.apply ? R.apply
@ -136,13 +137,19 @@ EventEmitter.prototype.emit = function emit(type) {
var er; var er;
if (args.length > 0) if (args.length > 0)
er = args[0]; er = args[0];
if (er instanceof Error) { if (er instanceof Error || er instanceof Throwable) {
// Note: The comments on the `throw` lines are intentional, they show // Note: The comments on the `throw` lines are intentional, they show
// up in Node's output if this results in an unhandled exception. // up in Node's output if this results in an unhandled exception.
throw er; // Unhandled 'error' event throw er; // Unhandled 'error' event
} }
if (er.error instanceof Error || er.error instanceof Throwable) {
throw er.error; // Unhandled 'error' event
}
if (er.cause instanceof Error || er.error instanceof Throwable) {
throw er.error; // Unhandled 'error' event
}
// At least give some kind of context to the user // At least give some kind of context to the user
var err = new Error('Unhandled error.' + (er ? ' (' + er.message + ')' : '')); var err = new Error('Unhandled error.' + (er ? ' (' + (er.message || er.error || er.cause || er) + ')' : ''));
// @ts-ignore // @ts-ignore
err.context = er; err.context = er;
throw err; // Unhandled 'error' event throw err; // Unhandled 'error' event

3
packages/nodejs/src/fs/index.ts
View File

@ -5,6 +5,7 @@ const Path = Java.type("java.nio.file.Path");
const JavaString = Java.type("java.lang.String"); const JavaString = Java.type("java.lang.String");
const File = Java.type("java.io.File"); const File = Java.type("java.io.File");
const Files = Java.type("java.nio.file.Files"); const Files = Java.type("java.nio.file.Files");
const Paths = Java.type("java.nio.file.Paths");
const Collector = Java.type("java.util.stream.Collector") const Collector = Java.type("java.util.stream.Collector")
const separatorChar = File.separatorChar; const separatorChar = File.separatorChar;
const StandardCopyOption = Java.type("java.nio.file.StandardCopyOption"); const StandardCopyOption = Java.type("java.nio.file.StandardCopyOption");
@ -37,7 +38,7 @@ function javaFile(...opts: any[]) {
} }
export function renameSync(oldPath: PathLike, newPath: PathLike): void { export function renameSync(oldPath: PathLike, newPath: PathLike): void {
Files.move(Paths.get(oldPath), Paths.get(oldPath), StandardCopyOption['ATOMIC_MOVE'])
} }
export function truncateSync() { export function truncateSync() {

9
packages/polyfill/src/node-shim.ts
View File

@ -126,6 +126,7 @@ class Process extends EventEmitter {
} }
class EventLoop { class EventLoop {
private threadCount = new AtomicInteger(0)
private eventLoopMainThread = undefined private eventLoopMainThread = undefined
private eventLoopTaskQueue = new DelayQueue() private eventLoopTaskQueue = new DelayQueue()
private taskExecuteTimeout = 3000 private taskExecuteTimeout = 3000
@ -134,10 +135,10 @@ class EventLoop {
constructor() { constructor() {
this.taskExecuteTimeout = parseInt(process.env.MS_TASK_EXECUTE_TIMEOUT) || 3000 this.taskExecuteTimeout = parseInt(process.env.MS_TASK_EXECUTE_TIMEOUT) || 3000
this.fixedThreadPool = new ThreadPoolExecutor( this.fixedThreadPool = new ThreadPoolExecutor(
1, 1, 0, TimeUnit.SECONDS, 8, 16, 0, TimeUnit.SECONDS,
new LinkedBlockingQueue(1024), new LinkedBlockingQueue(1024),
new ThreadFactory((run: any) => { new ThreadFactory((run: any) => {
let thread = new Thread(run, "@ccms/event-loop") let thread = new Thread(run, "@ccms/event-loop-" + this.threadCount.incrementAndGet())
thread.setDaemon(true) thread.setDaemon(true)
return thread return thread
})) }))
@ -198,7 +199,7 @@ class EventLoop {
try { try {
callback.apply(undefined, args) callback.apply(undefined, args)
} catch (cause: any) { } catch (cause: any) {
cause = cause.getCause && cause.getCause() || cause cause = cause.getCause ? cause.getCause() : cause
try { try {
process.emit('error', cause) process.emit('error', cause)
} catch (error: any) { } catch (error: any) {
@ -213,7 +214,7 @@ class EventLoop {
return console.warn(`FixedThreadPool isInterrupted exit! Task ${name} exec exit!`) return console.warn(`FixedThreadPool isInterrupted exit! Task ${name} exec exit!`)
} }
if (error instanceof TimeoutException) { if (error instanceof TimeoutException) {
return console.warn(`Task ${name} => ${callback} exec time greater than ${this.taskExecuteTimeout}s!`) return console.warn(`Task ${name} => ${callback} exec time greater than ${this.taskExecuteTimeout}ms!`)
} }
throw error.getCause && error.getCause() || error throw error.getCause && error.getCause() || error
} }

43
packages/websocket/src/client/index.ts
View File

@ -28,12 +28,14 @@ export class WebSocketManager {
export const manager = new WebSocketManager() export const manager = new WebSocketManager()
export class WebSocket extends EventEmitter { export class WebSocket extends EventEmitter {
public static manager: WebSocketManager = manager
public static CONNECTING = 0 public static CONNECTING = 0
public static OPEN = 1 public static OPEN = 1
public static CLOSING = 2 public static CLOSING = 2
public static CLOSED = 3 public static CLOSED = 3
public binaryType: 'blob' | 'arraybuffer' public binaryType: 'blob' | 'arraybuffer'
protected manager: WebSocketManager
protected _url: string protected _url: string
protected _headers: WebSocketHeader = {} protected _headers: WebSocketHeader = {}
@ -42,7 +44,6 @@ export class WebSocket extends EventEmitter {
constructor(url: string, subProtocol: string | string[] = '', headers: WebSocketHeader = {}) { constructor(url: string, subProtocol: string | string[] = '', headers: WebSocketHeader = {}) {
super() super()
this.manager = manager
this._url = url this._url = url
this._headers = headers this._headers = headers
try { try {
@ -54,7 +55,14 @@ export class WebSocket extends EventEmitter {
console.ex(error) console.ex(error)
return return
} }
// mamanger connected client
manager.add(this) manager.add(this)
this.client.on(ClientEvent.close, (_) => manager.del(this))
// add event forward
this.client.on(ClientEvent.open, (event) => this.onopen?.(event))
this.client.on(ClientEvent.message, (event) => this.onmessage?.(event))
this.client.on(ClientEvent.close, (event) => this.onclose?.(event))
this.client.on(ClientEvent.error, (event) => this.onerror?.(event))
setTimeout(() => this.client.connect(), 20) setTimeout(() => this.client.connect(), 20)
} }
get id() { get id() {
@ -70,30 +78,31 @@ export class WebSocket extends EventEmitter {
return this.client.protocol return this.client.protocol
} }
get readyState() { get readyState() {
return this.client.readyStatus return this.client.readyState
} }
get url() { get url() {
return this._url return this._url
} }
set onopen(func: (event: Event) => void) {
this.client.on(ClientEvent.open, func) public onopen: (event: Event) => void
public onmessage: (event: MessageEvent) => void
public onclose: (event: CloseEvent) => void
public onerror: (event: ErrorEvent) => void
public on(eventName: string | symbol, listener: (...args: any[]) => void): this {
this.client.on(eventName, listener)
return this
} }
set onmessage(func: (event: MessageEvent) => void) { public emit(eventName: string | symbol, ...args: any[]): boolean {
this.client.on(ClientEvent.message, func) return this.client.emit(eventName, ...args)
}
set onclose(func: (event: CloseEvent) => void) {
this.client.on(ClientEvent.close, func)
manager.del(this)
}
set onerror(func: (event: ErrorEvent) => void) {
this.client.on(ClientEvent.error, func)
} }
public send(data: any) { public send(data: any) {
this.client.send(data) return this.client.send(data)
} }
public close(code?: number, reason?: string) { public close(code?: number, reason?: string) {
this.client.close(code, reason) return this.client.close(code, reason)
manager.del(this)
} }
} }
global.setGlobal('WebSocket', WebSocket) global.setGlobal('WebSocket', WebSocket)

28
packages/websocket/src/client/netty/handler.ts
View File

@ -1,6 +1,9 @@
import { NettyWebSocket } from '.' import { NettyWebSocket } from '.'
import { WebSocketClientHandlerAdapter } from './adapter/handler' import { WebSocketClientHandlerAdapter } from './adapter/handler'
const Throwable = Java.type('java.lang.Throwable')
const RuntimeException = Java.type('java.lang.RuntimeException')
const CharsetUtil = Java.type('io.netty.util.CharsetUtil') const CharsetUtil = Java.type('io.netty.util.CharsetUtil')
const TextWebSocketFrame = Java.type('io.netty.handler.codec.http.websocketx.TextWebSocketFrame') const TextWebSocketFrame = Java.type('io.netty.handler.codec.http.websocketx.TextWebSocketFrame')
const CloseWebSocketFrame = Java.type('io.netty.handler.codec.http.websocketx.CloseWebSocketFrame') const CloseWebSocketFrame = Java.type('io.netty.handler.codec.http.websocketx.CloseWebSocketFrame')
@ -20,6 +23,7 @@ export class WebSocketClientHandler extends WebSocketClientHandlerAdapter {
return true return true
} }
handlerAdded(ctx: any) { handlerAdded(ctx: any) {
this.client.onconnection({})
if (ctx.newPromise) { if (ctx.newPromise) {
this.handshakeFuture = ctx.newPromise() this.handshakeFuture = ctx.newPromise()
} else { } else {
@ -28,18 +32,21 @@ export class WebSocketClientHandler extends WebSocketClientHandlerAdapter {
} }
channelActive(ctx: any) { channelActive(ctx: any) {
this.handshaker.handshake(ctx.channel()) this.handshaker.handshake(ctx.channel())
setTimeout(() => {
this.abortHandshake(new Error('handshake timed out.'))
}, 10000)
} }
channelInactive(ctx: any) { channelInactive(ctx: any) {
if (this.client.readyStatus != WebSocket.CLOSED) { this.client.close(1006, 'connection was closed abnormally.', true)
this.client.onclose({ code: 1006, reason: 'client connection channel inactive.' })
}
} }
channelRead0(ctx: any, msg: any) { channelRead0(ctx: any, msg: any) {
let ch = ctx.channel() let ch = ctx.channel()
if (!this.handshaker.isHandshakeComplete()) { if (!this.handshaker.isHandshakeComplete()) {
// web socket client connected console.debug(`Netty Handler channelRead0 websocket client connected`)
// websocket client connected
this.handshaker.finishHandshake(ch, msg) this.handshaker.finishHandshake(ch, msg)
this.handshakeFuture.setSuccess() this.handshakeFuture.setSuccess()
this.client.onconnect({})
return return
} }
@ -52,14 +59,19 @@ export class WebSocketClientHandler extends WebSocketClientHandlerAdapter {
if (frame instanceof TextWebSocketFrame) { if (frame instanceof TextWebSocketFrame) {
this.client.onmessage({ data: frame.text() }) this.client.onmessage({ data: frame.text() })
} else if (frame instanceof CloseWebSocketFrame) { } else if (frame instanceof CloseWebSocketFrame) {
this.client.close(1000, 'server close connection.') this.client.receiverClose(frame.statusCode(), frame.reasonText())
} }
} }
abortHandshake(reason: Error) {
if (this.handshakeFuture.isDone()) { return }
if (!(reason instanceof Throwable)) {
reason = new RuntimeException(reason)
}
this.handshakeFuture.setFailure(reason)
}
exceptionCaught(ctx: any, cause: Error) { exceptionCaught(ctx: any, cause: Error) {
console.debug(`${ctx} exceptionCaught ${cause}`) console.debug(`${ctx} exceptionCaught ${cause}`)
this.client.abortHandshake(cause)
this.client.onerror({ error: cause }) this.client.onerror({ error: cause })
if (!this.handshakeFuture.isDone()) {
this.handshakeFuture.setFailure(cause)
}
} }
} }

51
packages/websocket/src/client/netty/index.ts
View File

@ -60,6 +60,8 @@ export class NettyWebSocket extends Transport {
private channel: any private channel: any
private b: any private b: any
private handler: any
constructor(url: string, subProtocol: string = '', headers: WebSocketHeader = {}) { constructor(url: string, subProtocol: string = '', headers: WebSocketHeader = {}) {
super(url, subProtocol, headers) super(url, subProtocol, headers)
if (!url) { if (!url) {
@ -88,7 +90,6 @@ export class NettyWebSocket extends Transport {
return `${this.channel?.id()}` || `NettyWebSocket#${channelCount.incrementAndGet()}` return `${this.channel?.id()}` || `NettyWebSocket#${channelCount.incrementAndGet()}`
} }
doConnect() { doConnect() {
console.debug('client NettyWebSocket doConnect', this._url)
let uri = URI.create(this._url) let uri = URI.create(this._url)
let headers = new DefaultHttpHeaders() let headers = new DefaultHttpHeaders()
for (const key of Object.getOwnPropertyNames(this._headers || {})) { for (const key of Object.getOwnPropertyNames(this._headers || {})) {
@ -97,7 +98,7 @@ export class NettyWebSocket extends Transport {
// Connect with V13 (RFC 6455 aka HyBi-17). You can change it to V08 or V00. // Connect with V13 (RFC 6455 aka HyBi-17). You can change it to V08 or V00.
// If you change it to V00, ping is not supported and remember to change // If you change it to V00, ping is not supported and remember to change
// HttpResponseDecoder to WebSocketHttpResponseDecoder in the pipeline. // HttpResponseDecoder to WebSocketHttpResponseDecoder in the pipeline.
let handler = new WebSocketClientHandler(WebSocketClientHandshakerFactory this.handler = new WebSocketClientHandler(WebSocketClientHandshakerFactory
.newHandshaker(uri, WebSocketVersion.V13, null, false, headers), this) .newHandshaker(uri, WebSocketVersion.V13, null, false, headers), this)
this.b = new Bootstrap() this.b = new Bootstrap()
this.b.group(group) this.b.group(group)
@ -108,7 +109,7 @@ export class NettyWebSocket extends Transport {
if (this._schema == "wss") { if (this._schema == "wss") {
if (SslContextBuilder) { if (SslContextBuilder) {
let sslCtx = SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build() let sslCtx = SslContextBuilder.forClient().trustManager(InsecureTrustManagerFactory.INSTANCE).build()
pipeline.addLast(sslCtx.newHandler(ch.alloc(), this._host, this._port)) pipeline.addLast('ssl', sslCtx.newHandler(ch.alloc(), this._host, this._port))
} else { } else {
let sslEngine = SSLContext.getDefault().createSSLEngine() let sslEngine = SSLContext.getDefault().createSSLEngine()
sslEngine.setUseClientMode(true) sslEngine.setUseClientMode(true)
@ -117,37 +118,37 @@ export class NettyWebSocket extends Transport {
} }
pipeline.addLast("http-codec", new HttpClientCodec()) pipeline.addLast("http-codec", new HttpClientCodec())
pipeline.addLast("aggregator", new HttpObjectAggregator(65536)) pipeline.addLast("aggregator", new HttpObjectAggregator(65536))
pipeline.addLast("websocket", handler.getHandler()) pipeline.addLast("websocket", this.handler.getHandler())
} }
})) }))
this.b.connect(this._host, this._port).addListener(new ChannelFutureListener((future: any) => { try {
try { this.channel = this.b.connect(this._host, this._port).sync().channel()
this.channel = future.sync().channel() this.handler.handshakeFuture.sync()
this.onconnection({}) } catch (error) {
handler.handshakeFuture.addListener(new ChannelFutureListener((future: any) => { // ignore connect error
try { // tigger error at handshakeFuture
future.sync() }
// only trigger onconnect when not have error
this.onconnect({})
} catch (error: any) {
// ignore error exceptionCaught from handler
// this.onerror({ error })
}
}))
} catch (error: any) {
this.onerror({ error })
}
}))
} }
doSend(text: string) { doSend(text: string) {
this.channel.writeAndFlush(new TextWebSocketFrame(text)) this.channel.writeAndFlush(new TextWebSocketFrame(text))
} }
doClose(code: number, reason: string) { doClose(code: number, reason: string, wasClean: boolean = false) {
this.channel.writeAndFlush(new CloseWebSocketFrame()) console.debug(`Netty Client doClose code: ${code} reason: ${reason}`)
this.channel.closeFuture().addListener(new ChannelFutureListener(() => { if (this.readyState == WebSocket.CLOSING) {
if (!this._closeFrameSent) {
console.debug(`Netty Client doClose send close frame`)
this.channel?.writeAndFlush(new CloseWebSocketFrame(code, reason))
this._closeFrameSent = true
}
if (!this._closeFrameReceived && !wasClean) { return console.debug(`Netty Client doClose wait server send close`) }
}
this.channel?.closeFuture().addListener(new ChannelFutureListener(() => {
this.onclose({ code, reason }) this.onclose({ code, reason })
})) }))
} }
abortHandshake(reason: Error): void {
this.handler.abortHandshake(reason)
}
getChannel() { getChannel() {
return this.channel return this.channel
} }

65
packages/websocket/src/client/transport.ts
View File

@ -8,6 +8,9 @@ export abstract class Transport extends EventEmitter {
protected _protocol: string protected _protocol: string
protected _headers: WebSocketHeader = {} protected _headers: WebSocketHeader = {}
protected _closeFrameReceived = false;
protected _closeFrameSent = false;
constructor(uri: string, subProtocol: string = '', headers: WebSocketHeader = {}) { constructor(uri: string, subProtocol: string = '', headers: WebSocketHeader = {}) {
super() super()
this._url = uri this._url = uri
@ -23,24 +26,22 @@ export abstract class Transport extends EventEmitter {
return this._protocol return this._protocol
} }
get readyStatus() { get readyState() {
return this._state return this._state
} }
set readyStatus(state: number) { set readyState(state: number) {
this._state = state this._state = state
} }
connect() { connect() {
try { this.doConnect()
this.doConnect()
} catch (error: any) {
console.ex(error)
this.onerror({ error })
}
} }
send(text: string) { send(text: string) {
if (this.readyState === WebSocket.CONNECTING) {
throw new Error('WebSocket is not open: readyState 0 (CONNECTING)');
}
try { try {
this.doSend(text) this.doSend(text)
} catch (error: any) { } catch (error: any) {
@ -48,16 +49,24 @@ export abstract class Transport extends EventEmitter {
} }
} }
close(code: number = 1000, reason: string = '') { close(code: number = 1000, reason: string = '', wasClean: boolean = false) {
if (this.readyStatus < WebSocket.CLOSING) { if (this.readyState === WebSocket.CLOSED) return;
this.readyStatus = WebSocket.CLOSING if (this.readyState === WebSocket.CONNECTING) {
try { const msg = 'WebSocket was closed before the connection was established';
this.doClose(code, reason) this.abortHandshake(new Error(msg));
} catch (error: any) { return;
this.onerror({ error }) }
if (this.readyState === WebSocket.CLOSING) {
if (this._closeFrameSent && this._closeFrameReceived) {
this.onclose({ code, reason });
} }
} else { return;
console.debug(`WebSocket Transport ${this.id} call close code ${code} reason ${reason} but state is ${this.readyStatus}`) }
this.readyState = WebSocket.CLOSING
try {
this.doClose(code, reason, wasClean)
} catch (error: any) {
this.onerror({ error })
} }
} }
@ -67,11 +76,11 @@ export abstract class Transport extends EventEmitter {
} }
onconnect(event: Event) { onconnect(event: Event) {
if (this.readyStatus != WebSocket.OPEN) { if (this.readyState != WebSocket.OPEN) {
this.readyStatus = WebSocket.OPEN this.readyState = WebSocket.OPEN
this.emit(ClientEvent.open, event) this.emit(ClientEvent.open, event)
} else { } else {
console.debug(`WebSocket Transport ${this.id} call onconnect but state is ${this.readyStatus}`) console.debug(`WebSocket Transport ${this.id} call onconnect but state is ${this.readyState}`)
} }
} }
@ -84,16 +93,24 @@ export abstract class Transport extends EventEmitter {
} }
onclose(event: CloseEvent) { onclose(event: CloseEvent) {
if (this.readyStatus != WebSocket.CLOSED) { console.debug(`WebSocket Transport ${this.id} call onclose CloseEvent[code: ${event.code}, reason: ${event.reason}]`)
this.readyStatus = WebSocket.CLOSED if (this.readyState != WebSocket.CLOSED) {
this.readyState = WebSocket.CLOSED
this.emit(ClientEvent.close, event) this.emit(ClientEvent.close, event)
} else { } else {
console.debug(`WebSocket Transport ${this.id} call onclose but state is ${this.readyStatus} CloseEvent[code: ${event.code}, reason: ${event.reason}]`) console.debug(`WebSocket Transport ${this.id} call onclose but state is ${this.readyState} CloseEvent[code: ${event.code}, reason: ${event.reason}]`)
} }
} }
receiverClose(code: number, reason: string) {
console.debug(`Netty Handler receeve close code: ${code} reason: ${reason}`)
this._closeFrameReceived = true;
this.close(code, reason)
}
abstract getId(): string abstract getId(): string
abstract doConnect(): void abstract doConnect(): void
abstract doSend(text: string): void abstract doSend(text: string): void
abstract doClose(code: number, reason: string): void abstract doClose(code: number, reason: string, wasClean?: boolean): void
abstract abortHandshake(reason: Error): void
} }