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, type 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     onlineHandler: () => {
  21       pool
  22         .execute({
  23           port: 8080,
  24           maxWorkers: Math.round(availableParallelism() / 4),
  25           workerFile: requestHandlerWorkerFile,
  26           enableTasksQueue: true,
  27           tasksQueueOptions: {
  28             concurrency: 8
  29           },
  30           errorHandler: (e: Error) => {
  31             console.error('Thread worker error:', e)
  32           }
  33         })
  34         .then((response) => {
  35           if (response.status) {
  36             console.info(
  37               // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
  38               `WebSocket server is listening in cluster worker on port ${response.port}`
  39             )
  40           }
  41           return null
  42         })
  43         .catch((error) => {
  44           console.error(
  45             'WebSocket server failed to start in cluster worker:',
  46             error
  47           )
  48         })
  49     },
  50     errorHandler: (e: Error) => {
  51       console.error('Cluster worker error', e)
  52     }
  53   }
  54 )