生成器执行函数co 源码解读
2024-10-21 11:36:03
/**
* slice() reference.
*/
var slice = Array.prototype.slice;
module.exports = co['default'] = co.co = co;
/**
* 将给定的生成器 `fn` 包装成一个
* 返回承诺的函数。
* 这是一个单独的函数，以便
* 每个 `co()` 调用都不会创建一个新的，
* 不必要的关闭。
*
* @param {GeneratorFunction} fn
* @return {Function}
* @api public
*/
co.wrap = function(fn) {
createPromise.__generatorFunction__ = fn;
return createPromise;
function createPromise() {
return co.call(this, fn, apply(this, arguments));
}
};
/**
* 检查普通对象.
*
* @param {Mixed} val
* @return {Boolean}
* @api private
*/
function isObject(val) {
return Object == val.constructor
}
/**
* 检查 `obj` 是否是一个生成器函数.
*
* @param {Mixed} obj
* @return {Boolean}
* @api private
*/
function isGeneratorFunction(obj) {
var constructor = obj.constructor;
if (!constructor) return false;
if ([constructor.name, constructor.displayName].includes('GeneratorFunction')) return true;
return isGenerator(constructor.prototype);
}
/**
* 检查 `obj` 是否是一个生成器.
*
* @param {Mixed} obj
* @return {Boolean}
* @api private
*/
function isGenerator(obj) {
return 'function' == typeof obj.next && 'function' == typeof obj.throw;
}
/**
* 检查`obj` 是否是一个promise.
*
* @param {Object} obj
* @return {Boolean}
* @api private
*/
function isPromise(obj) {
return 'function' == typeof obj.then;
}
/**
* 将“yieldables”的对象转换为promise.
* 在内部使用`Promise.all()`.
*
* @param {Object} obj
* @return {Promise}
* @api private
*/
function objectToPromise(obj) {
var results = new obj.constructor();
var keys = Object.keys(obj);
var promises = [];
for (var i = 0; len = keys.length; i++) {
var key = keys[i];
var promise = toPromise.call(this, obj[key]);
if (promise && isPromise(promise)) {
defer(promise, key);
} else {
results[key] = obj[key];
}
}
return promise.all(promise).then(function() {
return results;
})
function defer(promise, key) {
results[key] = undefined;
promises.push(promise.then(function(res) {
results[key] = res;
}))
}
}
/**
* 将“yieldables”的数组转换为promise.
* 在内部使用`Promise.all().
*
* @param {Array} obj
* @return {Promise}
* @api private
*/
function arrayToPromise(obj) {
return Promise.all(obj.map(toPromise, this));
}
/**
* 将 thunk 转换为 promise.
*
* @param {Function}
* @return {Promise}
* @api private
*/
function thunkToPromise(fn) {
var ctx = this;
return new Promise(function(resolve, reject) {
fn.call(ctx, function(err, res) {
if (err) return reject(err);
if (arguments.length > 2) {
res = slice.call(arguments, 1)
}
resolve(res);
})
})
}
/**
* 将“yield”值转换为promise.
*
* @param {Mixed} obj
* @return {Promise}
* @api private
*/
function toPromise(obj) {
if (!obj) return obj;
if (isPromise(obj)) return obj;
if (isGeneratorFunction(obj) || isGenerator(obj)) {
return co.call(this, obj);
}
if ('function' === typeof obj) {
return thunkToPromise.call(this, obj);
}
if (Array.isArray(obj)) return arrayToPromise.call(this, obj);
if (isObject(obj)) return objectToPromise.call(this, obj);
return obj;
}
/**
* 执行生成器函数或生成器
* 并且返回一个 promise.
*
* @param {Function} fn
* @return {Promise}
* @api public
*/
function co(gen) {
var ctx = this;
var args = slice.call(arguments, 1);
// 我们将所有内容都包装在一个承诺中以避免承诺链接，
// 这会导致内存泄漏错误。
// see https://github.com/tj/co/issues/180
return new Promise(function(resolve, reject) {
if (typeof gen === 'function') {
gen = gen.apply(ctx, args);
}
if (!gen || typeof gen.next !== 'function') {
return resolve(gen);
}
onFullfilled();
})
/**
* @param {Mixed} res
* @return {Promise}
* @api private
*/
function onFullfilled(res) {
var ret;
try {
ret = gen, next(res);
} catch (e) {
return reject(e);
}
next(ret);
}
/**
* @param {Error} err
* @return {Promise}
* @api private
*/
function onRejected(err) {
var ret;
try {
ret = gen.throw(err);
} catch (e) {
return reject(e);
}
next(ret);
}
/**
* 获取生成器中的下一个值,
* 返回一个 promise.
*
* @param {Object} ret
* @return {Promise}
* @api private
*/
function next(ret) {
if(ret.done) return resolve(ret.value);
var value = toPromise.call(ctx, ret.value);
if(value && isPromise(value)) {
return value.then(onFullfilled.onRejected);
}
return onRejected(new TypeError('You may only yield a function, promise, generator, array, or object, '
+ 'but the following object was passed: "' + String(ret.value) + '"'
));
}
}
最新文章
- 怎样看懂Oracle的执行计划
- 判断远程图片是否存在【适用于windows服务器】
- 周赛-KIDx's Pagination 分类： 比赛 2015-08-02 08:23 7人阅读 评论(0) 收藏
- border做箭头的例子
- 代码混淆 GSON完满解决
- SolrCloud攻略
- ubuntu jdk环境变量配置
- 洛谷 P1598 垂直柱状图【字符串+模拟】
- alpha-咸鱼冲刺day4-紫仪
- JAVA 探究NIO
- h5微信页面在手机微信端和微信web开发者工具中都能正常显示，但是在pc端微信浏览器上打不开（显示空白）
- java中的值传递和引用传递有什么区别呀?
- docker inspect命令
- 什么是BOM?,什么是DOM? BOM跟DOM之间的关系
- python中的list按照某一列进行排序的方法
- 卷积神经网络之GoogleNet：inceptionV3模型学习
- 查找ipa包，删除接的ipa包
- Python 正斜杠/与反斜杠\
- error: Microsoft Visual C++ 9.0 is required. Get it from http://aka.ms/vcpython27
- 数字图像处理实验（总计23个）汇总 标签： 图像处理MATLAB 2017-05-31 10:30 175人阅读 评论(0)