You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.3 KiB
35 lines
1.3 KiB
4 months ago
|
import { MESSAGE_TYPE_RPC_TRANSFERABLE } from './message-types.js';
|
||
|
|
/**
|
||
|
|
* Setup worker methods which receive transferables from worker method params. This function should be executed on the main thread.
|
||
|
|
* @param worker worker instance
|
||
|
|
* @param methods an object whose key is method name and whose value is options how to pick transferables from method params
|
||
|
|
*/
|
||
|
|
export function setupTransferableMethodsOnMain(worker, methods) {
|
||
|
|
let c = 0;
|
||
|
|
const callbacks = {};
|
||
|
|
worker.addEventListener('message', e => {
|
||
|
|
const d = e.data;
|
||
|
|
if (d.type !== MESSAGE_TYPE_RPC_TRANSFERABLE) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
const f = callbacks[d.id];
|
||
|
|
if (f) {
|
||
|
|
delete callbacks[d.id];
|
||
|
|
if (d.error) {
|
||
|
|
f[1](Object.assign(Error(d.error.message), d.error));
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
f[0](d.result);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
Object.keys(methods).forEach(method => {
|
||
|
|
worker[method] = (...params) => new Promise((a, b) => {
|
||
|
|
const id = ++c;
|
||
|
|
callbacks[id] = [a, b];
|
||
|
|
const opts = methods[method];
|
||
|
|
worker.postMessage({ type: MESSAGE_TYPE_RPC_TRANSFERABLE, id, method, params }, opts.pickTransferablesFromParams ? opts.pickTransferablesFromParams(params) : []);
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
//# sourceMappingURL=on-main.js.map
|