如何基于Socket数据中断Node.js中的长运行循环？

Node.js 是单线程事件驱动模型，你写的同步for循环会完全阻塞事件队列——在循环跑完之前，Node.js 根本没机会处理新的事件（包括你前端触发的Socket回调）。只有等循环彻底执行完毕，事件队列里的cancelAPIRequest才会被调用，这就是为啥没法中途终止循环。

要让Socket事件能及时被处理，必须把阻塞的同步循环拆成非阻塞的异步任务，每次迭代后给事件循环腾出手处理其他请求。

方法1：用setImmediate拆分循环 #

把每一次迭代逻辑放到setImmediate里，这样每次迭代结束后，Node.js会优先处理事件队列里的其他任务（比如Socket消息）：

let booleanFlag = false;

function cancelAPIRequest() {
    booleanFlag = true;
}

async function runAsyncLoop(body) {
    for (let i = 0; i < body.length; i++) {
        // 每次迭代先检查终止标记
        if (booleanFlag) {
            console.log('循环已终止');
            return;
        }
        // 这里写你原来的循环业务逻辑
        console.log(`处理第 ${i} 项数据`);
        // 让出事件循环，让Socket事件有机会执行
        await new Promise(resolve => setImmediate(resolve));
    }
}

方法2：用setTimeout实现异步迭代（可选） #

如果业务逻辑允许，也可以用setTimeout给每次迭代加个0延迟，同样能让出事件循环：

let booleanFlag = false;

function cancelAPIRequest() {
    booleanFlag = true;
}

function runTimedLoop(body) {
    let i = 0;
    function next() {
        if (booleanFlag) {
            console.log('循环已终止');
            return;
        }
        if (i >= body.length) return;
        
        // 执行当前迭代的业务逻辑
        console.log(`处理第 ${i} 项数据`);
        
        i++;
        setTimeout(next, 0); // 0延迟会把下一次迭代放到事件队列末尾
    }
    next();
}

方法3：用Worker线程处理CPU密集型循环 #

如果你的循环是CPU密集型操作（比如大量计算），上面的异步迭代可能还是会有性能问题，这时可以把循环放到单独的Worker线程里，主线程只负责处理Socket事件并发送终止信号：

主线程代码 #

const { Worker } = require('worker_threads');
let loopWorker;

// 启动循环
function startLoop(body) {
    loopWorker = new Worker('./loop-worker.js', { workerData: body });
}

// 触发终止
function cancelAPIRequest() {
    if (loopWorker) {
        loopWorker.postMessage('cancel');
        loopWorker.terminate();
    }
}

loop-worker.js 子线程代码 #

const { workerData, parentPort } = require('worker_threads');
const body = workerData;

// 监听主线程的终止信号
parentPort.on('message', (msg) => {
    if (msg === 'cancel') {
        console.log('循环已终止');
        process.exit();
    }
});

// 原来的同步循环
for (let i = 0; i < body.length; i++) {
    // 执行CPU密集型业务逻辑
    console.log(`处理第 ${i} 项数据`);
}

• 同步阻塞代码会直接卡死Node.js事件循环，所有异步事件（Socket、定时器、I/O）都得等阻塞代码跑完才能处理
• 异步迭代的关键是每次迭代后让出事件循环，给其他事件留执行机会
• CPU密集型任务优先用Worker线程，避免拖垮主线程

内容的提问来源于stack exchange，提问作者Akshay Dusane