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,而无需担心所有权问题。