} from './types.js'
const emptyFunction = (): void => {
- /** Intentional */
+ /* Intentional */
}
class WebSocketServerWorker extends ClusterWorker<
private static readonly startWebSocketServer = (
workerData?: ClusterWorkerData
): ClusterWorkerResponse => {
- const { port } = workerData as ClusterWorkerData
- WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
- console.info(
- `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
- )
- })
+ const { port, workerFile, minWorkers, maxWorkers, ...poolOptions } =
+ workerData as ClusterWorkerData
WebSocketServerWorker.requestHandlerPool = new DynamicThreadPool<
ThreadWorkerData<DataPayload>,
ThreadWorkerResponse<DataPayload>
>(
- workerData?.minWorkers ?? 1,
- workerData?.maxWorkers ?? availableParallelism(),
- workerData?.workerFile as string
+ minWorkers ?? 1,
+ maxWorkers ?? availableParallelism(),
+ workerFile,
+ poolOptions
)
- WebSocketServerWorker.wss.on('connection', (ws) => {
+ WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
+ console.info(
+ `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
+ )
+ })
+
+ WebSocketServerWorker.wss.on('connection', ws => {
ws.on('error', console.error)
ws.on('message', (message: RawData) => {
const { type, data } = JSON.parse(
case MessageType.echo:
WebSocketServerWorker.requestHandlerPool
.execute({ data }, 'echo')
- .then((response) => {
+ .then(response => {
ws.send(
JSON.stringify({
type: MessageType.echo,
data: response.data
})
)
- return null
+ return undefined
})
.catch(emptyFunction)
break
case MessageType.factorial:
WebSocketServerWorker.requestHandlerPool
.execute({ data }, 'factorial')
- .then((response) => {
+ .then(response => {
ws.send(
JSON.stringify({
type: MessageType.factorial,
data: response.data
})
)
- return null
+ return undefined
})
.catch(emptyFunction)
break
public constructor () {
super(WebSocketServerWorker.startWebSocketServer, {
- killHandler: () => {
+ killHandler: async () => {
+ await WebSocketServerWorker.requestHandlerPool.destroy()
WebSocketServerWorker.wss.close()
}
})