Julia中如何阻止关键代码块被Ctrl+C/SIGINT立即中断？

你可以通过自定义SIGINT信号处理器，实现和Python方案逻辑完全一致的不可中断代码块：关键代码运行期间收到的Ctrl+C信号会被暂存，等代码块执行完成后再触发中断，保证块内操作的原子性，避免中间状态被打断导致数据无效。

核心实现代码 #

直接复制即可使用，不需要安装第三方依赖：

function uninterruptible(f)
    got_interrupt = Ref(false)
    # 临时替换SIGINT处理器，仅标记中断状态不立即抛出
    prev_handler = Sys.set_signal_handler(SIGINT, _ -> got_interrupt[] = true)
    try
        return f()
    finally
        # 恢复原有信号处理器
        Sys.set_signal_handler(SIGINT, prev_handler)
        # 若运行期间收到中断，块执行完成后再抛出中断异常
        got_interrupt[] && throw(InterruptException())
    end
end

MCMC场景的正确用法 #

不要把整个采样循环包裹在不可中断块中，仅包裹单次迭代里更新链状态的原子操作段即可，既保证状态一致性，也不会让Ctrl+C失去及时终止的作用：

for iter in 1:max_iter
    # 不可中断的核心状态更新段
    uninterruptible() do
        # 此处放所有和链状态更新相关的原子操作：
        # 提议新参数、计算接受概率、更新链状态、计数接受/拒绝次数
        propose!(chain)
        log_alpha = calc_log_acceptance(chain)
        if rand() < exp(log_alpha)
            accept!(chain)
        else
            reject!(chain)
        end
    end

    # 块外为可中断区域，可放置非核心操作
    if iter % checkpoint_interval == 0
        save_checkpoint(chain, "mcmc_tmp.jld2")
    end
    if iter % log_interval == 0
        @printf "迭代%d，接受率%.3f\n" iter chain.accept_cnt/iter
    end
end

配套优化建议 #

• 不可中断块要尽可能短，仅保留必须保证原子性的状态修改逻辑，不要把IO、进度打印、长耗时非状态操作放入块内，避免按下Ctrl+C后长时间无响应。
• 可以在最外层捕获InterruptException，触发中断时自动保存当前链状态，下次运行直接从断点续跑，不需要重新走burn-in流程，示例代码如下：

try
    run_mcmc(chain, n_iter=100000)
catch e
    e isa InterruptException || rethrow(e)
    println("\n检测到用户中断，正在保存断点...")
    save_checkpoint(chain, "mcmc_interrupted.jld2")
    println("断点已保存，重启后可从当前迭代位置继续采样")
end

原理说明 #

Julia默认收到SIGINT时，会在执行流的任意安全点直接抛出InterruptException，抛出位置可能刚好落在状态更新的中间步骤（比如刚写入部分参数、还没同步更新计数），最终导致链状态无效。通过临时替换信号处理器的方式，把中断的抛出时机严格控制在关键块执行完成后的安全点，就能保证链状态始终完整有效。该方案在REPL、脚本执行环境下均兼容，不需要额外配置启动参数。

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