Rust有没有信号量机制,在缓存有数据的时候才允许等待的进程取数据?

Rust提供了信号量(Semaphore)机制,尽管它并没有直接提供类似于某些其他编程语言中的Semaphore类。然而,你可以使用std::sync::Mutex和std::sync::Condvar(条件变量)组合来实现信号量的功能。

信号量通常用于控制对共享资源的访问,通过维护一个计数器来限制同时访问资源的线程数。当资源可用时(即缓存中有数据时),等待的线程可以被唤醒并获取资源。

在Rust中,你可以使用Mutex来保护共享资源并提供互斥访问,同时使用Condvar来让等待的线程在条件满足时被唤醒。下面是一个简单的示例,展示了如何使用Mutex和Condvar来实现一个信号量,该信号量允许在缓存有数据时唤醒等待的线程:

rust 复制代码
use std::sync::{Mutex, Condvar, Arc};  
use std::thread;  
use std::time::Duration;  
  
// 缓存结构  
struct Cache {  
    data: Mutex<Vec<i32>>,  // 互斥锁保护的数据  
    not_empty: Condvar,     // 条件变量,用于等待数据可用  
    not_full: Condvar,      // 条件变量,用于等待空间可用(如果需要的话)  
}  
  
impl Cache {  
    fn new(capacity: usize) -> Cache {  
        Cache {  
            data: Mutex::new(Vec::with_capacity(capacity)),  
            not_empty: Condvar::new(),  
            not_full: Condvar::new(),  
        }  
    }  
  
    // 尝试从缓存中取出数据  
    fn try_pop(&self) -> Option<i32> {  
        let mut data = self.data.lock().unwrap();  
        if data.is_empty() {  
            None  
        } else {  
            Some(data.remove(0))  
        }  
    }  
  
    // 向缓存中添加数据  
    fn push(&self, value: i32) {  
        let mut data = self.data.lock().unwrap();  
        data.push(value);  
        self.not_empty.notify_one(); // 通知可能等待的消费者线程  
    }  
  
    // 从缓存中获取数据,如果没有数据则阻塞  
    fn pop(&self) -> i32 {  
        let mut data = self.data.lock().unwrap();  
        while data.is_empty() {  
            // 等待直到有数据可用或超时  
            self.not_empty.wait(data.as_mut()).unwrap();  
        }  
        data.remove(0)  
    }  
}  
  
fn main() {  
    let cache = Arc::new(Cache::new(10)); // 假设缓存容量为10  
  
    // 生产者线程  
    let producer = thread::spawn(move || {  
        for i in 1..=5 {  
            thread::sleep(Duration::from_secs(1));  
            cache.push(i);  
            println!("Produced: {}", i);  
        }  
    });  
  
    // 消费者线程  
    let consumer = thread::spawn(move || {  
        for _ in 1..=5 {  
            let value = cache.pop();  
            println!("Consumed: {}", value);  
        }  
    });  
  
    // 等待生产者和消费者线程完成  
    producer.join().unwrap();  
    consumer.join().unwrap();  
}

在这个示例中,Cache结构体有一个互斥锁data来保护对Vec的访问,以及两个条件变量not_empty和not_full(尽管在这个例子中我们并没有使用not_full,因为我们没有实现缓存满时的等待逻辑)。

pop方法尝试从缓存中取出数据。如果缓存为空,它会调用not_empty.wait(data.as_mut())来阻塞当前线程,同时释放互斥锁,允许其他线程运行。当生产者线程调用push方法并向缓存中添加数据时,它会调用not_empty.notify_one()来唤醒可能正在等待的消费者线程。

这样,我们就实现了一个简单的信号量机制,它允许消费者线程在缓存有数据时获取数据,并在没有数据时等待。

相关推荐
爱吃牛肉的大老虎13 分钟前
Rust对象之结构体,枚举,特性
开发语言·后端·rust
TunerT_TQ3 小时前
Valhalla 静态工程审阅 #008|RisingWave 源码证据驱动评测【大厂开源基础设施特辑】
rust·开源·github·实时数据处理·流式数据库·apacheflink·streamingsql
踏着七彩祥云的小丑3 小时前
忘记Redis是否安装过时查看
数据库·redis·缓存
TunerT_TQ7 小时前
Valhalla 静态工程审阅 #007|AgentENV 源码证据驱动评测【大厂开源基础设施特辑】
rust·开源·go·github·sandbox·分布式系统·ai基础设施
麻瓜code8 小时前
【Redis 】数据类型、持久化与过期删除
数据库·redis·缓存
NutShell Wang9 小时前
Vite 8.1 深度拆解:Rolldown 统一打包器如何终结前端构建的「双引擎时代」
前端·rust·开源·vite·开发者工具·vibe coding
寒水馨9 小时前
macOS下载、安装uv-0.12.0(附安装包uv-aarch64-apple-darwin.tar.gz)
python·macos·rust·项目管理·包管理器·astral·pip替代
changshuaihua00110 小时前
Redis 详解:Redis 与 MySQL 的区别、数据类型与缓存三大问题
数据库·redis·缓存
寒水馨10 小时前
macOS下载、安装ripgrep-15.2.0(附安装包ripgrep-15.2.0-aarch64-apple-darwin.tar.gz)
macos·正则表达式·rust·ripgrep·rg·命令行搜索·代码搜索
love530love10 小时前
Windows 原生编译 SGLang(4/8):环境关——VS 版本、venv 顺序、CUDA 多版本、生成器缓存
windows·缓存·sglang