如何在 Julia 中传递类型安全的函数

Question

假设我想将一个函数传递给另一个函数：

function foo()
    return 0;
end
function bar(func)
    return func();
end
print(bar(foo));

但是你可以使函数类型安全：

function func(t::Int)
    print(t);
end
func(0);                 #produces no error
func("Hello world");     #produces an error

我没有发现，我如何将两者结合起来，也就是说，如何将bar 的参数显式定义为函数，例如func，可能具有某些输入/输出参数类型。

提前感谢您的帮助。

Answer 1

函数的类型为Function。您可以轻松检查：

julia> foo() = 1;

julia> T = typeof(foo)
typeof(foo)

julia> supertype(T)
Function

julia> foo isa Function
true

这不一定涵盖所有可调用类型，因为您可以将任何类型设为可调用：

julia> struct Callable end

julia> (::Callable)(x::Number) = x + one(x)

julia> callable = Callable()
Callable()

julia> callable(5)
6

julia> callable isa Function
false

Answer 2

如果我对您的理解正确，您想确保传递的函数返回特定类型吗？最简单的方法是在运行时对返回值进行类型断言：

julia> function f(func)
           val = func()::Int # Error if the return value is not of type Int
           return val
       end
f (generic function with 1 method)

julia> f(() -> 1)
1

julia> f(() -> 1.0)
ERROR: TypeError: in typeassert, expected Int64, got Float64
Stacktrace:
 [1] f(::var"#7#8") at ./REPL[5]:2
 [2] top-level scope at REPL[8]:1

您也可以使用FunctionWrappers.jl 包（它将转换为指定的返回类型，如果无法转换则错误）：

julia> using FunctionWrappers: FunctionWrapper

julia> function f(func::FunctionWrapper{Int,<:Tuple})
           val = func()
           return val
       end;

julia> function f(func)
           fw = FunctionWrapper{Int,Tuple{}}(func)
           return f(fw)
       end;

julia> f(() -> 1)
1

julia> f(() -> 1.0) # Can convert to Int
1

julia> f(() -> 1.2) # Can not convert to Int
ERROR: InexactError: Int64(1.2)