highperfocused/nostream
1
0
Fork 0
mirror of https://github.com/Cameri/nostream.git synced 2025-03-17 21:31:48 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: inc. pong timeout

Browse Source
This commit is contained in:
Ricardo Arturo Cabral Mejía 2023-01-29 23:51:56 -05:00
parent 48ca4fd9bf
commit 2a33d748e4
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/adapters/web-socket-adapter.ts
View File

@ -1,7 +1,7 @@
import ws, { WebSocket } from 'ws'
import cluster from 'cluster'
import { EventEmitter } from 'stream'
import { IncomingMessage as IncomingHttpMessage } from 'http'
import { WebSocket } from 'ws'
import { ContextMetadata, Factory } from '../@types/base'
import { createNoticeMessage, createOutgoingEventMessage } from '../utils/messages'
@ -22,7 +22,7 @@ import { Settings } from '../@types/settings'
import { SocketAddress } from 'net'
(() => {
(WebSocket as any).Receiver.prototype._write = function _write (chunk: any, _encoding: any, cb: any) {
(ws as any).Receiver.prototype._write = function _write (chunk: any, _encoding: any, cb: any) {
if (this._opcode === 0x08 && this._state == 0) return cb()
this._bufferedBytes += chunk.length
@ -68,6 +68,8 @@ export class WebSocketAdapter extends EventEmitter implements IWebSocketAdapter
family: address.indexOf(':') >= 0 ? 'ipv6' : 'ipv4',
})
console.log(`web-socket-adapter: new client ${this.clientId} (${this.getClientAddress()})`)
this.client
.on('error', (error) => {
if (error.name === 'RangeError' && error.message === 'Max payload size exceeded') {
@ -274,6 +276,6 @@ export class WebSocketAdapter extends EventEmitter implements IWebSocketAdapter
this.removeAllListeners()
this.client.removeAllListeners()
debug('client %s closed', this.clientId)
console.error(`web-socket-adapter: disconnected client ${this.clientId} (${this.getClientAddress()})`)
}
}

4
src/adapters/web-socket-server-adapter.ts
View File

@ -14,7 +14,7 @@ import { WebServerAdapter } from './web-server-adapter'
const debug = createLogger('web-socket-server-adapter')
const WSS_CLIENT_HEALTH_PROBE_INTERVAL = 30000
const WSS_CLIENT_HEALTH_PROBE_INTERVAL = 60000
export class WebSocketServerAdapter extends WebServerAdapter implements IWebSocketServerAdapter {
private webSocketsAdapters: WeakMap<WebSocket, IWebSocketAdapter>
@ -95,6 +95,8 @@ export class WebSocketServerAdapter extends WebServerAdapter implements IWebSock
return
}
console.log('web-socket-server-adapter: connected clients:', this.webSocketServer.clients.size)
this.webSocketsAdapters.set(client, this.createWebSocketAdapter([client, req, this]))
}