// SPDX-FileCopyrightText: 2022-Present Nils Andreas Svee mailto:contact@lochnair.net (github @Lochnair)
//
// SPDX-License-Identifier: MPL-2.0

use anyhow::anyhow;
use std::sync::{Arc, RwLock, RwLockReadGuard, RwLockWriteGuard};

pub type ArcRwLock<T> = Arc<RwLock<T>>;

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}"))
    }
}