@ghini/kit/dist/lib/queue.js

js practical tools to assist efficient development

export { queue };
function queue(num = 3) {
    if (typeof num !== "number" || num < 1) {
        throw new TypeError("并发数(num)必须是一个大于等于1的数字。");
    }
    const taskQueue = [];
    const availableWorkerIds = Array.from({ length: num }, (_, i) => i);
    async function next() {
        if (availableWorkerIds.length === 0 || taskQueue.length === 0)
            return;
        const workerId = availableWorkerIds.shift();
        const { task, resolve, reject } = taskQueue.shift();
        try {
            resolve(await task(workerId));
        }
        catch (error) {
            reject(error);
        }
        finally {
            availableWorkerIds.push(workerId);
            next();
        }
    }
    return (task) => {
        return new Promise((resolve, reject) => {
            taskQueue.push({ task, resolve, reject });
            next();
        });
    };
}