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

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