examples/typescript/websocket-server-pool/ws-cluster/src/worker.ts

   1 import { ClusterWorker } from 'poolifier'
   2 import { type RawData, WebSocketServer } from 'ws'
   3
   4 import {
   5   type DataPayload,
   6   type MessagePayload,
   7   MessageType,
   8   type WorkerData,
   9   type WorkerResponse
  10 } from './types.js'
  11
  12 class WebSocketServerWorker extends ClusterWorker<WorkerData, WorkerResponse> {
  13   private static wss: WebSocketServer
  14
  15   private static readonly factorial = (n: number | bigint): bigint => {
  16     if (n === 0 || n === 1) {
  17       return 1n
  18     } else {
  19       n = BigInt(n)
  20       let factorial = 1n
  21       for (let i = 1n; i <= n; i++) {
  22         factorial *= i
  23       }
  24       return factorial
  25     }
  26   }
  27
  28   private static readonly startWebSocketServer = (
  29     workerData?: WorkerData
  30   ): WorkerResponse => {
  31     const { port } = workerData!
  32
  33     WebSocketServerWorker.wss = new WebSocketServer({ port }, () => {
  34       console.info(
  35         `⚡️[ws server]: WebSocket server is started in cluster worker at ws://localhost:${port}/`
  36       )
  37     })
  38
  39     WebSocketServerWorker.wss.on('connection', ws => {
  40       ws.on('error', console.error)
  41       ws.on('message', (message: RawData) => {
  42         const { type, data } = JSON.parse(
  43           // eslint-disable-next-line @typescript-eslint/no-base-to-string
  44           message.toString()
  45         ) as MessagePayload<DataPayload>
  46         switch (type) {
  47           case MessageType.echo:
  48             ws.send(
  49               JSON.stringify({
  50                 type: MessageType.echo,
  51                 data
  52               })
  53             )
  54             break
  55           case MessageType.factorial:
  56             ws.send(
  57               JSON.stringify(
  58                 {
  59                   type: MessageType.factorial,
  60                   data: {
  61                     number: WebSocketServerWorker.factorial(data.number!)
  62                   }
  63                 },
  64                 (_, v) => (typeof v === 'bigint' ? v.toString() : v)
  65               )
  66             )
  67             break
  68         }
  69       })
  70     })
  71     return {
  72       status: true,
  73       port: WebSocketServerWorker.wss.options.port
  74     }
  75   }
  76
  77   public constructor () {
  78     super(WebSocketServerWorker.startWebSocketServer, {
  79       killHandler: () => {
  80         WebSocketServerWorker.wss.close()
  81       }
  82     })
  83   }
  84 }
  85
  86 export const webSocketServerWorker = new WebSocketServerWorker()