7.TypeScript(ts) worker_threads.Worker类多线程代码示范
TypeScript(ts) worker_threads.Worker类代码示范
TypeScript中worker_threads.Worker类的典型用法示范
1: Promise
return new Promise((resolve, reject) => {
const worker = new workerThreads.Worker(__filename, {
workerData: script
});
worker.on('message', resolve);
worker.on('error', reject);
worker.on('exit', (code) => {
if (code !== 0)
reject(new Error(`Worker stopped with exit code ${code}`));
});
});
2: initialize
initialize() {
this._worker = new Worker(path.resolve(__dirname, './threadChild.js'), {
eval: false,
stderr: true,
stdout: true,
workerData: {
cwd: process.cwd(),
env: {
...process.env,
JEST_WORKER_ID: String(this._options.workerId),
} as NodeJS.ProcessEnv,
// Suppress --debug / --inspect flags while preserving others (like --harmony).
execArgv: process.execArgv.filter(v => !/^--(debug|inspect)/.test(v)),
silent: true,
...this._options.forkOptions,
},
});
this._worker.on('message', this.onMessage.bind(this));
this._worker.on('exit', this.onExit.bind(this));
this._worker.postMessage([
CHILD_MESSAGE_INITIALIZE,
false,
this._options.workerPath,
this._options.setupArgs,
]);
this._retries++;
// If we exceeded the amount of retries, we will emulate an error reply
// tt coming from the child. This avoids code duplication related with cleaning
// the queue, and scheduling the next call.
if (this._retries > this._options.maxRetries) {
const error = new Error('Call retries were exceeded');
this.onMessage([
PARENT_MESSAGE_CLIENT_ERROR,
error.name,
error.message,
error.stack!,
{type: 'WorkerError'},
]);
}
}
3: createContext
};
} else {
const script = workerThreads.workerData;
workerThreads.parentPort!.postMessage(script);
}
}
{
const { port1, port2 } = new workerThreads.MessageChannel();
port1.on('message', (message) => console.log('received', message));
port2.postMessage({ foo: 'bar' });
}
{
if (workerThreads.isMainThread) {
const worker = new workerThreads.Worker(__filename);
const subChannel = new workerThreads.MessageChannel();
worker.postMessage({ hereIsYourPort: subChannel.port1 }, [subChannel.port1]);
subChannel.port2.on('message', (value) => {
console.log('received value:', value);
});//
worker.moveMessagePortToContext(new workerThreads.MessagePort(), createContext());
} else {
workerThreads.parentPort!.once('message', (value) => {
assert(value.hereIsYourPort instanceof workerThreads.MessagePort);
value.hereIsYourPort.postMessage('the worker is sending this');
value.hereIsYourPort.close();
});
}//
}
4: assert
});
};
} else {
const script = workerThreads.workerData;
workerThreads.parentPort!.postMessage(script);
}
}
{
const { port1, port2 } = new workerThreads.MessageChannel();
port1.on('message', (message) => console.log('received', message));
port2.postMessage({ foo: 'bar' });
}
{
if (workerThreads.isMainThread) {
const worker = new workerThreads.Worker(__filename);
const subChannel = new workerThreads.MessageChannel();
worker.postMessage({ hereIsYourPort: subChannel.port1 }, [subChannel.port1]);
subChannel.port2.on('message', (value) => {
console.log('received:', value);
});
} else {
workerThreads.parentPort!.once('message', (value) => {
assert(value.hereIsYourPort instanceof workerThreads.MessagePort);
value.hereIsYourPort.postMessage('the worker is sending this');
value.hereIsYourPort.close();
});
}//
}
5: send
send(request: ChildMessage, onProcessStart: OnStart, onProcessEnd: OnEnd) {
onProcessStart(this);
this._onProcessEnd = onProcessEnd;
this._retries = 0;
this._worker.postMessage(request);
}
6: onExit
onExit(exitCode: number) {
if (exitCode !== 0) {
this.initialize();
if (this._request) {
this._worker.postMessage(this._request);
}
} else {
this._shutdown();
}
}
7: onProcessEnd
send(request: ChildMessage, onProcessStart: OnStart, onProcessEnd: OnEnd) {
onProcessStart(this);
this._onProcessEnd = (...args) => {
// ttClean the request to avoid sending past requests to workers that fail
// ttwhile waiting for a new request (timers, unhandled rejections...)
this._request = null;
return onProcessEnd(...args);
};
this._request = request;
this._retries = 0;
this._worker.postMessage(request);
}
欢迎讨论,相互学习。
cdtxw@foxmail.com