Rust函数能否设置可选泛型类型参数 可否通过宏实现相关功能

基础结论 #

• Rust原生支持可选泛型类型参数，只要给泛型声明设置默认值即可，比如fn bar<F: Foo = ()>()，调用时如果不指定泛型就会默认使用()类型
• 你示例中的if F != ()运行时类型判断原生语法不支持：Rust的类型信息会在编译期完全擦除，运行时没有对应的元数据可以做这种判断，但可以通过编译期的特性实现等价逻辑，不需要运行时判断

推荐实现：空实现特化（稳定可用，无额外开销） #

最符合Rust惯例的方案是给unit类型()实现Foo trait的空逻辑，完全不需要写判断分支，编译期会自动选择对应实现：

fn main() {
    // 不指定泛型，默认用()，不会执行实际逻辑
    bar();
    // 指定自定义类型，执行对应逻辑
    bar::<MyFoo>();
}

// 泛型默认值设为()，实现可选参数
fn bar<F: Foo = ()>() {
    let x = some_computation();
    let foo = F::new(x);
    foo.foo();
}

trait Foo {
    fn new(x: u64) -> Self;
    fn foo(&self);
}

// 给()实现空的Foo逻辑，无任何开销
impl Foo for () {
    fn new(_x: u64) -> Self { () }
    fn foo(&self) {}
}

// 自定义类型的正常实现
struct MyFoo;
impl Foo for MyFoo {
    fn new(x: u64) -> Self {
        println!("初始化MyFoo，值为{x}");
        MyFoo
    }
    fn foo(&self) {
        println!("执行MyFoo的foo方法");
    }
}

fn some_computation() -> u64 { 123 }

这种方案完全运行在稳定版Rust上，无用的空实现会被编译器完全优化掉，没有任何运行时开销。

可选方案：宏实现 #

如果你不想给()实现Foo trait，也可以通过声明宏实现不同调用的分支匹配：

macro_rules! bar {
    // 无参数调用，不执行Foo逻辑
    () => {{
        let x = some_computation();
        // 其他公共逻辑
    }};
    // 传入类型参数，执行对应Foo逻辑
    ($f:ty) => {{
        let x = some_computation();
        let foo = <$f as Foo>::new(x);
        foo.foo();
    }};
}

// 调用方式
bar!();
bar!(MyFoo);

宏会在编译期展开成对应代码，同样没有运行时开销。

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