node-worker.ts 1.2 KB

123456789101112131415161718192021222324252627282930
  1. // Mediocre shim
  2. let Worker: typeof import('worker_threads').Worker;
  3. 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";
  4. try {
  5. Worker = require('worker_threads').Worker;
  6. } catch(e) {
  7. }
  8. export default Worker ? <T>(c: string, _: number, msg: unknown, transfer: ArrayBuffer[], cb: (err: Error, msg: T) => void) => {
  9. let done = false;
  10. const w = new Worker(c + workerAdd, { eval: true })
  11. .on('error', e => cb(e, null))
  12. .on('message', m => cb(null, m))
  13. .on('exit', c => {
  14. if (c && !done) cb(new Error('exited with code ' + c), null);
  15. });
  16. w.postMessage(msg, transfer);
  17. w.terminate = () => {
  18. done = true;
  19. return Worker.prototype.terminate.call(w);
  20. }
  21. return w;
  22. } : (_: string, __: number, ___: unknown, ____: ArrayBuffer[], cb: (err: Error, msg: null) => void) => {
  23. setImmediate(() => cb(new Error('async operations unsupported - update to Node 12+ (or Node 10-11 with the --experimental-worker CLI flag)'), null));
  24. const NOP = () => {};
  25. return {
  26. terminate: NOP,
  27. postMessage: NOP
  28. } as unknown as import('worker_threads').Worker;
  29. }