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.
374 lines
9.4 KiB
374 lines
9.4 KiB
2 weeks ago
|
"use strict";
|
||
|
|
||
|
// These use the global symbol registry so that multiple copies of this
|
||
|
// library can work together in case they are not deduped.
|
||
|
const GENSYNC_START = Symbol.for("gensync:v1:start");
|
||
|
const GENSYNC_SUSPEND = Symbol.for("gensync:v1:suspend");
|
||
|
|
||
|
const GENSYNC_EXPECTED_START = "GENSYNC_EXPECTED_START";
|
||
|
const GENSYNC_EXPECTED_SUSPEND = "GENSYNC_EXPECTED_SUSPEND";
|
||
|
const GENSYNC_OPTIONS_ERROR = "GENSYNC_OPTIONS_ERROR";
|
||
|
const GENSYNC_RACE_NONEMPTY = "GENSYNC_RACE_NONEMPTY";
|
||
|
const GENSYNC_ERRBACK_NO_CALLBACK = "GENSYNC_ERRBACK_NO_CALLBACK";
|
||
|
|
||
|
module.exports = Object.assign(
|
||
|
function gensync(optsOrFn) {
|
||
|
let genFn = optsOrFn;
|
||
|
if (typeof optsOrFn !== "function") {
|
||
|
genFn = newGenerator(optsOrFn);
|
||
|
} else {
|
||
|
genFn = wrapGenerator(optsOrFn);
|
||
|
}
|
||
|
|
||
|
return Object.assign(genFn, makeFunctionAPI(genFn));
|
||
|
},
|
||
|
{
|
||
|
all: buildOperation({
|
||
|
name: "all",
|
||
|
arity: 1,
|
||
|
sync: function(args) {
|
||
|
const items = Array.from(args[0]);
|
||
|
return items.map(item => evaluateSync(item));
|
||
|
},
|
||
|
async: function(args, resolve, reject) {
|
||
|
const items = Array.from(args[0]);
|
||
|
|
||
|
if (items.length === 0) {
|
||
|
Promise.resolve().then(() => resolve([]));
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
let count = 0;
|
||
|
const results = items.map(() => undefined);
|
||
|
items.forEach((item, i) => {
|
||
|
evaluateAsync(
|
||
|
item,
|
||
|
val => {
|
||
|
results[i] = val;
|
||
|
count += 1;
|
||
|
|
||
|
if (count === results.length) resolve(results);
|
||
|
},
|
||
|
reject
|
||
|
);
|
||
|
});
|
||
|
},
|
||
|
}),
|
||
|
race: buildOperation({
|
||
|
name: "race",
|
||
|
arity: 1,
|
||
|
sync: function(args) {
|
||
|
const items = Array.from(args[0]);
|
||
|
if (items.length === 0) {
|
||
|
throw makeError("Must race at least 1 item", GENSYNC_RACE_NONEMPTY);
|
||
|
}
|
||
|
|
||
|
return evaluateSync(items[0]);
|
||
|
},
|
||
|
async: function(args, resolve, reject) {
|
||
|
const items = Array.from(args[0]);
|
||
|
if (items.length === 0) {
|
||
|
throw makeError("Must race at least 1 item", GENSYNC_RACE_NONEMPTY);
|
||
|
}
|
||
|
|
||
|
for (const item of items) {
|
||
|
evaluateAsync(item, resolve, reject);
|
||
|
}
|
||
|
},
|
||
|
}),
|
||
|
}
|
||
|
);
|
||
|
|
||
|
/**
|
||
|
* Given a generator function, return the standard API object that executes
|
||
|
* the generator and calls the callbacks.
|
||
|
*/
|
||
|
function makeFunctionAPI(genFn) {
|
||
|
const fns = {
|
||
|
sync: function(...args) {
|
||
|
return evaluateSync(genFn.apply(this, args));
|
||
|
},
|
||
|
async: function(...args) {
|
||
|
return new Promise((resolve, reject) => {
|
||
|
evaluateAsync(genFn.apply(this, args), resolve, reject);
|
||
|
});
|
||
|
},
|
||
|
errback: function(...args) {
|
||
|
const cb = args.pop();
|
||
|
if (typeof cb !== "function") {
|
||
|
throw makeError(
|
||
|
"Asynchronous function called without callback",
|
||
|
GENSYNC_ERRBACK_NO_CALLBACK
|
||
|
);
|
||
|
}
|
||
|
|
||
|
let gen;
|
||
|
try {
|
||
|
gen = genFn.apply(this, args);
|
||
|
} catch (err) {
|
||
|
cb(err);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
evaluateAsync(gen, val => cb(undefined, val), err => cb(err));
|
||
|
},
|
||
|
};
|
||
|
return fns;
|
||
|
}
|
||
|
|
||
|
function assertTypeof(type, name, value, allowUndefined) {
|
||
|
if (
|
||
|
typeof value === type ||
|
||
|
(allowUndefined && typeof value === "undefined")
|
||
|
) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
let msg;
|
||
|
if (allowUndefined) {
|
||
|
msg = `Expected opts.${name} to be either a ${type}, or undefined.`;
|
||
|
} else {
|
||
|
msg = `Expected opts.${name} to be a ${type}.`;
|
||
|
}
|
||
|
|
||
|
throw makeError(msg, GENSYNC_OPTIONS_ERROR);
|
||
|
}
|
||
|
function makeError(msg, code) {
|
||
|
return Object.assign(new Error(msg), { code });
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* Given an options object, return a new generator that dispatches the
|
||
|
* correct handler based on sync or async execution.
|
||
|
*/
|
||
|
function newGenerator({ name, arity, sync, async, errback }) {
|
||
|
assertTypeof("string", "name", name, true /* allowUndefined */);
|
||
|
assertTypeof("number", "arity", arity, true /* allowUndefined */);
|
||
|
assertTypeof("function", "sync", sync);
|
||
|
assertTypeof("function", "async", async, true /* allowUndefined */);
|
||
|
assertTypeof("function", "errback", errback, true /* allowUndefined */);
|
||
|
if (async && errback) {
|
||
|
throw makeError(
|
||
|
"Expected one of either opts.async or opts.errback, but got _both_.",
|
||
|
GENSYNC_OPTIONS_ERROR
|
||
|
);
|
||
|
}
|
||
|
|
||
|
if (typeof name !== "string") {
|
||
|
let fnName;
|
||
|
if (errback && errback.name && errback.name !== "errback") {
|
||
|
fnName = errback.name;
|
||
|
}
|
||
|
if (async && async.name && async.name !== "async") {
|
||
|
fnName = async.name.replace(/Async$/, "");
|
||
|
}
|
||
|
if (sync && sync.name && sync.name !== "sync") {
|
||
|
fnName = sync.name.replace(/Sync$/, "");
|
||
|
}
|
||
|
|
||
|
if (typeof fnName === "string") {
|
||
|
name = fnName;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (typeof arity !== "number") {
|
||
|
arity = sync.length;
|
||
|
}
|
||
|
|
||
|
return buildOperation({
|
||
|
name,
|
||
|
arity,
|
||
|
sync: function(args) {
|
||
|
return sync.apply(this, args);
|
||
|
},
|
||
|
async: function(args, resolve, reject) {
|
||
|
if (async) {
|
||
|
async.apply(this, args).then(resolve, reject);
|
||
|
} else if (errback) {
|
||
|
errback.call(this, ...args, (err, value) => {
|
||
|
if (err == null) resolve(value);
|
||
|
else reject(err);
|
||
|
});
|
||
|
} else {
|
||
|
resolve(sync.apply(this, args));
|
||
|
}
|
||
|
},
|
||
|
});
|
||
|
}
|
||
|
|
||
|
function wrapGenerator(genFn) {
|
||
|
return setFunctionMetadata(genFn.name, genFn.length, function(...args) {
|
||
|
return genFn.apply(this, args);
|
||
|
});
|
||
|
}
|
||
|
|
||
|
function buildOperation({ name, arity, sync, async }) {
|
||
|
return setFunctionMetadata(name, arity, function*(...args) {
|
||
|
const resume = yield GENSYNC_START;
|
||
|
if (!resume) {
|
||
|
// Break the tail call to avoid a bug in V8 v6.X with --harmony enabled.
|
||
|
const res = sync.call(this, args);
|
||
|
return res;
|
||
|
}
|
||
|
|
||
|
let result;
|
||
|
try {
|
||
|
async.call(
|
||
|
this,
|
||
|
args,
|
||
|
value => {
|
||
|
if (result) return;
|
||
|
|
||
|
result = { value };
|
||
|
resume();
|
||
|
},
|
||
|
err => {
|
||
|
if (result) return;
|
||
|
|
||
|
result = { err };
|
||
|
resume();
|
||
|
}
|
||
|
);
|
||
|
} catch (err) {
|
||
|
result = { err };
|
||
|
resume();
|
||
|
}
|
||
|
|
||
|
// Suspend until the callbacks run. Will resume synchronously if the
|
||
|
// callback was already called.
|
||
|
yield GENSYNC_SUSPEND;
|
||
|
|
||
|
if (result.hasOwnProperty("err")) {
|
||
|
throw result.err;
|
||
|
}
|
||
|
|
||
|
return result.value;
|
||
|
});
|
||
|
}
|
||
|
|
||
|
function evaluateSync(gen) {
|
||
|
let value;
|
||
|
while (!({ value } = gen.next()).done) {
|
||
|
assertStart(value, gen);
|
||
|
}
|
||
|
return value;
|
||
|
}
|
||
|
|
||
|
function evaluateAsync(gen, resolve, reject) {
|
||
|
(function step() {
|
||
|
try {
|
||
|
let value;
|
||
|
while (!({ value } = gen.next()).done) {
|
||
|
assertStart(value, gen);
|
||
|
|
||
|
// If this throws, it is considered to have broken the contract
|
||
|
// established for async handlers. If these handlers are called
|
||
|
// synchronously, it is also considered bad behavior.
|
||
|
let sync = true;
|
||
|
let didSyncResume = false;
|
||
|
const out = gen.next(() => {
|
||
|
if (sync) {
|
||
|
didSyncResume = true;
|
||
|
} else {
|
||
|
step();
|
||
|
}
|
||
|
});
|
||
|
sync = false;
|
||
|
|
||
|
assertSuspend(out, gen);
|
||
|
|
||
|
if (!didSyncResume) {
|
||
|
// Callback wasn't called synchronously, so break out of the loop
|
||
|
// and let it call 'step' later.
|
||
|
return;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return resolve(value);
|
||
|
} catch (err) {
|
||
|
return reject(err);
|
||
|
}
|
||
|
})();
|
||
|
}
|
||
|
|
||
|
function assertStart(value, gen) {
|
||
|
if (value === GENSYNC_START) return;
|
||
|
|
||
|
throwError(
|
||
|
gen,
|
||
|
makeError(
|
||
|
`Got unexpected yielded value in gensync generator: ${JSON.stringify(
|
||
|
value
|
||
|
)}. Did you perhaps mean to use 'yield*' instead of 'yield'?`,
|
||
|
GENSYNC_EXPECTED_START
|
||
|
)
|
||
|
);
|
||
|
}
|
||
|
function assertSuspend({ value, done }, gen) {
|
||
|
if (!done && value === GENSYNC_SUSPEND) return;
|
||
|
|
||
|
throwError(
|
||
|
gen,
|
||
|
makeError(
|
||
|
done
|
||
|
? "Unexpected generator completion. If you get this, it is probably a gensync bug."
|
||
|
: `Expected GENSYNC_SUSPEND, got ${JSON.stringify(
|
||
|
value
|
||
|
)}. If you get this, it is probably a gensync bug.`,
|
||
|
GENSYNC_EXPECTED_SUSPEND
|
||
|
)
|
||
|
);
|
||
|
}
|
||
|
|
||
|
function throwError(gen, err) {
|
||
|
// Call `.throw` so that users can step in a debugger to easily see which
|
||
|
// 'yield' passed an unexpected value. If the `.throw` call didn't throw
|
||
|
// back to the generator, we explicitly do it to stop the error
|
||
|
// from being swallowed by user code try/catches.
|
||
|
if (gen.throw) gen.throw(err);
|
||
|
throw err;
|
||
|
}
|
||
|
|
||
|
function isIterable(value) {
|
||
|
return (
|
||
|
!!value &&
|
||
|
(typeof value === "object" || typeof value === "function") &&
|
||
|
!value[Symbol.iterator]
|
||
|
);
|
||
|
}
|
||
|
|
||
|
function setFunctionMetadata(name, arity, fn) {
|
||
|
if (typeof name === "string") {
|
||
|
// This should always work on the supported Node versions, but for the
|
||
|
// sake of users that are compiling to older versions, we check for
|
||
|
// configurability so we don't throw.
|
||
|
const nameDesc = Object.getOwnPropertyDescriptor(fn, "name");
|
||
|
if (!nameDesc || nameDesc.configurable) {
|
||
|
Object.defineProperty(
|
||
|
fn,
|
||
|
"name",
|
||
|
Object.assign(nameDesc || {}, {
|
||
|
configurable: true,
|
||
|
value: name,
|
||
|
})
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (typeof arity === "number") {
|
||
|
const lengthDesc = Object.getOwnPropertyDescriptor(fn, "length");
|
||
|
if (!lengthDesc || lengthDesc.configurable) {
|
||
|
Object.defineProperty(
|
||
|
fn,
|
||
|
"length",
|
||
|
Object.assign(lengthDesc || {}, {
|
||
|
configurable: true,
|
||
|
value: arity,
|
||
|
})
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return fn;
|
||
|
}
|