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