Commit | Line | Data |
---|---|---|
72855e92 JB |
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 | errorHandler: (e: Error) => { | |
16 | console.error(e) | |
17 | } | |
18 | } | |
19 | ) | |
20 | ||
21 | // Start one ws server instance per cluster worker in the pool | |
22 | for (let i = 1; i <= pool.info.maxSize; i++) { | |
23 | pool | |
24 | .execute({ port: 8080 }) | |
25 | .then(response => { | |
26 | if (response.status) { | |
27 | console.info( | |
28 | // eslint-disable-next-line @typescript-eslint/restrict-template-expressions | |
29 | `WebSocket server is listening on worker ${i} on port ${response.port}` | |
30 | ) | |
31 | } else { | |
32 | console.error( | |
33 | `WebSocket server failed to start on worker ${i}:`, | |
34 | response.error | |
35 | ) | |
36 | } | |
37 | return null | |
38 | }) | |
39 | .catch(error => { | |
40 | console.error(error) | |
41 | }) | |
42 | } |