src/charging-station/ChargingStationWorker.ts

   1 // Partial Copyright Jerome Benoit. 2021-2024. All Rights Reserved.
   2
   3 import { parentPort } from 'node:worker_threads'
   4
   5 import { ThreadWorker } from 'poolifier'
   6
   7 import { ChargingStation } from './ChargingStation.js'
   8 import { BaseError } from '../exception/index.js'
   9 import type {
  10   ChargingStationData,
  11   ChargingStationWorkerData,
  12   ChargingStationWorkerEventError,
  13   ChargingStationWorkerMessage
  14 } from '../types/index.js'
  15 import { Configuration, buildChargingStationDataPayload } from '../utils/index.js'
  16 import { type WorkerMessage, WorkerMessageEvents } from '../worker/index.js'
  17
  18 export let chargingStationWorker: object
  19 if (Configuration.workerPoolInUse()) {
  20   chargingStationWorker = new ThreadWorker<ChargingStationWorkerData>(
  21     (data?: ChargingStationWorkerData): void => {
  22       // eslint-disable-next-line @typescript-eslint/no-non-null-assertion, no-new
  23       new ChargingStation(data!.index, data!.templateFile)
  24     }
  25   )
  26 } else {
  27   // eslint-disable-next-line @typescript-eslint/no-extraneous-class
  28   class ChargingStationWorker<Data extends ChargingStationWorkerData> {
  29     constructor () {
  30       parentPort?.on('message', (message: WorkerMessage<Data>) => {
  31         switch (message.event) {
  32           case WorkerMessageEvents.addWorkerElement:
  33             try {
  34               const chargingStation = new ChargingStation(
  35                 message.data.index,
  36                 message.data.templateFile
  37               )
  38               parentPort?.postMessage({
  39                 event: WorkerMessageEvents.addedWorkerElement,
  40                 data: buildChargingStationDataPayload(chargingStation)
  41               } satisfies ChargingStationWorkerMessage<ChargingStationData>)
  42             } catch (error) {
  43               parentPort?.postMessage({
  44                 event: WorkerMessageEvents.workerElementError,
  45                 data: {
  46                   event: WorkerMessageEvents.addWorkerElement,
  47                   name: (error as Error).name,
  48                   message: (error as Error).message,
  49                   stack: (error as Error).stack
  50                 }
  51               } satisfies ChargingStationWorkerMessage<ChargingStationWorkerEventError>)
  52             }
  53             break
  54           default:
  55             throw new BaseError(
  56               `Unknown worker event: '${message.event}' received with data: '${JSON.stringify(
  57                 message.data,
  58                 undefined,
  59                 2
  60               )}'`
  61             )
  62         }
  63       })
  64     }
  65   }
  66   chargingStationWorker = new ChargingStationWorker<ChargingStationWorkerData>()
  67 }