type WorkerResponse
} from './types.js'
-const factorial: (n: number) => number = n => {
+const factorial: (n: number) => number = (n) => {
if (n === 0) {
return 1
}
return factorial(n - 1) * n
}
-const startWebSocketServer = (workerData?: WorkerData): WorkerResponse => {
- try {
- const wss = new WebSocketServer({ port: workerData?.port }, () => {
+class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
+ private static wss: WebSocketServer
+
+ private static readonly startWebSocketServer = (
+ workerData?: WorkerData
+ ): WorkerResponse => {
+ const { port } = workerData as WorkerData
+
+ WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
console.info(
- `⚡️[ws server]: WebSocket server is started at ws://localhost:${
- workerData?.port as number
- }/`
+ `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
)
})
- wss.on('connection', ws => {
+ WebSocketServerWorker.wss.on('connection', (ws) => {
ws.on('error', console.error)
ws.on('message', (message: RawData) => {
const { type, data } = JSON.parse(
})
return {
status: true,
- port: wss.options.port
- }
- } catch (err) {
- return {
- status: false,
- error: err as Error
+ port: WebSocketServerWorker.wss.options.port
}
}
-}
-class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
public constructor () {
- super(startWebSocketServer)
+ super(WebSocketServerWorker.startWebSocketServer, {
+ killHandler: () => {
+ WebSocketServerWorker.wss.close()
+ }
+ })
}
}