rust使用lazy_static对全局变量多线程并发读写示例

首先需要在项目依赖Cargo.toml添加lazy_static依赖项

[dependencies]
lazy_static = "1.4.0"

示例代码如下:

use lazy_static::lazy_static;
use std::sync::{RwLock, RwLockReadGuard, RwLockWriteGuard};
use std::thread;

#[derive(Debug)]
struct SharedData {
    version: i32,
    data: Vec<u8>,
}

impl SharedData {
    fn new() -> Self {
        SharedData {
            version: 0,
            data: Vec::new(),
        }
    }

    fn display(&self) {
        println!("{:?}", self);
    }

    fn add_version(&mut self) {
        self.version = self.version + 1;
    }

    fn add_item(&mut self, item: u8) {
        self.data.push(item);
        self.add_version();
    }

    fn remove_item(&mut self, item: u8) {
        self.data.retain(|&x| x != 42);
        self.add_version();
    }
}

lazy_static! {
    static ref GLOBAL_DATA: RwLock<SharedData> = RwLock::new(SharedData::new());
}

fn get_write_shared_data() -> RwLockWriteGuard<'static, SharedData> {
    GLOBAL_DATA.write().unwrap()
}

fn get_read_shared_data() -> RwLockReadGuard<'static, SharedData> {
    GLOBAL_DATA.read().unwrap()
}

fn main() {
    // 读取最初的数据
    thread::spawn(|| {
        let shared_data = get_read_shared_data();
        shared_data.display();
    })
    .join()
    .unwrap();

    // 添加一个元素
    thread::spawn(|| {
        let mut shared_data = get_write_shared_data();
        shared_data.add_item(42);
    })
    .join()
    .unwrap();

    // 读取修改后的数据
    thread::spawn(|| {
        let shared_data = get_read_shared_data();
        shared_data.display();
    })
    .join()
    .unwrap();

    // 移除一个元素
    thread::spawn(|| {
        let mut shared_data = get_write_shared_data();
        shared_data.remove_item(42);
    })
    .join()
    .unwrap();

    // 读取修改后的数据
    thread::spawn(|| {
        let shared_data = get_read_shared_data();
        shared_data.display();
    })
    .join()
    .unwrap();
}
posted @ 2024-01-25 11:41  ALXPS  阅读(65)  评论(0编辑  收藏  举报