Rust 并行库 crossbeam 的 Channel 示例

下面是一个完整的示例程序:

rust 复制代码
use crossbeam::channel;
use crossbeam::thread;
use std::thread::sleep;
use std::time::Duration;

// 定义Task结构体
struct Task {
    data: usize, // 假设任务包含一个数据字段
    call_box: Box<dyn FnMut()>, // 假设任务包含一个可调用对象的装箱指针
}

impl Task {
    fn new(data: usize, call_box: impl FnMut() + 'static) -> Self {
        Task {
            data,
            call_box: Box::new(call_box),
        }
    }

    // 实现call_box方法
    fn call_box(&mut self) {
        (self.call_box)();
    }
}

fn main() {
    const NUMBER_OF_WORKERS: usize = 4; // 假设有4个工作线程
    let (tx, rx) = channel::unbounded::<Task>();
    let mut handlers = vec![];

    // 启动工作线程
    for _ in 0..NUMBER_OF_WORKERS {
        let rx = rx.clone();
        let handle = thread::spawn(move || {
            while let Some(task) = rx.recv() {
                task.call_box(); // 执行任务
            }
        });
        handlers.push(handle);
    }

    // 发送任务到通道
    for i in 0..10 { // 假设发送10个任务
        let task = Task::new(i, || {
            println!("Executing task with data: {}", i);
            sleep(Duration::from_secs(1)); // 模拟耗时操作
            println!("Finished task with data: {}", i);
        });
        tx.send(task).unwrap();
    }

    // 关闭发送通道
    drop(tx);

    // 等待所有工作线程完成
    for handle in handlers {
        handle.join().unwrap();
    }

    println!("All tasks are processed.");
}

在这个程序中,我们定义了一个Task结构体,它包含一个data字段和一个call_box字段,后者是一个装箱的可调用对象。我们实现了call_box方法,它调用这个装箱的可调用对象。

main函数中,我们创建了一个无界通道,用于在工作线程和主线程之间传递Task实例。我们启动了NUMBER_OF_WORKERS个工作线程,它们不断地从通道接收Task实例并调用call_box方法执行它们。

然后,主线程创建了一些Task实例,并通过通道发送它们给工作线程。一旦所有任务都被发送,主线程通过drop(tx)关闭了发送通道,这样工作线程在尝试接收任务时,如果没有更多任务可用,将会得到一个None,从而退出循环。

最后,主线程等待所有工作线程完成,并打印出消息表示所有任务都已经处理完毕。

请注意,为了简化示例,我使用了Box<dyn FnMut()>来允许Task存储任何可调用对象的装箱指针。这意味着任务中的可调用对象必须能够单独编译成一个独立的、无状态的函数,这样才能安全地在多个线程之间共享。在实际应用中,你可能需要根据你的具体需求调整Task结构体的设计和使用方式。

相关推荐
沐知全栈开发6 分钟前
MongoDB 创建数据库
开发语言
ErizJ25 分钟前
Golang | 迭代器模式
开发语言·golang·迭代器模式
牙痛不能吃糖,哭28 分钟前
C++面试复习日记(8)2025.4.25,malloc,free和new,delete的区别
开发语言·c++
健康的猪31 分钟前
golang的cgo的一点小心得
开发语言·后端·golang
夜夜敲码1 小时前
C语言教程(十六): C 语言字符串详解
c语言·开发语言
宋康1 小时前
C语言结构体和union内存对齐
c语言·开发语言
居然是阿宋1 小时前
Kotlin高阶函数 vs Lambda表达式:关键区别与协作关系
android·开发语言·kotlin
机构师1 小时前
<rust><iced><GUI>iced中的复合列表部件:combo_box
后端·rust
Cao1234567893212 小时前
简易学生成绩管理系统(C语言)
c语言·开发语言
The Future is mine2 小时前
C# new Bitmap(32043, 32043, PixelFormat.Format32bppArgb)报错:参数无效,如何将图像分块化处理?
开发语言·c#