examples/typescript/websocket-server-pool/ws-worker_threads/src/worker.ts

   1 import { ThreadWorker } from 'poolifier'
   2
   3 import type { DataPayload, WorkerData, WorkerResponse } from './types.js'
   4
   5 class RequestHandlerWorker<
   6   Data extends WorkerData<DataPayload>,
   7   Response extends WorkerResponse<DataPayload>
   8 > extends ThreadWorker<Data, Response> {
   9   private static readonly factorial = (n: number | bigint): bigint => {
  10     if (n === 0 || n === 1) {
  11       return 1n
  12     } else {
  13       n = BigInt(n)
  14       let factorial = 1n
  15       for (let i = 1n; i <= n; i++) {
  16         factorial *= i
  17       }
  18       return factorial
  19     }
  20   }
  21
  22   public constructor () {
  23     super({
  24       echo: (workerData?: Data) => {
  25         return workerData as unknown as Response
  26       },
  27       factorial: (workerData?: Data) => {
  28         return {
  29           data: {
  30             number: RequestHandlerWorker.factorial(workerData!.data.number!)
  31           }
  32         } as unknown as Response
  33       }
  34     })
  35   }
  36 }
  37
  38 export const requestHandlerWorker = new RequestHandlerWorker<
  39 WorkerData<DataPayload>,
  40 WorkerResponse<DataPayload>
  41 >()