Julia中如何后台收集深层函数的统计数据？

以下几个Julia包可以满足你无需大量修改调用链路、后台收集数据并支持后续检索的需求：

1. LoggingExtras.jl（基于官方Logging扩展） #

官方Logging.jl本身支持自定义日志接收器，LoggingExtras.jl进一步简化了这个过程，你可以轻松将日志数据存储到内存结构（如Dict、Vector）中，而非立即输出。

示例代码： #

using Logging, LoggingExtras

# 初始化内存日志存储
log_store = Dict{Symbol, Vector{Any}}()

# 自定义日志处理函数，将数据存入log_store
function store_log(event)
    func_key = event.metadata[:func]
    if !haskey(log_store, func_key)
        log_store[func_key] = []
    end
    push!(log_store[func_key], (timestamp=event.time, msg=event.message))
end

# 替换全局日志器，仅将日志存入内存不输出到控制台
global_logger(TransformerLogger(ConsoleLogger()) do event
    store_log(event)
    return false  # 阻止默认控制台输出
end)

# 深层函数f，仅需添加日志语句
function f(obj)
    # 记录对象的类型、大小等统计信息
    @logmsg :info "Object stats: type=$(typeof(obj)), size=$(sizeof(obj))" func=:f
end

# 原调用链路无需修改
function b(obj)
    f(obj)
end

function a(obj)
    b(obj)
end

# 测试调用
a([1, 2, 3])
a("Julia is awesome")

# 后续检索f的日志数据
for entry in log_store[:f]
    println("[$(entry.timestamp)] $(entry.msg)")
end

2. Telemetry.jl（专注遥测数据收集） #

这个包专为后台收集运行时遥测数据设计，提供了轻量级的埋点接口，无需修改调用链路中的其他函数，仅需在目标函数f中添加数据收集语句即可。

示例代码： #

using Telemetry

# 初始化遥测系统
init_telemetry()

# 在目标函数f中添加数据收集
function f(obj)
    # 记录对象的统计信息，支持自定义键值对
    track_event(:function_f, obj_type=typeof(obj), obj_size=sizeof(obj))
end

# 原调用链路保持不变
function b(obj)
    f(obj)
end

function a(obj)
    b(obj)
end

# 测试调用
a(42)
a(Dict(:x => 1, :y => 2))

# 检索所有来自f的事件数据
f_events = get_events(:function_f)
for evt in f_events
    println("Object type: $(evt[:obj_type]), Size: $(evt[:obj_size]) bytes")
end

3. 轻量方案：基于全局存储的封装（无需额外包） #

如果你不想引入第三方包，可以基于你最初的思路，用线程安全结构封装全局存储，避免手动管理全局变量的线程安全问题：

using DataStructures

# 线程安全的有序字典存储日志
const logs = OrderedDict{Symbol, Vector{Any}}()

function f(obj)
    lock(logs) do
        if !haskey(logs, :f)
            logs[:f] = []
        end
        push!(logs[:f], (obj_type=typeof(obj), obj_size=sizeof(obj)))
    end
end

# 调用链路无需修改
function b(obj)
    f(obj)
end

function a(obj)
    b(obj)
end

# 测试与检索
a([1.0, 2.0])
a("test string")

println(logs[:f])

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