minilink

README.md

Minilink

comlink から proxy やらなんやら抜いたもの。 気が向いたら npm に投げる

使い方

// worker thread
import {expose} from './minilink'
expose({
  async foo(args) {
    return args.v + 1;
  }
})

// main thread
import {wrap} from './minilink'
const api = wrap(new Worker('/myworker.js'));
const ret = await api.call('foo', {v:1});
console.log(ret); // => 2

minilink.ts

export function expose(api: any) {
  self.addEventListener("message", ev => {
    if (ev.data && ev.data.scope === "minilink") {
      const cmd = ev.data.cmd;
      const id = ev.data.id;
      const args = ev.data.args || {};
      const func = api[cmd];
      func(args)
        .then((result: any) => {
          // @ts-ignore
          self.postMessage({ result, id });
        })
        .catch((e: any) => {
          // @ts-ignore
          self.postMessage({ result: e, id, error: true });
        });
    }
  });
}

const _sender = new Map();
export function wrap(endpoint: Worker) {
  endpoint.addEventListener("message", ev => {
    const { error, id, result } = ev.data;
    if (id) {
      const obj = _sender.get(id);
      _sender.delete(id);
      if (error) {
        obj.reject(result);
      } else {
        obj.resolve(result);
      }
    }
  });
  return {
    call(cmd: string, args: any, transferables?: Array<any>) {
      const id = Math.random();
      return new Promise((resolve, reject) => {
        _sender.set(id, {
          resolve,
          reject
        });
        endpoint.postMessage(
          {
            scope: "minilink",
            cmd,
            id,
            args
          },
          transferables || []
        );
      });
    }
  };
}