Julia测试中如何替换函数内加法运算符实现行为追踪？

核心思路：用局部作用域重载或参数化类型实现无侵入追踪 #

不用修改全局的Base.:+，而是把追踪逻辑限制在execute函数的局部环境内，让测试函数在这个隔离环境中执行——既不影响全局基础功能，也不用改动测试函数的代码。

方法一：局部作用域重载加法（最简单直接） #

先定义你的Tracking结构体：

mutable struct Tracking
    add_count::Int
    add_pairs::Vector{Tuple{Float64, Float64}}
    Tracking() = new(0, [])
end

然后编写execute函数，关键是用local +在函数内部重新定义加法行为，测试函数作为闭包会自动捕获这个局部的加法实现：

function execute(track::Tracking, test_func::Function)
    # 局部重载加法，仅在当前execute调用范围内生效
    local +
    function +(a::Float64, b::Float64)
        track.add_count += 1
        push!(track.add_pairs, (a, b))
        return Base.:+(a, b)  # 调用原生加法保证计算结果正确
    end
    # 执行用户传入的测试函数
    return test_func()
end

测试示例：

track = Tracking()
execute(track) do
    1.0 + 2.0
    3.0 + 4.0
end

println(track.add_count)  # 输出2，追踪成功
println(track.add_pairs)  # 输出[(1.0, 2.0), (3.0, 4.0)]

这个方案轻量简洁，完全隔离全局环境，测试函数不需要做任何修改。

方法二：参数化包装类型（适合扩展多操作追踪） #

如果后续需要追踪减法、乘法等更多操作，用包装类型的方式扩展性更强：

mutable struct Tracking
    add_count::Int
    add_pairs::Vector{Tuple{Float64, Float64}}
    Tracking() = new(0, [])
end

# 定义绑定追踪上下文的浮点类型
struct TrackedFloat{T<:AbstractFloat, C} <: AbstractFloat
    value::T
    context::C
end

# 重载TrackedFloat的加法逻辑
function Base.:+(a::TrackedFloat{Float64, Tracking}, b::TrackedFloat{Float64, Tracking})
    a.context.add_count += 1
    push!(a.context.add_pairs, (a.value, b.value))
    return TrackedFloat(a.value + b.value, a.context)
end

# 自动转换普通Float64为TrackedFloat，避免手动修改测试函数内的数值
Base.promote_rule(::Type{TrackedFloat{Float64, Tracking}}, ::Type{Float64}) = TrackedFloat{Float64, Tracking}
function Base.convert(::Type{TrackedFloat{Float64, Tracking}}, x::Float64, track::Tracking)
    return TrackedFloat(x, track)
end

# 编写execute函数，用宏自动替换测试函数中的Float64为TrackedFloat
function execute(track::Tracking, test_func::Function)
    # 宏：遍历表达式，将所有Float64常量替换为带追踪的版本
    macro auto_track(expr)
        return esc(
            MacroTools.prewalk(x -> x isa Float64 ? convert(TrackedFloat{Float64, Tracking}, x, track) : x, expr)
        )
    end
    # 执行经过宏转换的测试函数
    return @auto_track test_func()
end

这个方案需要导入MacroTools包（using MacroTools），但后续扩展其他运算符时，只需要给TrackedFloat重载对应方法即可，同样不需要修改测试函数。

为什么你之前的实现没生效？ #

你之前重载的是带Tracking参数的加法（比如+(a::Float64, b::Float64, track::Tracking)），但测试函数中调用的是普通的a + b，Julia的调度系统不会自动为你补充Tracking参数。而上面两种方案，要么直接替换了局部的加法逻辑，要么把数值包装成带上下文的类型，让调度系统能正确匹配到追踪版本。

注意事项 #

• 方法一适合仅需追踪加法的场景，代码量少、无额外依赖。
• 方法二适合需要扩展多操作追踪的场景，灵活性更强。
• 两种方案都不会影响全局的Base.:+，完全隔离追踪逻辑。

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