@travetto/worker 7.0.0-rc.0 → 7.0.0-rc.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/package.json +2 -2
- package/src/ipc.ts +20 -20
- package/src/pool.ts +37 -37
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@travetto/worker",
|
|
3
|
-
"version": "7.0.0-rc.
|
|
3
|
+
"version": "7.0.0-rc.2",
|
|
4
4
|
"description": "Process management utilities, with a focus on inter-process communication",
|
|
5
5
|
"keywords": [
|
|
6
6
|
"exec",
|
|
@@ -25,7 +25,7 @@
|
|
|
25
25
|
"directory": "module/worker"
|
|
26
26
|
},
|
|
27
27
|
"dependencies": {
|
|
28
|
-
"@travetto/runtime": "^7.0.0-rc.
|
|
28
|
+
"@travetto/runtime": "^7.0.0-rc.2",
|
|
29
29
|
"generic-pool": "^3.9.0"
|
|
30
30
|
},
|
|
31
31
|
"travetto": {
|
package/src/ipc.ts
CHANGED
|
@@ -9,20 +9,20 @@ import { ShutdownManager, Util } from '@travetto/runtime';
|
|
|
9
9
|
export class IpcChannel<V = unknown> {
|
|
10
10
|
|
|
11
11
|
#emitter = new EventEmitter();
|
|
12
|
-
|
|
12
|
+
subProcess: NodeJS.Process | ChildProcess;
|
|
13
13
|
parentId: number;
|
|
14
14
|
|
|
15
|
-
constructor(
|
|
16
|
-
this.
|
|
17
|
-
this.parentId =
|
|
15
|
+
constructor(subProcess: NodeJS.Process | ChildProcess = process) {
|
|
16
|
+
this.subProcess = subProcess;
|
|
17
|
+
this.parentId = subProcess instanceof ChildProcess ? process.pid : process.ppid;
|
|
18
18
|
|
|
19
19
|
// Close on shutdown
|
|
20
20
|
ShutdownManager.onGracefulShutdown(() => this.destroy());
|
|
21
21
|
|
|
22
|
-
this.
|
|
23
|
-
console.debug('Received', { pid: this.parentId, id: this.id, type:
|
|
24
|
-
this.#emitter.emit(
|
|
25
|
-
this.#emitter.emit('*',
|
|
22
|
+
this.subProcess.on('message', (event: { type: string }) => {
|
|
23
|
+
console.debug('Received', { pid: this.parentId, id: this.id, type: event.type });
|
|
24
|
+
this.#emitter.emit(event.type, event);
|
|
25
|
+
this.#emitter.emit('*', event);
|
|
26
26
|
});
|
|
27
27
|
}
|
|
28
28
|
|
|
@@ -30,14 +30,14 @@ export class IpcChannel<V = unknown> {
|
|
|
30
30
|
* Gets channel unique identifier
|
|
31
31
|
*/
|
|
32
32
|
get id(): number | undefined {
|
|
33
|
-
return this.
|
|
33
|
+
return this.subProcess.pid;
|
|
34
34
|
}
|
|
35
35
|
|
|
36
36
|
/**
|
|
37
37
|
* Determines if channel is active
|
|
38
38
|
*/
|
|
39
39
|
get active(): boolean {
|
|
40
|
-
return (this.
|
|
40
|
+
return (this.subProcess instanceof ChildProcess) ? !this.subProcess.killed : !!this.subProcess.connected;
|
|
41
41
|
}
|
|
42
42
|
|
|
43
43
|
/**
|
|
@@ -47,17 +47,17 @@ export class IpcChannel<V = unknown> {
|
|
|
47
47
|
console.debug('Sending', { pid: this.parentId, id: this.id, eventType });
|
|
48
48
|
if (!this.active) {
|
|
49
49
|
throw new Error('Cannot send message to inactive process');
|
|
50
|
-
} else if (this.
|
|
51
|
-
this.
|
|
50
|
+
} else if (this.subProcess.send && this.subProcess.connected) {
|
|
51
|
+
this.subProcess.send({ ...(data ?? {}), type: eventType }, undefined, undefined, (error) => error && console.error(error));
|
|
52
52
|
} else {
|
|
53
|
-
throw new Error('this.
|
|
53
|
+
throw new Error('this.subProcess.send was not defined');
|
|
54
54
|
}
|
|
55
55
|
}
|
|
56
56
|
|
|
57
57
|
/**
|
|
58
58
|
* Listen for a specific message type
|
|
59
59
|
*/
|
|
60
|
-
on(eventType: string, callback: (
|
|
60
|
+
on(eventType: string, callback: (event: V & { type: string }) => unknown | void): () => void {
|
|
61
61
|
this.#emitter.on(eventType, callback);
|
|
62
62
|
return () => this.off(eventType, callback);
|
|
63
63
|
}
|
|
@@ -65,7 +65,7 @@ export class IpcChannel<V = unknown> {
|
|
|
65
65
|
/**
|
|
66
66
|
* Remove event listener
|
|
67
67
|
*/
|
|
68
|
-
off(eventType: string, callback: (
|
|
68
|
+
off(eventType: string, callback: (event: V & { type: string }) => unknown | void): void {
|
|
69
69
|
this.#emitter.off(eventType, callback);
|
|
70
70
|
}
|
|
71
71
|
|
|
@@ -73,7 +73,7 @@ export class IpcChannel<V = unknown> {
|
|
|
73
73
|
* Listen for a specific message type, once
|
|
74
74
|
*/
|
|
75
75
|
once(eventType: string): Promise<V & { type: string }> {
|
|
76
|
-
return new Promise(
|
|
76
|
+
return new Promise(resolve => this.#emitter.once(eventType, resolve));
|
|
77
77
|
}
|
|
78
78
|
|
|
79
79
|
/**
|
|
@@ -83,9 +83,9 @@ export class IpcChannel<V = unknown> {
|
|
|
83
83
|
if (this.active) {
|
|
84
84
|
try {
|
|
85
85
|
console.debug('Killing', { pid: this.parentId, id: this.id });
|
|
86
|
-
if (this.
|
|
87
|
-
const complete = new Promise<void>(
|
|
88
|
-
this.
|
|
86
|
+
if (this.subProcess instanceof ChildProcess) {
|
|
87
|
+
const complete = new Promise<void>(resolve => this.subProcess.on('close', resolve));
|
|
88
|
+
this.subProcess.kill();
|
|
89
89
|
await Promise.race([complete, Util.nonBlockingTimeout(1000)]);
|
|
90
90
|
}
|
|
91
91
|
} catch { }
|
|
@@ -98,7 +98,7 @@ export class IpcChannel<V = unknown> {
|
|
|
98
98
|
*/
|
|
99
99
|
release(): void {
|
|
100
100
|
console.debug('Released', { pid: this.parentId, id: this.id });
|
|
101
|
-
this.
|
|
101
|
+
this.subProcess.removeAllListeners();
|
|
102
102
|
this.#emitter.removeAllListeners();
|
|
103
103
|
}
|
|
104
104
|
}
|
package/src/pool.ts
CHANGED
|
@@ -3,7 +3,7 @@ import { Options, Pool, createPool } from 'generic-pool';
|
|
|
3
3
|
|
|
4
4
|
import { Env, Util, AsyncQueue } from '@travetto/runtime';
|
|
5
5
|
|
|
6
|
-
type
|
|
6
|
+
type IterableSource<I> = Iterable<I> | AsyncIterable<I>;
|
|
7
7
|
type WorkerExecutor<I, O> = (input: I, idx: number) => Promise<O>;
|
|
8
8
|
|
|
9
9
|
/**
|
|
@@ -22,11 +22,11 @@ type WorkerFactoryInput<I, O = unknown> = Partial<Worker<I, O>> & { execute: Wor
|
|
|
22
22
|
type WorkerInput<I, O> = (() => WorkerFactoryInput<I, O>) | WorkerExecutor<I, O>;
|
|
23
23
|
type WorkPoolConfig<I, O> = Options & {
|
|
24
24
|
onComplete?: (output: O, input: I, finishIdx: number) => void;
|
|
25
|
-
onError?(
|
|
25
|
+
onError?(event: Error, input: I, finishIdx: number): (unknown | Promise<unknown>);
|
|
26
26
|
shutdown?: AbortSignal;
|
|
27
27
|
};
|
|
28
28
|
|
|
29
|
-
const isWorkerFactory = <I, O>(
|
|
29
|
+
const isWorkerFactory = <I, O>(value: WorkerInput<I, O>): value is (() => WorkerFactoryInput<I, O>) => value.length === 0;
|
|
30
30
|
|
|
31
31
|
/**
|
|
32
32
|
* Work pool support
|
|
@@ -37,10 +37,10 @@ export class WorkPool {
|
|
|
37
37
|
static DEFAULT_SIZE = Math.max(Math.trunc(WorkPool.MAX_SIZE * .75), 4);
|
|
38
38
|
|
|
39
39
|
/** Build worker pool */
|
|
40
|
-
static #buildPool<I, O>(input: WorkerInput<I, O>,
|
|
40
|
+
static #buildPool<I, O>(input: WorkerInput<I, O>, options?: WorkPoolConfig<I, O>): Pool<Worker<I, O>> {
|
|
41
41
|
let pendingAcquires = 0;
|
|
42
42
|
|
|
43
|
-
const trace = /@travetto\/worker/.test(Env.DEBUG.
|
|
43
|
+
const trace = /@travetto\/worker/.test(Env.DEBUG.value ?? '');
|
|
44
44
|
|
|
45
45
|
// Create the pool
|
|
46
46
|
const pool = createPool({
|
|
@@ -58,22 +58,22 @@ export class WorkPool {
|
|
|
58
58
|
pendingAcquires -= 1;
|
|
59
59
|
}
|
|
60
60
|
},
|
|
61
|
-
async destroy(
|
|
61
|
+
async destroy(worker) {
|
|
62
62
|
if (trace) {
|
|
63
|
-
console.debug('Destroying', { pid: process.pid, worker:
|
|
63
|
+
console.debug('Destroying', { pid: process.pid, worker: worker.id });
|
|
64
64
|
}
|
|
65
|
-
return
|
|
65
|
+
return worker.destroy?.();
|
|
66
66
|
},
|
|
67
|
-
validate: async (
|
|
67
|
+
validate: async (worker: Worker<I, O>) => worker.active
|
|
68
68
|
}, {
|
|
69
69
|
evictionRunIntervalMillis: 5000,
|
|
70
|
-
...(
|
|
71
|
-
max:
|
|
72
|
-
min:
|
|
70
|
+
...(options ?? {}),
|
|
71
|
+
max: options?.max ?? WorkPool.DEFAULT_SIZE,
|
|
72
|
+
min: options?.min ?? 1,
|
|
73
73
|
});
|
|
74
74
|
|
|
75
75
|
// Listen for shutdown
|
|
76
|
-
|
|
76
|
+
options?.shutdown?.addEventListener('abort', async () => {
|
|
77
77
|
while (pendingAcquires) {
|
|
78
78
|
await Util.nonBlockingTimeout(10);
|
|
79
79
|
}
|
|
@@ -87,17 +87,17 @@ export class WorkPool {
|
|
|
87
87
|
/**
|
|
88
88
|
* Process a given input source and worker, and fire on completion
|
|
89
89
|
*/
|
|
90
|
-
static async run<I, O>(workerFactory: WorkerInput<I, O>,
|
|
90
|
+
static async run<I, O>(workerFactory: WorkerInput<I, O>, source: IterableSource<I>, options: WorkPoolConfig<I, O> = {}): Promise<void> {
|
|
91
91
|
|
|
92
|
-
const trace = /@travetto\/worker/.test(Env.DEBUG.
|
|
92
|
+
const trace = /@travetto\/worker/.test(Env.DEBUG.value ?? '');
|
|
93
93
|
const pending = new Set<Promise<unknown>>();
|
|
94
94
|
const errors: Error[] = [];
|
|
95
95
|
let inputIdx = 0;
|
|
96
96
|
let finishIdx = 0;
|
|
97
97
|
|
|
98
|
-
const pool = this.#buildPool(workerFactory,
|
|
98
|
+
const pool = this.#buildPool(workerFactory, options);
|
|
99
99
|
|
|
100
|
-
for await (const nextInput of
|
|
100
|
+
for await (const nextInput of source) {
|
|
101
101
|
const worker = await pool.acquire()!;
|
|
102
102
|
|
|
103
103
|
if (trace) {
|
|
@@ -105,10 +105,10 @@ export class WorkPool {
|
|
|
105
105
|
}
|
|
106
106
|
|
|
107
107
|
const completion = worker.execute(nextInput, inputIdx += 1)
|
|
108
|
-
.then(
|
|
109
|
-
.catch(
|
|
110
|
-
errors.push(
|
|
111
|
-
|
|
108
|
+
.then(output => options.onComplete?.(output, nextInput, finishIdx += 1))
|
|
109
|
+
.catch(error => {
|
|
110
|
+
errors.push(error);
|
|
111
|
+
options?.onError?.(error, nextInput, finishIdx += 1);
|
|
112
112
|
}) // Catch error
|
|
113
113
|
.finally(async () => {
|
|
114
114
|
if (trace) {
|
|
@@ -140,36 +140,36 @@ export class WorkPool {
|
|
|
140
140
|
/**
|
|
141
141
|
* Process a given input source as an async iterable
|
|
142
142
|
*/
|
|
143
|
-
static runStream<I, O>(worker: WorkerInput<I, O>, input:
|
|
144
|
-
const
|
|
143
|
+
static runStream<I, O>(worker: WorkerInput<I, O>, input: IterableSource<I>, options?: WorkPoolConfig<I, O>): AsyncIterable<O> {
|
|
144
|
+
const queue = new AsyncQueue<O>();
|
|
145
145
|
const result = this.run(worker, input, {
|
|
146
|
-
...
|
|
147
|
-
onComplete: (
|
|
148
|
-
|
|
149
|
-
|
|
146
|
+
...options,
|
|
147
|
+
onComplete: (event, value, finishIdx) => {
|
|
148
|
+
queue.add(event);
|
|
149
|
+
options?.onComplete?.(event, value, finishIdx);
|
|
150
150
|
}
|
|
151
151
|
});
|
|
152
|
-
result.finally(() =>
|
|
153
|
-
return
|
|
152
|
+
result.finally(() => queue.close());
|
|
153
|
+
return queue;
|
|
154
154
|
}
|
|
155
155
|
|
|
156
156
|
/**
|
|
157
157
|
* Process a given input source as an async iterable with progress information
|
|
158
158
|
*/
|
|
159
|
-
static runStreamProgress<I, O>(worker: WorkerInput<I, O>, input:
|
|
159
|
+
static runStreamProgress<I, O>(worker: WorkerInput<I, O>, input: IterableSource<I>, total: number, options?: WorkPoolConfig<I, O>): AsyncIterable<{
|
|
160
160
|
idx: number;
|
|
161
161
|
value: O;
|
|
162
162
|
total: number;
|
|
163
163
|
}> {
|
|
164
|
-
const
|
|
164
|
+
const queue = new AsyncQueue<{ idx: number, value: O, total: number }>();
|
|
165
165
|
const result = this.run(worker, input, {
|
|
166
|
-
...
|
|
167
|
-
onComplete: (
|
|
168
|
-
|
|
169
|
-
|
|
166
|
+
...options,
|
|
167
|
+
onComplete: (event, value, finishIdx) => {
|
|
168
|
+
queue.add({ value: event, idx: finishIdx, total });
|
|
169
|
+
options?.onComplete?.(event, value, finishIdx);
|
|
170
170
|
}
|
|
171
171
|
});
|
|
172
|
-
result.finally(() =>
|
|
173
|
-
return
|
|
172
|
+
result.finally(() => queue.close());
|
|
173
|
+
return queue;
|
|
174
174
|
}
|
|
175
175
|
}
|