03:维护线程池
我们可以维护一个运行池和一个等待队列,运行池中始终保持10个请求并发。
当运行池中一个请求完成后,就从等待队列中取出一个新请求放入运行池中运行,保证运行池始终满负荷运转,即使出现慢接口,也不会阻塞后续接口入池。
// 运行池,用于存储当前正在执行的请求
const pool = new Set();
// 等待队列,用于存储等待执行的请求
const waitQueue = [];
/**
* @description: 限制并发请求的数量
* @param {*} reqFn: 请求方法(返回一个 Promise 的函数)
* @param {*} max: 最大并发数
* @returns {Promise} 返回一个 Promise,当请求完成时 resolve 或 reject
*/
const request = (reqFn, max) => {
return new Promise((resolve, reject) => {
// 检查运行池是否已满
const isFull = pool.size >= max;
// 包装新的请求方法
const newReqFn = () => {
reqFn()
.then(res => {
resolve(res); // 请求成功时 resolve
})
.catch(err => {
reject(err); // 请求失败时 reject
})
.finally(() => {
// 请求完成后,将其从运行池中移除
pool.delete(newReqFn);
// 从等待队列中取出新的请求并放入运行池中执行
const next = waitQueue.shift();
if (next) {
pool.add(next);
next();
}
});
};
if (isFull) {
// 如果运行池已满,将新的请求放入等待队列
waitQueue.push(newReqFn);
} else {
// 如果运行池未满,将新的请求加入运行池并执行
pool.add(newReqFn);
newReqFn();
}
});
};
// 遍历 requestList,并发执行每个请求,限制最大并发数为 10
requestList.forEach(async item => {
const res = await request(item, 10); // 调用 request 函数,并发执行请求
console.log(res); // 输出每个请求的结果
});