众所周知,浏览器发起的请求最大并发数量有限制,这是因为浏览器会限制同一域名下的并发请求数量,以避免对服务器造成过大的压力。
模拟一下大量请求的场景
1 2 3 4 5 6 7 8
| const ids = new Array(100).fill('')
console.time() for (let i = 0; i < ids.length; i++) { console.log(i) } console.timeEnd()
|
一次发送上百次请求,虽然浏览器有队列机制(上图pending),但是我们还是需要细颗粒度的控制并发请求。
解决思路:请求池
利用Promise模拟任务队列,从而实现请求池效果
step 1:定义请求池主函数
1 2 3 4
| export const handQueue = ( reqs ) => {}
|
接受一个参数reqs
,它是一个数组,包含需要发送的请求。函数的主要目的是对这些请求进行队列管理,确保并发请求的数量不会超过设定的上限。
step 2:定义 dequeue函数
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
| const dequeue = () => { while (current < concurrency && queue.length) { current++; const requestPromiseFactory = queue.shift() requestPromiseFactory() .then(() => { }) .catch(error => { console.log(error) }) .finally(() => { current-- dequeue() }); } }
|
这个函数用于从请求池中取出请求并发送。它在一个循环中运行,直到当前并发请求数current
达到最大并发数concurrency
或请求池queue
为空。对于每个出队的请求,它首先增加current
的值,然后调用请求函数requestPromiseFactory
来发送请求。当请求完成(无论成功还是失败)后,它会减少current
的值并再次调用dequeue
,以便处理下一个请求。
step 3: 定义返回请求入队函数
1 2 3 4
| return (requestPromiseFactory) => { queue.push(requestPromiseFactory) dequeue() }
|
函数返回一个函数,这个函数接受一个参数requestPromiseFactory
,表示一个返回Promise的请求工厂函数。这个返回的函数将请求工厂函数加入请求池queue
,并调用dequeue
来尝试发送新的请求,当然也可以自定义axios,利用Promise.all
统一处理返回后的结果。
完整代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
| import axios from 'axios'
export const handQueue = ( reqs ) => { reqs = reqs || []
const requestQueue = (concurrency) => { concurrency = concurrency || 6 const queue = [] let current = 0
const dequeue = () => { while (current < concurrency && queue.length) { current++; const requestPromiseFactory = queue.shift() requestPromiseFactory() .then(() => { }) .catch(error => { console.log(error) }) .finally(() => { current-- dequeue() }); }
}
return (requestPromiseFactory) => { queue.push(requestPromiseFactory) dequeue() }
}
const enqueue = requestQueue(6)
for (let i = 0; i < reqs.length; i++) {
enqueue(() => axios.get('/api/test' + i)) } }
|