「手写题」7、实现网络请求并发限制

298 阅读1分钟

实现网络请求并发限制的管理类 NetManager(concurrency),构造函数传入一个number,是并发的限制数,提供 request(url, options) => Promise方法。

  1. 实现一个类 class ,内部会有网络请求的并发限制,如果达到限制,则内部维持队列处理
  2. 调用方在使用 NetManagerrequest 方法时,不考虑是否达到了并发限制。也就是说,并发限制对调用方是透明 的,简单说,就是不管是否达到了并发限制, request 方法始终返回的是一个 Promise
function httpRequest(url, options){
  // 真实的网络请求省略
  return new Promise((resolve, reject) => {
    // xxx
  });
}

class NetManager {
  constructor(concurrency) {
    // 保存并发数
    this.concurrency = concurrency;
    // 待处理队列
    this.queue = [];
    // 处理中的个数
    this.processCount = 0;
  }
  request(url, data) {
    return new Promise((resolve, reject) => {
      // 这里不再直接发请求,而是放到队列里
      // 这里把这个 promise 的 resolve/reject 先和请求参数一起放到队列里,
      // 等待实际请求之后再改变promise状态,把最终数据(异常)返回给调用方
      this.queue.push({
        url,
        data,
        resolve,
        reject,
      });
      // 尝试处理第一个任务
      this._processNext();
    });
  }
  // 内部私有函数,判断当前是否超过并发数,未超过就从队列取出第1个发送请求
  _processNext() {
    if (this.processCount >= this.concurrency) {
      // 超过并发数,啥都不做
      return;
    }
    const obj = this.queue.shift();
    if (obj) {
      // 有待处理任务,取出来发送
      this.processCount++;
      const p = httpRequest(obj.url, obj.data);
      p.finally((res) => {
        // 网络请求完成,不管成功还是失败,都将进行中减1
        this.processCount--;
        // 尝试处理下一个任务
        this._processNext();
      });
      // 把实际的网络返回数据(失败错误)传给调用方
      p.then(obj.resolve, obj.reject);
    }
  }
}