rust : condvar中一对一和多对一模式初探

condvar是不经常碰到的,但其实在tokio之类库中,还是非常核心的作用。

想进一步体会condvar的使用,还是从场景出发。

一、一个通知发送者,一个接收者

假定一个员工收到一个任务,就是模拟是一个时间片,到时了,就会触发通知发出(notify_one)。

通知也有不断发出通知,还是事件触发后再发出通知。这些根据场景需要都可以。

不管形式如何,但mutex中data值的不同状态,wait判断事件是否已经发生的标志。

复制代码
use std::sync::{ Arc, Condvar, Mutex };
use std::thread;
use std::time::{Duration,Instant};
use std::collections::HashMap;
fn main_single() {
    let builder = thread::Builder::new();
    let pair = Arc::new((Mutex::new(false), Condvar::new())); // data为bool值,以此判断事件是否发出
    let pair2 = Arc::clone(&pair);
    //单个通知者
    let handle = builder.spawn(move || {
        let start = Instant::now();
        let mut num_send = 0;
        let mut is_ok =false;
        while !is_ok {
            thread::sleep(Duration::from_secs(1));
            let (lock, cvar) = &*pair2;
            let mut started = lock.lock().unwrap();
            num_send +=1;
            println!("员工:第 {:?} 次报告;任务执行中.....,已耗时: {:?} 秒",num_send, start.elapsed().as_secs());
            cvar.notify_one();
            let is_event_happened = start.elapsed().as_secs() >6;
            if is_event_happened{
                println!("员工: 报告boss,任务已经完成了!共耗时: {:?} 秒",start.elapsed().as_secs());
                *started = true;
                //cvar.notify_all();
                is_ok = true;
            }
        }
    });
    let mut is_received = false;
    let seconds = 1;
    let start = Instant::now();
    let mut num_receive = 0;
    while !is_received {
        let (lock, cvar) = &*pair;
        let mut started = lock.lock().unwrap();
        
        started = cvar.wait(started).unwrap();
        if *started == false{
            println!("boss: 时间过去 {:?} 秒,任务还没完成!",start.elapsed().as_secs());
        }else {
            println!("boss: 好!收到员工任务完成信息!现在过去了 {:?} 秒!",start.elapsed().as_secs());
            is_received = true;
        }
        num_receive +=1;
        println!("boss: 收到! 收到{:?} 次报告,共过去了: {:?} 秒",num_receive, start.elapsed().as_secs());
        thread::sleep(Duration::from_secs(seconds));
    }
    if is_received{
        println!("任务完成!");
    }else{
        println!("任务失败!");
    }
    handle.expect("任务失败!boss要崩溃了!").join().unwrap();
    
}
fn main(){
	main_single()
}

这种场景比较简单,是一对一的。但其它典型的场景可能还有多对一,不断发送通知给一个接收者。象N个员工和一个上级。

二、多个通知者,一个接收者

通过Mutex中包裹一个Hashmap结构,来对多个通知的状态进行管理。

下例构造了3个员工汇报对应一个老板接收的方式。

mutex中值的结构是hashmap<usize,(nums,bool)>结构,分别指的是员工id编号usize,以及此员工通知发送次数nums,以及事件是否发生标志状态bool。

这个数据结构根据自己需要来设定,这里只是一个假设了一个相对信息量复杂的场景。

复制代码
use std::sync::{ Arc, Condvar, Mutex };
use std::thread;
use std::time::{Duration,Instant};
use std::collections::HashMap;
fn main_multi(){

    let pair = Arc::new((Mutex::new(HashMap::new()), Condvar::new()));
    let mut handles  = Vec::new();
    const PERSONS :usize = 3;//多个成员向1个
    let each_send_seconds = 1;
    for i in 0..PERSONS {
        let _pair = Arc::clone(&pair);
        let handle = thread::spawn(move || {
            let start = Instant::now();
            let mut num_send = 0;
            let mut is_ok = false;
            while !is_ok {
                thread::sleep(Duration::from_secs(each_send_seconds));
                let (lock, cvar) = &*_pair;
                let mut locked = lock.lock().unwrap();
                num_send +=1;
                let is_event_happened = start.elapsed().as_secs() >= 6;//定义为事件
                if is_event_happened{
                    println!("员工: 我是员工{:?} ,报告boss,我任务执行完成!共耗时: {:?} 秒",i,start.elapsed().as_secs());
                    (*locked).insert(i,(num_send,true));
                    is_ok = true;
                }else{
                    println!("员工:我是员工{:?} ,第 {:?} 次报告,任务执行中.....,已耗时:: {:?} 秒",i,num_send, start.elapsed().as_secs());
                    (*locked).insert(i,(num_send,false));
                }
                cvar.notify_one();
            }
        });
        handles.push(handle);
    }

    let mut is_all_received = false;
    //let each_recv_seconds = 1;
    let start = Instant::now();
    //let mut num_receive = 0;
    while !is_all_received {
        let mut is_status = vec![false;PERSONS];
        //每次轮询全部一次        
        let (lock, cvar) = &*pair;
        let mut locked = lock.lock().unwrap();
        locked = cvar.wait(locked).unwrap();
        for i in 0..PERSONS{
            if (*locked).contains_key(&i){
                let num = (*locked).get(&i).unwrap();
                let status = num.1;
                if status{
                    println!("员工:{:?}, 通知次数: {:?} status : {:?}",i,num.0,status);
                    is_status[i] = true;
                }
            }
        }
        if is_status.iter().all(|&i| i == true){
            println!("boss: 很好,收到全部员工任务完成的信息!现在过去了 {:?} 秒!",start.elapsed().as_secs());
            is_all_received = true;
        }

    }
    if is_all_received {
        println!("任务完成!");
    }else{
        println!("任务失败!");
    }
    println!("任务总共花了: {:?} 秒", start.elapsed().as_secs());
    handles.into_iter().for_each(|w| w.join().unwrap());
}

