examples/typescript/websocket-server-pool/ws-cluster/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 WorkerData, type WorkerResponse } from './types.js'
   5
   6 const workerFile = join(
   7   dirname(fileURLToPath(import.meta.url)),
   8   `worker${extname(fileURLToPath(import.meta.url))}`
   9 )
  10
  11 const pool = new FixedClusterPool<WorkerData, WorkerResponse>(
  12   availableParallelism(),
  13   workerFile,
  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 worker on port ${response.port}`
  23             )
  24           } else {
  25             console.error(
  26               'WebSocket server failed to start on worker:',
  27               response.error
  28             )
  29           }
  30           return null
  31         })
  32         .catch(error => {
  33           console.error(error)
  34         })
  35     },
  36     errorHandler: (e: Error) => {
  37       console.error(e)
  38     }
  39   }
  40 )