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 pool = new FixedClusterPool<ClusterWorkerData, ClusterWorkerResponse>(
  12   Math.round(availableParallelism() / 2),
  13   webSocketServerWorkerFile,
  14   {
  15     onlineHandler: () => {
  16       pool
  17         .execute({ port: 8080 })
  18         .then(response => {
  19           if (response.status) {
  20             console.info(
  21               // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
  22               `WebSocket server is listening on cluster worker on port ${response.port}`
  23             )
  24           }
  25           return null
  26         })
  27         .catch(error => {
  28           console.error(
  29             'WebSocket server failed to start on cluster worker:',
  30             error
  31           )
  32         })
  33     },
  34     errorHandler: (e: Error) => {
  35       console.error('Cluster worker error', e)
  36     }
  37   }
  38 )