熟知p-limit的运作方式，才能玩转它控制并发数的艺术

2024-01-14 19:01:16

p-limit的工作原理

p-limit使用令牌桶算法来限制并发数。令牌桶算法是一种流行的并发控制算法，它使用一个固定大小的令牌桶来管理并发操作的数量。当一个异步操作需要执行时，它必须先从令牌桶中获取一个令牌。如果没有令牌可用，则该操作必须等待，直到令牌桶中出现空闲令牌。

p-limit的令牌桶算法由两个参数控制：

并发限制： 并发限制指定了令牌桶的大小。它决定了同时可以执行的异步操作的数量。
令牌生成速率： 令牌生成速率指定了令牌桶中令牌生成的速率。它决定了令牌桶中令牌可用的速度。

如何使用p-limit来控制并发数

使用p-limit来控制并发数非常简单。您只需要创建一个p-limit实例，然后将您的异步操作传递给它的run()方法。p-limit将确保这些操作不会同时执行超过并发限制的数量。

以下是一个示例，演示如何使用p-limit来控制并发数：

const pLimit = require('p-limit');
const limit = pLimit(2);

const tasks = [
  () => {
    return new Promise((resolve, reject) => {
      setTimeout(() => {
        console.log('Task 1');
        resolve();
      }, 1000);
    });
  },
  () => {
    return new Promise((resolve, reject) => {
      setTimeout(() => {
        console.log('Task 2');
        resolve();
      }, 2000);
    });
  },
  () => {
    return new Promise((resolve, reject) => {
      setTimeout(() => {
        console.log('Task 3');
        resolve();
      }, 3000);
    });
  },
];

tasks.forEach((task) => {
  limit(task).then(() => {
    console.log('Task completed');
  });
});