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

ikun-fans

已于 2023-10-12 11:41:01 修改

阅读量499

点赞数

分类专栏： Rust 文章标签： rust 开发语言

于 2023-10-12 11:40:38 首次发布

本文链接：https://blog.csdn.net/u012456479/article/details/133787331

版权

Rust 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

文章讨论了在Rust中如何正确传递FnOnce类型的函数给需要Fntrait的回调，以及如何避免因std::thread::JoinHandle类型不符合Fn要求而引发的问题。作者提供了使用Arc<Mutex<...>>来共享JoinHandle的解决方案。

摘要由CSDN通过智能技术生成

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