一起阅读 p-limit 源码，学习限制并发数的技巧

2024-01-27 15:09:16

前言

在 Node.js 中，并发编程是一个非常重要的概念。通过并发编程，我们可以充分利用多核 CPU 的优势，提高程序的性能。但是，并发编程也存在一个问题，那就是如果并发数过大，可能会导致系统资源枯竭，从而使程序崩溃。

为了解决这个问题，我们可以使用 p-limit 库来限制并发数。p-limit 是一个非常流行的 Node.js 库，它可以帮助我们轻松地限制并发数。在本文中，我们将一起阅读 p-limit 的源码，并学习如何使用它来限制并发数。

阅读源码

p-limit 的源码非常简单，只有不到 100 行代码。我们可以轻松地阅读它，并理解它的实现原理。

首先，我们来看看 p-limit 的构造函数。构造函数接受一个参数，即并发数限制。这个参数指定了 p-limit 允许同时运行的任务数。

function PLimit(limit) {
  if (!(this instanceof PLimit)) {
    return new PLimit(limit);
  }

  this._limit = limit;
  this._queue = [];
  this._active = 0;
}

接下来，我们来看看 p-limit 的 run 方法。run 方法接受一个参数，即任务函数。任务函数是一个异步函数，它会在 p-limit 允许的情况下运行。

P.prototype.run = function(fn) {
  var self = this;

  if (self._active >= self._limit) {
    return new Promise(function(resolve, reject) {
      self._queue.push({ fn: fn, resolve: resolve, reject: reject });
    });
  }

  self._active++;

  return new Promise(function(resolve, reject) {
    fn(function(err, data) {
      self._active--;

      if (err) {
        reject(err);
      } else {
        resolve(data);
      }

      if (self._queue.length) {
        var task = self._queue.shift();
        self.run(task.fn).then(task.resolve, task.reject);
      }
    });
  });
};

run 方法首先检查 p-limit 的 active 属性是否大于等于并发数限制。如果大于等于并发数限制，则将任务函数放入队列中等待执行。否则，则立即执行任务函数。

任务函数执行完成后，run 方法会将 p-limit 的 active 属性减 1，并检查队列中是否还有等待执行的任务函数。如果有，则继续执行队列中的任务函数。

使用 p-limit

现在，我们已经了解了 p-limit 的实现原理。接下来，我们来看看如何使用 p-limit。

var pLimit = require('p-limit');

var limit = new pLimit(10);

var tasks = [
  function() {
    return new Promise(function(resolve, reject) {
      setTimeout(function() {
        resolve('Task 1');
      }, 1000);
    });
  },
  function() {
    return new Promise(function(resolve, reject) {
      setTimeout(function() {
        resolve('Task 2');
      }, 2000);
    });
  },
  function() {
    return new Promise(function(resolve, reject) {
      setTimeout(function() {
        resolve('Task 3');
      }, 3000);
    });
  }
];

Promise.all(tasks.map(function(task) {
  return limit.run(task);
})).then(function(results) {
  console.log(results);
});