28 lines
857 B
Rust
28 lines
857 B
Rust
use anyhow::anyhow;
|
|
use std::sync::{Mutex, MutexGuard, RwLock, RwLockReadGuard, RwLockWriteGuard};
|
|
|
|
pub trait MutexExt<T> {
|
|
fn lock_anyhow(&self) -> anyhow::Result<MutexGuard<'_, T>>;
|
|
}
|
|
|
|
impl<T> MutexExt<T> for Mutex<T> {
|
|
fn lock_anyhow(&self) -> anyhow::Result<MutexGuard<'_, T>> {
|
|
self.lock().map_err(|e| anyhow!("mutex poisoned: {e}"))
|
|
}
|
|
}
|
|
|
|
pub trait RwLockExt<T> {
|
|
fn read_anyhow(&self) -> anyhow::Result<RwLockReadGuard<'_, T>>;
|
|
fn write_anyhow(&self) -> anyhow::Result<RwLockWriteGuard<'_, T>>;
|
|
}
|
|
|
|
impl<T> RwLockExt<T> for RwLock<T> {
|
|
fn read_anyhow(&self) -> anyhow::Result<RwLockReadGuard<'_, T>> {
|
|
self.read().map_err(|e| anyhow!("rwlock poisoned: {e}"))
|
|
}
|
|
|
|
fn write_anyhow(&self) -> anyhow::Result<RwLockWriteGuard<'_, T>> {
|
|
self.write().map_err(|e| anyhow!("rwlock poisoned: {e}"))
|
|
}
|
|
}
|