examples/typescript/websocket-server-pool/ws-hybrid/src/main.ts

   1 import { dirname, extname, join } from 'node:path'
   2 import { fileURLToPath } from 'node:url'
   3 import { FixedClusterPool, availableParallelism } from 'poolifier'
   4 import type { ClusterWorkerData, ClusterWorkerResponse } from './types.js'
   5
   6 const webSocketServerWorkerFile = join(
   7   dirname(fileURLToPath(import.meta.url)),
   8   `websocket-server-worker${extname(fileURLToPath(import.meta.url))}`
   9 )
  10
  11 const requestHandlerWorkerFile = join(
  12   dirname(fileURLToPath(import.meta.url)),
  13   `request-handler-worker${extname(fileURLToPath(import.meta.url))}`
  14 )
  15
  16 const pool = new FixedClusterPool<ClusterWorkerData, ClusterWorkerResponse>(
  17   Math.round(availableParallelism() / 2),
  18   webSocketServerWorkerFile,
  19   {
  20     enableEvents: false,
  21     onlineHandler: () => {
  22       pool
  23         .execute({
  24           port: 8080,
  25           maxWorkers:
  26             Math.round(availableParallelism() / 4) < 1
  27               ? 1
  28               : Math.round(availableParallelism() / 4),
  29           workerFile: requestHandlerWorkerFile,
  30           enableTasksQueue: true,
  31           tasksQueueOptions: {
  32             concurrency: 8
  33           },
  34           errorHandler: (e: Error) => {
  35             console.error('Thread worker error:', e)
  36           }
  37         })
  38         .then(response => {
  39           if (response.status) {
  40             console.info(
  41               // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
  42               `WebSocket server is listening in cluster worker on port ${response.port}`
  43             )
  44           }
  45           return undefined
  46         })
  47         .catch(error => {
  48           console.error(
  49             'WebSocket server failed to start in cluster worker:',
  50             error
  51           )
  52         })
  53     },
  54     errorHandler: (e: Error) => {
  55       console.error('Cluster worker error', e)
  56     }
  57   }
  58 )