Rust中如何根据条件传递异步函数作为参数？

问题重现 #

当尝试根据条件选择不同异步函数并传递给另一个函数时，编译器抛出类型不兼容错误：

if and else have incompatible types
expected fn item fn() -> impl Future<Output = ()> {say_hello}
found fn item fn() -> impl Future<Output = ()> {say_hi}

对应的代码如下：

use std::future::Future;

async fn greet<Fut>(f: impl Fn() -> Fut) where Fut: Future<Output = ()> {
    f().await;
}

async fn say_hello() {
    println!("Hello");
}

async fn say_hi() {
    println!("Hi");
}

#[tokio::main]
async fn main() {
    let x = 5;

    let greet_func = if x < 10 {
        say_hello
    } else {
        say_hi
    };

    greet(&greet_func).await;
}

问题原因 #

Rust中每个async fn返回的impl Future是唯一的匿名类型——哪怕两个异步函数的返回值输出类型相同，它们的impl Future底层类型也完全不同。因此say_hello和say_hi是两种不同的函数类型，无法被赋值给同一个变量，导致if-else分支类型不匹配。

解决方案 #

通过**装箱（Box）**将不同的Future统一为动态类型，让if-else分支返回相同的类型即可解决问题。以下是两种可行的修改方式：

方式1：用闭包返回装箱后的Future #

修改main函数中的变量定义，将条件判断移到闭包内部，返回Pin<Box<dyn Future<Output = ()>>>类型：

#[tokio::main]
async fn main() {
    let x = 5;

    let greet_func = || {
        if x < 10 {
            Box::pin(say_hello())
        } else {
            Box::pin(say_hi())
        }
    };

    greet(greet_func).await;
}

此时greet函数无需修改，因为闭包返回的Pin<Box<dyn Future<Output = ()>>>满足Fut: Future<Output = ()>的约束。

方式2：显式指定函数指针类型 #

直接将greet_func指定为返回装箱Future的函数指针类型：

#[tokio::main]
async fn main() {
    let x = 5;

    let greet_func: fn() -> Pin<Box<dyn Future<Output = ()>>> = if x < 10 {
        || Box::pin(say_hello())
    } else {
        || Box::pin(say_hi())
    };

    greet(greet_func).await;
}

原理说明 #

Box::pin将不同的匿名Future类型包装成Pin<Box<dyn Future<Output = ()>>>，这是一个统一的动态类型，消除了两个异步函数返回值的类型差异，让编译器能够识别if-else分支返回的是相同类型的值。

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