妙趣横生的并发控制：设计一个优雅的批量请求函数

2024-01-28 19:15:04

巧用批量请求函数优化并发处理，提升应用程序性能

引言

在现代网络应用开发中，并发控制扮演着至关重要的角色，它能够有效提升应用程序的性能和用户体验。当需要同时处理多个请求时，我们需要一种优雅高效的并发控制机制，以确保请求能够有序执行，避免资源竞争和死锁。本文将介绍一种巧妙的批量请求函数设计，它能够在有限的并发量内同时处理多个请求，并在请求完成后返回一个 Promise，将结果按照请求顺序依次打出。

函数设计思路

批量请求函数的基本思路是利用 JavaScript 的 Promise 机制来控制请求的并发量。当函数被调用时，它会创建一个 Promise 对象，并将请求队列存储在该对象中。然后，函数会根据设定的并发量，依次从请求队列中取出请求并发送。每当有一个请求返回时，函数就会留下一个空位，并从队列中取出下一个请求发送。这样，函数就可以在有限的并发量内同时处理多个请求，而不会造成资源竞争或死锁。

函数实现细节

以下是一个批量请求函数的示例实现：

function batchRequest(urls, maxConcurrency) {
  // 创建一个 Promise 对象
  const promise = new Promise((resolve, reject) => {
    // 初始化请求队列和结果数组
    const queue = [...urls];
    const results = [];

    // 循环发送请求，直到队列为空
    (function sendRequest() {
      // 检查是否达到最大并发量
      if (queue.length === 0 || results.length >= maxConcurrency) {
        return;
      }

      // 从队列中取出一个请求
      const url = queue.shift();

      // 发送请求并处理结果
      fetch(url)
        .then((response) => {
          // 将结果添加到结果数组中
          results.push(response.json());

          // 检查是否还有请求需要发送
          sendRequest();
        })
        .catch((error) => {
          // 处理错误
          reject(error);
        });
    })();

    // 当所有请求都完成后，解析 Promise
    Promise.all(results).then((data) => {
      resolve(data);
    });
  });

  // 返回 Promise 对象
  return promise;
}

函数使用示例

以下是一个使用批量请求函数的示例：

const urls = [
  'https://example.com/api/users/1',
  'https://example.com/api/users/2',
  'https://example.com/api/users/3',
  'https://example.com/api/users/4',
  'https://example.com/api/users/5',
];

// 设置最大并发量为 3
const maxConcurrency = 3;

// 调用批量请求函数
batchRequest(urls, maxConcurrency)
  .then((data) => {
    // 处理结果
    console.log(data);
  })
  .catch((error) => {
    // 处理错误
    console.error(error);
  });