如何使用Promise实现并发请求数限制

没有Promise的并发请求

 在Web开发中，我们经常需要发起多个异步请求来获取数据。例如，我们可能需要从服务器获取一些用户信息、文章内容、评论列表等等。如果我们使用的是传统的JavaScript回调函数，可能会写出类似下面这样的代码： 

function getUsers(callback) {
  fetch('https://example.com/users', (response) => {
    if (response.status === 200) {
      response.json((data) => {
        callback(data);
      });
    } else {
      console.error('Error fetching users');
    }
  });
}

function getPosts(callback) {
  fetch('https://example.com/posts', (response) => {
    if (response.status === 200) {
      response.json((data) => {
        callback(data);
      });
    } else {
      console.error('Error fetching posts');
    }
  });
}

function getComments(callback) {
  fetch('https://example.com/comments', (response) => {
    if (response.status === 200) {
      response.json((data) => {
        callback(data);
      });
    } else {
      console.error('Error fetching comments');
    }
  });
}

getUsers((users) => {
  console.log(users);
});

getPosts((posts) => {
  console.log(posts);
});

getComments((comments) => {
  console.log(comments);
});

在这个例子中，我们定义了三个函数来获取不同的数据，每个函数都接受一个回调函数作为参数。当响应成功返回时，我们会调用回调函数并将响应数据传递给它。这种方式看起来很简单，但如果我们需要发起更多的请求，代码会变得越来越难以维护。

另外一个问题是，如果我们需要在多个请求都完成后进行处理，我们需要使用一些技巧来确保所有请求都已经完成。例如，我们可以使用计数器来跟踪已完成的请求数，并在所有请求都完成后执行回调函数。这种方式也很容易出错，特别是当请求失败时。

使用Promise限制并发请求

幸运的是，JavaScript中有一个非常有用的工具，可以帮助我们更好地处理异步请求：Promise。使用Promise，我们可以将异步请求看作是一个返回Promise对象的函数，然后使用链式调用来处理请求的结果。

例如，我们可以将上面的代码改写成使用Promise的形式：

function getUsers() {
  return fetch('https://example.com/users').then((response) => {
    if (response.status === 200) {
      return response.json();
    } else {
      throw new Error('Error fetching users');
    }
  });
}

function getPosts() {
  return fetch('https://example.com/posts').then((response) => {
    if (response.status === 200) {
      return response.json();
    } else {
      throw new Error('Error fetching posts');
    }
  });
}

function getComments() {
  return fetch('https://example.com/comments').then((response) => {
    if (response.status === 200) {
      return response.json();
    } else {
      throw new Error('Error fetching comments');
    }
  });
}

Promise.all([getUsers(), getPosts(), getComments()]).then(([users, posts, comments]) => {
  console.log(users);
  console.log(posts);
  console.log(comments);
}).catch((error) => {
  console.error(error);
});

在这个例子中，我们将每个获取数据的函数改写成返回Promise对象的形式。当响应成功返回时，我们会使用then方法来处理响应数据。如果响应失败，我们会使用throw语句抛出一个错误。

然后，我们使用Promise.all函数来等待所有请求完成，并将所有响应数据存储在一个数组中。当所有请求都完成后，我们可以使用解构赋值来获取每个响应数据，并进行处理。如果有任何请求失败，我们可以使用catch方法来处理错误。

这种方式比使用回调函数更加简洁和易于维护，但它仍然有一个问题：如果我们同时发起太多的请求，可能会导致服务器过载或者网络拥塞。因此，我们需要限制并发请求数，以确保我们的应用程序可以正常运行。

使用Promise实现并发请求数限制

在JavaScript中，我们可以使用Promise.all函数来等待所有请求完成。但是，Promise.all函数会等待所有Promise对象都已解析后才返回结果。如果我们同时发起太多的请求，可能会导致浏览器崩溃或者内存溢出。

幸运的是，我们可以使用一些技巧来限制并发请求数。例如，我们可以使用一个计数器来跟踪当前正在进行的请求数，然后在所有请求都完成后执行回调函数。这种方式可以确保我们不会同时发起太多的请求，从而避免服务器过载或者网络拥塞。

以下是一个使用Promise实现并发请求数限制的示例代码：

在此示例中，我们定义了一个limitRequest函数，用于限制并发请求数，并将所有请求的响应结果存储在results数组中。我们使用executeRequest函数来执行请求，并跟踪当前正在进行的请求。我们使用while循环来启动最初的一批请求，并在请求完成后继续执行后续请求，直到所有请求都完成。最后，我们使用Promise.all函数来等待所有请求完成，并返回results数组。
 

function limitRequest(urls, limit) {
  if (!Array.isArray(urls) || urls.length === 0 || limit <= 0) {
    // 对输入参数进行有效性检查
    return Promise.reject(new Error("Invalid input"));
  }

  let index = 0;
  const results = [];
  let completed = 0; // 计数器，记录已经完成的请求数量

  const executeRequest = (url, resolve, reject, retryCount = 0) => {
    // 添加重试机制
    const controller = new AbortController(); // 创建AbortController对象
    const timeoutId = setTimeout(() => {
      // 超时处理
      controller.abort(); // 取消请求
      reject(new Error("Request timeout"));
    }, 5000); // 设置超时时间为5秒

    fetch(url, { signal: controller.signal }) // 使用AbortController对象发起请求
      .then((response) => response.json())
      .then((result) => {
        clearTimeout(timeoutId); // 请求成功，清除超时定时器
        results.push(result);
        completed++;
        if (index < urls.length) {
          // 如果还有未完成的请求，继续执行下一个请求
          executeRequest(urls[index++], resolve, reject);
        } else if (completed === urls.length) {
          // 所有请求已经完成，返回结果
          resolve(results);
        }
      })
      .catch((error) => {
        clearTimeout(timeoutId); // 请求失败，清除超时定时器
        if (retryCount < 3) {
          // 最多重试3次
          executeRequest(url, resolve, reject, retryCount + 1);
        } else {
          completed++;
          console.error(error);
          if (completed === urls.length) {
            // 所有请求已经完成，返回结果
            resolve(results);
          }
        }
      });
  };

  return new Promise((resolve, reject) => {
    while (index < limit && index < urls.length) {
      // 启动最初的一批请求
      executeRequest(urls[index++], resolve, reject);
    }
  });
}

const urls = [
  "https://jsonplaceholder.typicode.com/posts/1",
  "https://jsonplaceholder.typicode.com/posts/2",
  "https://jsonplaceholder.typicode.com/posts/3",
  "https://jsonplaceholder.typicode.com/posts/4",
  "https://jsonplaceholder.typicode.com/posts/5",
  "https://jsonplaceholder.typicode.com/posts/6",
  "https://jsonplaceholder.typicode.com/posts/7",
  "https://jsonplaceholder.typicode.com/posts/8",
  "https://jsonplaceholder.typicode.com/posts/9",
  "https://jsonplaceholder.typicode.com/posts/10",
  "https://jsonplaceholder.typicode.com/posts/11",
  "https://jsonplaceholder.typicode.com/posts/12",
];

limitRequest(urls, 3)
  .then((results) => {
    console.log(results);
    console.log(`共收到 ${results.length} 条响应`);
  })
  .catch((error) => {
    console.error(error);
  });

 

总结

在本文中，我们介绍了如何使用Promise来处理异步请求，并使用Promise.all函数来等待所有请求完成。我们还讨论了如何限制并发请求数，以确保我们的应用程序可以正常运行。使用Promise，我们可以写出更加简洁、可维护的代码，并且可以更好地处理异步请求。