如何在Rust中将具体函数类型作为泛型参数使用？

核心思路 #

Rust无法直接写出函数项的匿名类型，但可以手动定义一个零大小结构体（ZST），让它实现Fn/FnMut/FnOnce trait，以此替代原始函数。这样既能得到可明确写出的类型，又能保持零内存开销的特性，让编译器充分发挥泛型优化能力。

手动实现方案 #

struct AddOneBefore<T: Fn(usize) -> usize>(T);

// 定义对应subtract_one的ZST结构体
struct SubtractOne;

// 实现Fn系列trait，复用原函数逻辑
impl FnOnce<(usize,)> for SubtractOne {
    type Output = usize;
    extern "rust-call" fn call_once(self, args: (usize,)) -> Self::Output {
        args.0 - 1
    }
}

impl FnMut<(usize,)> for SubtractOne {
    extern "rust-call" fn call_mut(&mut self, args: (usize,)) -> Self::Output {
        args.0 - 1
    }
}

impl Fn<(usize,)> for SubtractOne {
    extern "rust-call" fn call(&self, args: (usize,)) -> Self::Output {
        args.0 - 1
    }
}

// 现在可以明确写出目标类型
type Identity = AddOneBefore<SubtractOne>;
const IDENTITY: Identity = AddOneBefore(SubtractOne);

简化实现：用宏减少重复代码 #

如果需要定义多个这类ZST函数，可通过宏自动生成Fn系列trait的实现，避免重复代码：

struct AddOneBefore<T: Fn(usize) -> usize>(T);

macro_rules! zst_fn {
    ($name:ident, $($arg:ident: $ty:ty),* -> $ret:ty, $body:expr) => {
        struct $name;

        impl FnOnce<($($ty,)*)> for $name {
            type Output = $ret;
            extern "rust-call" fn call_once(self, args: ($($ty,)*)) -> Self::Output {
                let ($($arg,)*) = args;
                $body
            }
        }

        impl FnMut<($($ty,)*)> for $name {
            extern "rust-call" fn call_mut(&mut self, args: ($($ty,)*)) -> Self::Output {
                let ($($arg,)*) = args;
                $body
            }
        }

        impl Fn<($($ty,)*)> for $name {
            extern "rust-call" fn call(&self, args: ($($ty,)*)) -> Self::Output {
                let ($($arg,)*) = args;
                $body
            }
        }
    };
}

// 一键生成ZST函数
zst_fn!(SubtractOne, n: usize -> usize, n - 1);

// 目标类型清晰可写
type Identity = AddOneBefore<SubtractOne>;
const IDENTITY: Identity = AddOneBefore(SubtractOne);

方案优势 #

• 零内存开销：SubtractOne是ZST，AddOneBefore<SubtractOne>同样是ZST，不会像函数指针那样占用额外内存。
• 编译器优化友好：类型明确且无间接指针，编译器可完全内联函数逻辑，消除函数调用开销。
• 类型可明确写出：直接使用AddOneBefore<SubtractOne>即可作为关联类型，满足迭代器嵌套场景的组合需求。
• 兼容标准库迭代器：实现了标准Fn trait，可无缝配合std::iter::Map等类型，享受标准库提供的size_hint、DoubleEndedIterator等优化实现。

为什么原写法不可行 #

Rust中每个函数都有独有的匿名函数项类型，无法通过{subtract_one}这种语法直接写出。而函数指针类型fn(usize) -> usize是所有同签名函数的统一类型，会引入间接调用开销，不符合零优化的需求。

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