如何让abort()自动获取用户调用栈，无需手动指定回溯层数？

你可以通过遍历调用栈、过滤掉第三方工具包的内部调用环境，自动找到用户定义的函数调用，无需手动指定回溯层数。以下是具体实现方案：

方法1：精准过滤包内调用的通用方案 #

library(purrr)
library(rlang)

# 自定义错误函数，自动匹配用户函数调用
error <- function(call = find_user_call()) {
  abort("Oops!", call = call)
}

# 核心函数：找到第一个非第三方包的调用
find_user_call <- function() {
  tb <- trace_back()
  # 遍历回溯栈，跳过工具包的内部调用
  for (frame in tb$frames) {
    env_pkg_name <- env_pkg(frame$env)
    # 保留用户自己的代码或base包的调用
    if (is.null(env_pkg_name) || env_pkg_name == "base") {
      return(frame$call)
    }
  }
  # 兜底返回上层调用
  return(caller_env())
}

# 测试用函数
foo <- function() {
  walk(1, \(i) {
    error()
  })
}

bar <- function() {
  error()
}

# 运行测试
foo()
# 输出：Error in `foo()`:
# ! Oops!

bar()
# 输出：Error in `bar()`:
# ! Oops!

方法2：简化版——跳过匿名函数调用 #

如果不需要复杂的包过滤，也可以直接跳过\(i)这类匿名函数，定位到最近的具名函数：

error <- function(call = find_user_call()) {
  abort("Oops!", call = call)
}

find_user_call <- function() {
  current_env <- caller_env()
  # 向上遍历环境，直到找到非匿名函数的调用环境
  while (!is.null(current_env) && is.function(current_env$fn) && is.null(names(current_env$fn))) {
    current_env <- env_parent(current_env)
  }
  return(current_env)
}

foo()
# 同样会输出指向foo()的错误信息

原理说明 #

• trace_back()能获取完整调用栈，每个栈帧包含调用环境和对应的表达式
• env_pkg()可以判断环境所属的包，借此过滤掉purrr、rlang等工具包的内部调用
• 遍历过程中找到的第一个用户代码或base包的调用，就是最适合展示给用户的错误位置

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