Not sure why there's still stuff in the event loop, but whatever.

This commit is contained in:
Eddie Antonio Santos 2018-10-01 15:30:01 -06:00
parent 8e6203b29a
commit 260f237be2
No known key found for this signature in database
GPG key ID: DD411EA4D9509D87
2 changed files with 4 additions and 6 deletions

View file

@ -14,10 +14,11 @@ exports.LMLayer = class LMLayer {
predictWithContext({transform, context, _token}) {
let token = this._currentToken++;
let promise = new Promise((resolve, reject) => {
return new Promise((resolve, reject) => {
if (this._promises.has(token)) {
reject(`Existing request with token ${token}`);
}
this._promises.set(token, resolve);
this._worker.postMessage({
kind: 'predict',
@ -25,11 +26,7 @@ exports.LMLayer = class LMLayer {
transform,
context
});
this._promises.set(token, resolve);
});
return promise;
}
/**
@ -45,8 +42,8 @@ exports.LMLayer = class LMLayer {
}
if (kind === 'suggestions') {
accept(event.data);
this._promises.delete(token);
accept(event.data);
} else {
throw new Error(`Unknown message kind: ${kind}`);
}

View file

@ -10,6 +10,7 @@ test('Can provide context to LMLayer', async t => {
transform: { insert: 'D', deleteLeft: 0, deleteRight: 0 },
context: { left: '' }
});
let {suggestions} = message;
t.deepEqual(suggestions, [