fn main(){
    //main_single();
    main_multi();
}

运行结果:

复制代码
员工:我是员工2 ,第 1 次报告,任务执行中.....,已耗时:: 1 秒
员工:我是员工1 ,第 1 次报告,任务执行中.....,已耗时:: 1 秒
员工:我是员工0 ,第 1 次报告,任务执行中.....,已耗时:: 1 秒
员工:我是员工0 ,第 2 次报告,任务执行中.....,已耗时:: 2 秒
员工:我是员工1 ,第 2 次报告,任务执行中.....,已耗时:: 2 秒
员工:我是员工2 ,第 2 次报告,任务执行中.....,已耗时:: 2 秒
员工:我是员工1 ,第 3 次报告,任务执行中.....,已耗时:: 3 秒
员工:我是员工0 ,第 3 次报告,任务执行中.....,已耗时:: 3 秒
员工:我是员工2 ,第 3 次报告,任务执行中.....,已耗时:: 3 秒
员工:我是员工0 ,第 4 次报告,任务执行中.....,已耗时:: 4 秒
员工:我是员工1 ,第 4 次报告,任务执行中.....,已耗时:: 4 秒
员工:我是员工2 ,第 4 次报告,任务执行中.....,已耗时:: 4 秒
员工:我是员工2 ,第 5 次报告,任务执行中.....,已耗时:: 5 秒
员工:我是员工0 ,第 5 次报告,任务执行中.....,已耗时:: 5 秒
员工:我是员工1 ,第 5 次报告,任务执行中.....,已耗时:: 5 秒
员工: 我是员工2 ,报告boss,我任务执行完成!共耗时: 6 秒
员工:2, 通知次数: 6 status : true
员工: 我是员工1 ,报告boss,我任务执行完成!共耗时: 6 秒
员工: 我是员工0 ,报告boss,我任务执行完成!共耗时: 6 秒
员工:0, 通知次数: 6 status : true
员工:1, 通知次数: 6 status : true
员工:2, 通知次数: 6 status : true
boss: 很好,收到全部员工任务完成的信息!现在过去了 6 秒!
任务完成!
任务总共花了: 6 秒

三、多对多模式

在多对一的基础上,还可以衍生中多对多的模式。这个可以扩展一下。也可用到noftify_all。

四、其它相关问题

要注意notify_one()到wait()这个过程,需要特别小心,否则可能会想象中不一样。这个里面有一些操作系统线程 调度机制比较底层,是比较复杂的。

相关推荐
e***74952 小时前
Modbus报文详解
服务器·开发语言·php
lly2024062 小时前
ASP 发送电子邮件详解
开发语言
小徐敲java2 小时前
python使用s7协议与plc进行数据通讯(HslCommunication模拟)
开发语言·python
likuolei2 小时前
XSL-FO 软件
java·开发语言·前端·数据库
6***37942 小时前
PHP在电商中的BigCommerce
开发语言·php
Dev7z2 小时前
基于Matlab的多制式条形码识别与图形界面(GUI)系统设计与实现
开发语言·matlab
合作小小程序员小小店2 小时前
桌面开发,在线%信息管理%系统,基于vs2022,c#,winform,sql server数据。
开发语言·数据库·sql·microsoft·c#
FL16238631292 小时前
ONNX RuntimeC++ 静态库下载安装和使用教程
开发语言·c++
星释2 小时前
Rust 练习册 95:React与响应式编程
开发语言·react.js·rust
Evand J2 小时前
【MATLAB例程】3D雷达-IMU融合定位系统(基于扩展卡尔曼滤波)|雷达观测距离、俯仰角、方向角,IMU包括6维(加速度与角速度)。附下载链接
开发语言·matlab·跟踪·雷达观测·三维定位·ekf滤波