2024-06-08 某大厂面试官:100 个请求同时发送,咋办

2024-06-07  本文已影响0人  半眼鱼

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); // 输出每个请求的结果
});

https://mp.weixin.qq.com/s/zFTByheg7WfnTHUkyavT4g

上一篇 下一篇

猜你喜欢

热点阅读