58 lines
1.6 KiB
TypeScript
58 lines
1.6 KiB
TypeScript
|
/**
|
||
|
|
* @license
|
||
|
|
* Copyright Google Inc. All Rights Reserved.
|
||
|
|
*
|
||
|
|
* Use of this source code is governed by an MIT-style license that can be
|
||
|
|
* found in the LICENSE file at https://angular.io/license
|
||
|
|
*/
|
||
|
|
|
||
|
|
/// <reference types="node" />
|
||
|
|
|
||
|
|
import * as cluster from 'cluster';
|
||
|
|
|
||
|
|
import {CompileFn, CreateCompileFn} from '../api';
|
||
|
|
|
||
|
|
import {MessageToWorker} from './api';
|
||
|
|
import {sendMessageToMaster} from './utils';
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* A cluster worker is responsible for processing one task (i.e. one format property for a specific
|
||
|
|
* entry-point) at a time and reporting results back to the cluster master.
|
||
|
|
*/
|
||
|
|
export class ClusterWorker {
|
||
|
|
private compile: CompileFn;
|
||
|
|
|
||
|
|
constructor(createCompileFn: CreateCompileFn) {
|
||
|
|
if (cluster.isMaster) {
|
||
|
|
throw new Error('Tried to instantiate `ClusterWorker` on the master process.');
|
||
|
|
}
|
||
|
|
|
||
|
|
this.compile =
|
||
|
|
createCompileFn((_task, outcome) => sendMessageToMaster({type: 'task-completed', outcome}));
|
||
|
|
}
|
||
|
|
|
||
|
|
run(): Promise<void> {
|
||
|
|
// Listen for `ProcessTaskMessage`s and process tasks.
|
||
|
|
cluster.worker.on('message', (msg: MessageToWorker) => {
|
||
|
|
try {
|
||
|
|
switch (msg.type) {
|
||
|
|
case 'process-task':
|
||
|
|
return this.compile(msg.task);
|
||
|
|
default:
|
||
|
|
throw new Error(
|
||
|
|
`Invalid message received on worker #${cluster.worker.id}: ${JSON.stringify(msg)}`);
|
||
|
|
}
|
||
|
|
} catch (err) {
|
||
|
|
sendMessageToMaster({
|
||
|
|
type: 'error',
|
||
|
|
error: (err instanceof Error) ? (err.stack || err.message) : err,
|
||
|
|
});
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
// Return a promise that is never resolved.
|
||
|
|
return new Promise(() => undefined);
|
||
|
|
}
|
||
|
|
}
|