explorigin/portfolio
1
0
Fork 0
Code Releases Activity

Add cleanup() to the API.

Browse Source
This commit is contained in:
Timothy Farrell 2016-12-27 19:07:30 -06:00
parent a01a9a53b7
commit da86fc6045
1 changed files with 29 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
packages/portal/src/index.js
View File

@ -37,6 +37,7 @@ export function WorkerPortal(context, worker, serialize) {
? (type, destination, id, params) => serialize(type, destination, id, params, tryJSON) ? (type, destination, id, params) => serialize(type, destination, id, params, tryJSON)
: tryJSON; : tryJSON;
let callCount = 0; let callCount = 0;
let enabled = false;
function post(type, id, destination, params) { function post(type, id, destination, params) {
_worker.postMessage(_serialize(type, destination, id, params)); _worker.postMessage(_serialize(type, destination, id, params));
@ -85,6 +86,9 @@ export function WorkerPortal(context, worker, serialize) {
function injectionPointFactory(fnId, callbackFactory) { function injectionPointFactory(fnId, callbackFactory) {
return () => return () =>
new Promise((resolve, reject) => { new Promise((resolve, reject) => {
if (!enabled) {
reject(new Error('Portal disabled'));
}
const id = callCount; const id = callCount;
callCount += 1; callCount += 1;
responseMap.set(id, [callbackFactory ? callbackFactory(resolve) : resolve, reject]); responseMap.set(id, [callbackFactory ? callbackFactory(resolve) : resolve, reject]);
@ -98,17 +102,36 @@ export function WorkerPortal(context, worker, serialize) {
linkedFunctionNames.forEach((fnName, index) => { linkedFunctionNames.forEach((fnName, index) => {
externalInterface[fnName] = injectionPointFactory(index); externalInterface[fnName] = injectionPointFactory(index);
}); });
if (!isWorker()) {
externalInterface._cleanup = injectionPointFactory(linkedFunctionNames.length, resolve =>
resolve(cleanup())
);
}
enabled = true;
resolve(externalInterface); resolve(externalInterface);
return contextIndex; return contextIndex;
}; };
} }
function cleanup() {
enabled = false;
_worker.removeEventListener('message', dispatcher);
for (let key of responseMap.keys()) {
try {
responseMap.get(key)[1](new Error('Portal cleanup called.'));
} catch (e) {}
}
}
_worker.addEventListener('message', dispatcher); _worker.addEventListener('message', dispatcher);
return isWorker() if (isWorker()) {
? new Promise(resolve => { return new Promise(resolve => {
contextIndex.splice(0, 0, '__init'); contextIndex.splice(0, 0, '__init', '__cleanup');
context.__init = resolveExternalInterfaceFactory(resolve); context.__init = resolveExternalInterfaceFactory(resolve);
}) context.__cleanup = cleanup;
: injectionPointFactory(0, resolveExternalInterfaceFactory)(contextIndex); });
}
return injectionPointFactory(0, resolveExternalInterfaceFactory)(contextIndex);
} }