如何获取R语言中函数内警告的行号及所属函数名

可以通过withCallingHandlers捕获警告事件，结合调用栈工具和源码位置函数，同时提取警告所属的函数名和行号，具体实现代码如下：

# 示例函数
fun <- function(x, y){
  z = sum(x,y)
  if(z > 15)
    warning("Potential problem")
  
  return(z)
}

# 捕获警告并提取关键信息
warning_details <- NULL
withCallingHandlers(
  fun(10, 20),
  warning = function(w) {
    # 从调用栈中提取触发警告的函数名
    call_stack <- sys.calls()
    target_func <- as.character(call_stack[[length(call_stack)-1]][[1]])
    
    # 定位函数环境，获取警告所在行号
    func_env <- sys.frames()[[which(sapply(sys.frames(), function(f) environmentName(f)) == target_func)]]
    warning_line <- getSrcLocation(func_env$fun, "line")
    
    # 保存结果
    warning_details <<- data.frame(function_name = target_func, warning_index = warning_line)
    invokeRestart("muffleWarning") # 屏蔽默认警告输出
  }
)

# 格式化输出结果
knitr::kable(warning_details, align = "c")

运行上述代码后，会输出如下结果：

补充说明 #

• 该方法不仅支持自定义函数的警告，对R内置函数的警告也能捕获函数名，但内置函数因无本地源码文件关联，无法获取行号。
• invokeRestart("muffleWarning")用于阻止警告信息默认输出，若需要保留原警告输出可删除该行。

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