如何在用户中断R/RStudio计算时触发on.exit执行指定语句？

默认情况下，当用户按下ESC键或点击RStudio的停止按钮中断计算时，R会直接终止当前执行流程，导致on.exit中的代码无法运行。要实现中断时先执行指定清理逻辑（比如关闭并行集群）再终止，可以通过捕获interrupt条件来实现，这种方式兼容Mac（包括M系列芯片）和Windows系统。

核心思路 #

用户触发的中断会在R中抛出interrupt类型的条件，我们可以用withCallingHandlers函数捕获这个条件，先执行清理操作，再重新抛出中断让程序正常终止。同时保留on.exit确保正常执行完毕时也能清理资源。

针对并行集群的示例代码 #

run_parallel_task <- function() {
  # 创建PSOCK集群
  cl <- parallel::makePSOCKcluster(2)
  
  # 定义通用清理函数：关闭集群
  cleanup_cluster <- function() {
    if (!is.null(cl) && parallel::isCluster(cl)) {
      parallel::stopCluster(cl)
      cat("并行集群已成功关闭\n")
    }
  }
  
  # 捕获中断信号，执行清理后终止
  withCallingHandlers({
    # 这里编写你的并行计算逻辑
    task_result <- parallel::parLapply(cl, 1:10, function(x) {
      Sys.sleep(1)  # 模拟耗时计算
      x * 2
    })
    print(task_result)
  }, interrupt = function(e) {
    # 中断发生时优先执行清理
    cleanup_cluster()
    # 重新抛出中断，让程序正常终止
    stop(e)
  })
  
  # 正常执行完成时也触发清理
  on.exit(cleanup_cluster())
}

代码说明 #

1. cleanup_cluster函数：先检查集群是否存在且有效，避免重复关闭或操作无效集群导致报错。
2. withCallingHandlers捕获中断：当用户触发中断时，会优先执行interrupt处理函数中的清理逻辑，再通过stop(e)重新抛出中断，保证程序会终止。
3. on.exit兜底：确保函数正常执行完毕时，也能自动关闭集群，避免资源泄漏。

兼容性验证 #

这个方案在Mac（M系列芯片）和Windows系统下都能正常工作，因为R对interrupt条件做了跨平台封装，无需依赖系统级的信号处理差异。

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