如何使用Serialization包将Julia普通函数序列化到文件？

Julia的Serialization包对匿名函数（lambda）的序列化会保存其字节码与执行环境，因此能在新进程中直接反序列化使用。但对于普通命名函数，默认仅序列化函数的名称引用，反序列化时需要当前环境存在同名函数的定义，这就是出现UndefVarError的原因。以下是几种可行的解决方法：

方案一：序列化函数的定义表达式 #

直接将函数的定义表达式序列化，反序列化时通过eval重新定义函数：

序列化步骤 #

using Serialization, StatsBase

# 定义普通函数
f(x) = mean(x) + 1

# 提取函数的定义表达式
func_def = quote
    f(x) = mean(x) + 1
end

# 序列化表达式到磁盘
serialize("f_def.ser", func_def)

反序列化步骤 #

using Serialization, StatsBase

# 加载并求值表达式，重新定义函数
func_def = deserialize("f_def.ser")
eval(func_def)

# 验证函数可用性
f(5:7)  # 输出：7.0

注意：如果反序列化环境中已有同名函数，会覆盖原有定义。

方案二：将普通函数转为自包含的匿名函数序列化 #

通过提取函数的抽象语法树（AST），构建等价的匿名函数后序列化，避免依赖函数名称：

序列化步骤 #

using Serialization, StatsBase

# 定义普通函数
f(x) = mean(x) + 1

# 获取函数的方法和AST
m = first(methods(f))
func_ast = Base.uncompressed_ast(m)

# 构建等价的匿名函数表达式并求值
anon_f = eval(:(x -> $func_ast))

# 序列化匿名函数
serialize("anon_f.ser", anon_f)

反序列化步骤 #

using Serialization, StatsBase

# 直接加载匿名函数并使用
anon_f = deserialize("anon_f.ser")
anon_f(5:7)  # 输出：7.0

方案三：自定义函数的序列化/反序列化逻辑 #

通过扩展Serialization包的序列化方法，自动处理命名函数的定义保存与恢复：

定义自定义序列化逻辑 #

using Serialization

# 扩展序列化方法：保存函数的所有方法定义
function Serialization.serialize(s::AbstractSerializer, f::Function)
    method_exprs = []
    for m in methods(f)
        # 提取方法的参数名和AST
        arg_names = Base.argnames(m)[2:end]  # 跳过第一个隐含的self参数
        func_ast = Base.uncompressed_ast(m)
        # 构建函数定义表达式
        push!(method_exprs, :(function $(f.name)($(arg_names...))
            $func_ast
        end))
    end
    serialize(s, method_exprs)
end

# 扩展反序列化方法：重新定义函数
function Serialization.deserialize(s::AbstractSerializer, ::Type{Function})
    method_exprs = deserialize(s)
    for expr in method_exprs
        eval(expr)
    end
    # 返回最后一个定义的函数
    return eval(method_exprs[end].args[1].args[1])
end

使用自定义逻辑序列化 #

using StatsBase

f(x) = mean(x) + 1
serialize("f_custom.ser", f)

反序列化步骤 #

using Serialization, StatsBase

# 直接加载函数
f = deserialize("f_custom.ser")
f(5:7)  # 输出：7.0

注意：该方案对多方法函数、带关键字参数的函数需要进一步调整逻辑，但适用于大多数简单场景。

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