手写 p-limit ：40 行代码达成并发控制

在当今的软件开发中，并发控制是一个至关重要的话题。有效的并发控制可以确保系统的稳定性、性能和正确性。今天，我们将探讨如何通过手写仅 40 行代码来实现强大的并发控制，即 p-limit。

让我们理解一下为什么并发控制如此重要。在多线程或异步操作的环境中，如果不加以控制，多个任务可能会同时竞争资源，导致数据不一致、死锁或者性能下降等问题。

p-limit 的核心思想是限制同时运行的异步任务的数量。这可以避免系统因过多并发任务而不堪重负。

以下是一个简单的 JavaScript 实现示例：

class PLimit {
  constructor(concurrency) {
    this.concurrency = concurrency;
    this.running = 0;
    this.queue = [];
  }

  async run(task) {
    if (this.running >= this.concurrency) {
      await new Promise((resolve) => this.queue.push(resolve));
    }
    this.running++;
    try {
      await task();
    } finally {
      this.running--;
      if (this.queue.length > 0) {
        this.queue.shift()();
      }
    }
  }
}

// 使用示例
const limit = new PLimit(2); // 设置并发数量为 2

async function task1() {
  console.log('Task 1 started');
  await delay(1000);
  console.log('Task 1 completed');
}

async function task2() {
  console.log('Task 2 started');
  await delay(500);
  console.log('Task 2 completed');
}

async function delay(time) {
  return new Promise((resolve) => setTimeout(resolve, time));
}

async function main() {
  limit.run(task1);
  limit.run(task2);
}

main();

在上述代码中，PLimit 类接受一个并发数量参数。run 方法用于执行任务，当并发数量达到上限时，新任务会进入队列等待。

通过这种简单而有效的方式，我们可以在各种场景中轻松实现并发控制，确保系统的稳定性和性能。无论是处理大量的网络请求、文件操作还是其他异步任务，p-limit 都能为我们提供可靠的保障。

手写 p-limit 虽然代码量不大，但却能为我们的开发工作带来巨大的价值，帮助我们更好地应对并发带来的挑战。