Rust中FnOnce如何传递给一个约束Fn的回调

shuai132_2023-10-15 2:31

Rust中FnOnce如何传递给一个约束Fn的回调

下面的代码，set_cb(func);会报错，如何包装能够做到这样的效果：

fn set_cb<F: Fn() + 'static>(handler: F)
{
    handler();
}

fn main() {
    let join_handle = std::thread::spawn(|| {});
    let func = || {
        join_handle.join().unwrap();
    };
    set_cb(func);
    // func();
}

解决方法

代码中的问题是你试图在闭包中捕获 join_handle，但 std::thread::JoinHandle 不满足 Fn trait，因为它不是 Clone 和 Copy。这就是为什么你不能直接捕获它并在闭包中使用。解决方法是使用 Arc<Mutex<...>> 来共享 JoinHandle。

这是一个修改后的例子：

use std::sync::{Arc, Mutex};
use std::thread;

fn set_cb<F: Fn() + Send + 'static>(handler: F)
{
    handler();
}

fn main() {
    let join_handle = Arc::new(Mutex::new(Some(thread::spawn(|| {}))));
    let join_handle_clone = Arc::clone(&join_handle);

    let func = move || {
        let mut join_handle = join_handle_clone.lock().unwrap();
        join_handle.take().unwrap().join().unwrap();
    };

    set_cb(func);
}

在这个修改后的例子中，我们创建了一个 Arc<Mutex<Option>>。我们创建一个 Option，这样我们就可以在闭包中使用 take() 方法将其设置为 None，这样就可以将其移动出 Mutex，而无需在闭包中保持锁定。这样，我们就可以在 set_cb 调用中使用 join_handle，而无需担心所有权问题。