Remove WaitMap from existence.

This commit is contained in:
4lDO2
2025-04-10 15:32:29 +02:00
parent f3af1be69f
commit 10ef631fc3
2 changed files with 1 additions and 74 deletions
+1 -2
View File
@@ -1,5 +1,4 @@
pub use self::{wait_condition::WaitCondition, wait_map::WaitMap, wait_queue::WaitQueue};
pub use self::{wait_condition::WaitCondition, wait_queue::WaitQueue};
pub mod wait_condition;
pub mod wait_map;
pub mod wait_queue;
-72
View File
@@ -1,72 +0,0 @@
use alloc::collections::BTreeMap;
use core::mem;
use spin::Mutex;
use crate::sync::WaitCondition;
#[derive(Debug)]
pub struct WaitMap<K, V> {
// Using BTreeMap as this depends on .keys() providing elements in sorted order.
pub inner: Mutex<BTreeMap<K, V>>,
pub condition: WaitCondition,
}
impl<K, V> WaitMap<K, V>
where
K: Clone + Ord,
{
pub fn new() -> WaitMap<K, V> {
WaitMap {
inner: Mutex::new(BTreeMap::new()),
condition: WaitCondition::new(),
}
}
pub fn receive_nonblock(&self, key: &K) -> Option<V> {
self.inner.lock().remove(key)
}
pub fn receive(&self, key: &K, reason: &'static str) -> Option<V> {
loop {
let mut inner = self.inner.lock();
if let Some(value) = inner.remove(key) {
return Some(value);
}
if !self.condition.wait(inner, reason) {
return None;
}
}
}
pub fn receive_any_nonblock(&self) -> Option<(K, V)> {
let mut inner = self.inner.lock();
if let Some(key) = inner.keys().next().cloned() {
inner.remove(&key).map(|value| (key, value))
} else {
None
}
}
pub fn receive_any(&self, reason: &'static str) -> (K, V) {
loop {
let mut inner = self.inner.lock();
if let Some(key) = inner.keys().next().cloned() {
if let Some(entry) = inner.remove(&key).map(|value| (key, value)) {
return entry;
}
}
let _ = self.condition.wait(inner, reason);
}
}
pub fn receive_all(&self) -> BTreeMap<K, V> {
let mut ret = BTreeMap::new();
mem::swap(&mut ret, &mut *self.inner.lock());
ret
}
pub fn send(&self, key: K, value: V) {
self.inner.lock().insert(key, value);
self.condition.notify();
}
}