-
Notifications
You must be signed in to change notification settings - Fork 83
/
Copy pathnode-worker.ts
30 lines (29 loc) · 1.21 KB
/
node-worker.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
// Mediocre shim
let Worker: typeof import('worker_threads').Worker;
const workerAdd = ";var __w=require('worker_threads');__w.parentPort.on('message',function(m){onmessage({data:m})}),postMessage=function(m,t){__w.parentPort.postMessage(m,t)},close=process.exit;self=global";
try {
Worker = require('worker_threads').Worker;
} catch(e) {
}
export default Worker ? <T>(c: string, _: number, msg: unknown, transfer: ArrayBuffer[], cb: (err: Error, msg: T) => void) => {
let done = false;
const w = new Worker(c + workerAdd, { eval: true })
.on('error', e => cb(e, null))
.on('message', m => cb(null, m))
.on('exit', c => {
if (c && !done) cb(new Error('exited with code ' + c), null);
});
w.postMessage(msg, transfer);
w.terminate = () => {
done = true;
return Worker.prototype.terminate.call(w);
}
return w;
} : (_: string, __: number, ___: unknown, ____: ArrayBuffer[], cb: (err: Error, msg: null) => void) => {
setImmediate(() => cb(new Error('async operations unsupported - update to Node 12+ (or Node 10-11 with the --experimental-worker CLI flag)'), null));
const NOP = () => {};
return {
terminate: NOP,
postMessage: NOP
} as unknown as import('worker_threads').Worker;
}