使用 Promise 实现一个异步流量控制的函数(限制并发数)
下面是使用 Promise 实现异步流量控制的函数的示例:
function asyncFlowControl(tasks, limit) {
let runningCount = 0; // 当前正在运行的任务数
let index = 0; // 当前执行的任务索引
const results = []; // 存储任务的结果
return new Promise((resolve, reject) => {
function runTask() {
if (runningCount >= limit || index >= tasks.length) {
// 达到并发限制或所有任务已执行完毕,返回结果
if (results.length === tasks.length) {
resolve(results);
}
return;
}
const task = tasks[index];
const currentIndex = index; // 保存当前任务索引
index++;
runningCount++;
task().then((result) => {
results[currentIndex] = result; // 存储任务结果
runningCount--;
runTask(); // 递归执行下一个任务
}).catch((error) => {
reject(error);
});
runTask(); // 递归执行下一个任务
}
runTask(); // 开始执行任务
});
}
// 示例用法
function asyncTask(value) {
return new Promise((resolve) => {
setTimeout(() => {
console.log(value);
resolve(value);
}, Math.random() * 1000);
});
}
const tasks = [
() => asyncTask(1),
() => asyncTask(2),
() => asyncTask(3),
() => asyncTask(4),
() => asyncTask(5),
];
asyncFlowControl(tasks, 2).then((results) => {
console.log('All tasks completed:', results);
}).catch((error) => {
console.error('Error occurred:', error);
});
以上示例中的 asyncFlowControl
函数接受一个任务数组 tasks
和一个并发限制 limit
,它会按照并发限制逐个执行任务,并返回一个 Promise 对象。在示例中,任务数组中的每个任务都是一个返回 Promise 的函数,通过 setTimeout
模拟异步操作。
在执行过程中, asyncFlowControl
函数会维护一个 runningCount
变量来跟踪当前正在运行的任务数,并使用递归的方式执行任务。当达到并发限制或所有任务都已执行完毕时,函数会返回结果。
通过控制并发任务的数量,我们可以限制同时执行的异步操作,实现异步流量控制。在上述示例中,设置并发限制为 2,可以确保最多同时执行 2 个任务,并在任务执行完毕后再执行下一个任务。
延伸阅读:
暂无内容!
评论列表 (0条):
加载更多评论 Loading...