repositories
/
e-mobility-charging-stations-simulator.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
docs: improve code comments
[e-mobility-charging-stations-simulator.git]
/
src
/
worker
/
WorkerSet.ts
diff --git
a/src/worker/WorkerSet.ts
b/src/worker/WorkerSet.ts
index da5684b0345bb2546fd0094e30e991edbb12634a..61118a0d0a274d17871a71ec8e2ebec3efa2c2a9 100644
(file)
--- a/
src/worker/WorkerSet.ts
+++ b/
src/worker/WorkerSet.ts
@@
-1,6
+1,6
@@
// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
// Partial Copyright Jerome Benoit. 2021-2023. All Rights Reserved.
-import
EventEmitterAsyncResource
from 'node:events';
+import
{ EventEmitter }
from 'node:events';
import { SHARE_ENV, Worker } from 'node:worker_threads';
import { WorkerAbstract } from './WorkerAbstract';
import { SHARE_ENV, Worker } from 'node:worker_threads';
import { WorkerAbstract } from './WorkerAbstract';
@@
-16,29
+16,41
@@
import {
import { sleep } from './WorkerUtils';
export class WorkerSet extends WorkerAbstract<WorkerData> {
import { sleep } from './WorkerUtils';
export class WorkerSet extends WorkerAbstract<WorkerData> {
- public readonly emitter
: EventEmitterAsyncResource
;
+ public readonly emitter
!: EventEmitter
;
private readonly workerSet: Set<WorkerSetElement>;
/**
private readonly workerSet: Set<WorkerSetElement>;
/**
- * Create a new `WorkerSet`.
+ * Create
s
a new `WorkerSet`.
*
* @param workerScript -
* @param workerOptions -
*/
constructor(workerScript: string, workerOptions?: WorkerOptions) {
super(workerScript, workerOptions);
*
* @param workerScript -
* @param workerOptions -
*/
constructor(workerScript: string, workerOptions?: WorkerOptions) {
super(workerScript, workerOptions);
+ this.workerOptions.poolOptions = {
+ ...{
+ enableEvents: true,
+ restartWorkerOnError: true,
+ },
+ ...this.workerOptions.poolOptions,
+ };
this.workerSet = new Set<WorkerSetElement>();
this.workerSet = new Set<WorkerSetElement>();
- this.emitter = new EventEmitterAsyncResource();
+ if (this.workerOptions.poolOptions?.enableEvents) {
+ this.emitter = new EventEmitter();
+ }
}
get info(): SetInfo {
return {
}
get info(): SetInfo {
return {
+ version: WorkerConstants.version,
+ type: 'set',
+ worker: 'thread',
size: this.size,
size: this.size,
-
runningElements
: [...this.workerSet].reduce(
+
elementsExecuting
: [...this.workerSet].reduce(
(accumulator, workerSetElement) => accumulator + workerSetElement.numberOfWorkerElements,
(accumulator, workerSetElement) => accumulator + workerSetElement.numberOfWorkerElements,
- 0
+ 0
,
),
),
- elementsPerWorker: this.maxElementsPerWorker,
+ elementsPerWorker: this.maxElementsPerWorker
!
,
};
}
};
}
@@
-54,13
+66,19
@@
export class WorkerSet extends WorkerAbstract<WorkerData> {
public async start(): Promise<void> {
this.addWorkerSetElement();
// Add worker set element sequentially to optimize memory at startup
public async start(): Promise<void> {
this.addWorkerSetElement();
// Add worker set element sequentially to optimize memory at startup
- this.workerOptions.workerStartDelay
> 0 && (await sleep(this.workerOptions.workerStartDelay
));
+ this.workerOptions.workerStartDelay
! > 0 && (await sleep(this.workerOptions.workerStartDelay!
));
}
/** @inheritDoc */
public async stop(): Promise<void> {
for (const workerSetElement of this.workerSet) {
}
/** @inheritDoc */
public async stop(): Promise<void> {
for (const workerSetElement of this.workerSet) {
+ const workerExitPromise = new Promise<void>((resolve) => {
+ workerSetElement.worker.on('exit', () => {
+ resolve();
+ });
+ });
await workerSetElement.worker.terminate();
await workerSetElement.worker.terminate();
+ await workerExitPromise;
}
this.workerSet.clear();
}
}
this.workerSet.clear();
}
@@
-77,49
+95,57
@@
export class WorkerSet extends WorkerAbstract<WorkerData> {
});
++workerSetElement.numberOfWorkerElements;
// Add element sequentially to optimize memory at startup
});
++workerSetElement.numberOfWorkerElements;
// Add element sequentially to optimize memory at startup
- if (this.workerOptions.elementStartDelay > 0) {
- await sleep(this.workerOptions.elementStartDelay);
+ if (this.workerOptions.elementStartDelay
!
> 0) {
+ await sleep(this.workerOptions.elementStartDelay
!
);
}
}
/**
}
}
/**
- * Add a new `WorkerSetElement`.
+ * Add
s
a new `WorkerSetElement`.
*/
private addWorkerSetElement(): WorkerSetElement {
const worker = new Worker(this.workerScript, {
env: SHARE_ENV,
*/
private addWorkerSetElement(): WorkerSetElement {
const worker = new Worker(this.workerScript, {
env: SHARE_ENV,
- ...this.workerOptions.poolOptions.workerOptions,
+ ...this.workerOptions.poolOptions
?
.workerOptions,
});
worker.on(
'message',
});
worker.on(
'message',
- this.workerOptions
?.poolOptions?.messageHandler ?? WorkerConstants.EMPTY_FUNCTION
+ this.workerOptions
.poolOptions?.messageHandler ?? WorkerConstants.EMPTY_FUNCTION,
);
worker.on(
'error',
);
worker.on(
'error',
- this.workerOptions
?.poolOptions?.errorHandler ?? WorkerConstants.EMPTY_FUNCTION
+ this.workerOptions
.poolOptions?.errorHandler ?? WorkerConstants.EMPTY_FUNCTION,
);
worker.on('error', (error) => {
);
worker.on('error', (error) => {
- this.emitter.emit(WorkerSetEvents.error, error);
- this.addWorkerSetElement();
+ this.emitter?.emit(WorkerSetEvents.error, error);
+ if (this.workerOptions.poolOptions?.restartWorkerOnError) {
+ this.addWorkerSetElement();
+ }
});
worker.on(
'online',
});
worker.on(
'online',
- this.workerOptions
?.poolOptions?.onlineHandler ?? WorkerConstants.EMPTY_FUNCTION
+ this.workerOptions
.poolOptions?.onlineHandler ?? WorkerConstants.EMPTY_FUNCTION,
);
worker.on(
'exit',
);
worker.on(
'exit',
- this.workerOptions?.poolOptions?.exitHandler ?? WorkerConstants.EMPTY_FUNCTION
+ this.workerOptions.poolOptions?.exitHandler ?? WorkerConstants.EMPTY_FUNCTION,
+ );
+ worker.once('exit', () =>
+ this.removeWorkerSetElement(this.getWorkerSetElementByWorker(worker)!),
);
);
- worker.on('exit', () => this.workerSet.delete(this.getWorkerSetElementByWorker(worker)));
const workerSetElement: WorkerSetElement = { worker, numberOfWorkerElements: 0 };
this.workerSet.add(workerSetElement);
return workerSetElement;
}
const workerSetElement: WorkerSetElement = { worker, numberOfWorkerElements: 0 };
this.workerSet.add(workerSetElement);
return workerSetElement;
}
+ private removeWorkerSetElement(workerSetElement: WorkerSetElement): void {
+ this.workerSet.delete(workerSetElement);
+ }
+
private async getWorkerSetElement(): Promise<WorkerSetElement> {
private async getWorkerSetElement(): Promise<WorkerSetElement> {
- let chosenWorkerSetElement: WorkerSetElement;
+ let chosenWorkerSetElement: WorkerSetElement
| undefined
;
for (const workerSetElement of this.workerSet) {
for (const workerSetElement of this.workerSet) {
- if (workerSetElement.numberOfWorkerElements < this.workerOptions.elementsPerWorker) {
+ if (workerSetElement.numberOfWorkerElements < this.workerOptions.elementsPerWorker
!
) {
chosenWorkerSetElement = workerSetElement;
break;
}
chosenWorkerSetElement = workerSetElement;
break;
}
@@
-127,13
+153,14
@@
export class WorkerSet extends WorkerAbstract<WorkerData> {
if (!chosenWorkerSetElement) {
chosenWorkerSetElement = this.addWorkerSetElement();
// Add worker set element sequentially to optimize memory at startup
if (!chosenWorkerSetElement) {
chosenWorkerSetElement = this.addWorkerSetElement();
// Add worker set element sequentially to optimize memory at startup
- this.workerOptions.workerStartDelay > 0 && (await sleep(this.workerOptions.workerStartDelay));
+ this.workerOptions.workerStartDelay! > 0 &&
+ (await sleep(this.workerOptions.workerStartDelay!));
}
return chosenWorkerSetElement;
}
private getWorkerSetElementByWorker(worker: Worker): WorkerSetElement | undefined {
}
return chosenWorkerSetElement;
}
private getWorkerSetElementByWorker(worker: Worker): WorkerSetElement | undefined {
- let workerSetElt: WorkerSetElement;
+ let workerSetElt: WorkerSetElement
| undefined
;
for (const workerSetElement of this.workerSet) {
if (workerSetElement.worker.threadId === worker.threadId) {
workerSetElt = workerSetElement;
for (const workerSetElement of this.workerSet) {
if (workerSetElement.worker.threadId === worker.threadId) {
workerSetElt = workerSetElement;