diff options
Diffstat (limited to 'src/poisonable')
| -rwxr-xr-x | src/poisonable/poisonable.rs | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/poisonable/poisonable.rs b/src/poisonable/poisonable.rs index dd12cea..0134ff1 100755 --- a/src/poisonable/poisonable.rs +++ b/src/poisonable/poisonable.rs @@ -1,10 +1,11 @@ use std::panic::{RefUnwindSafe, UnwindSafe}; +use crate::collection::OwnedLockCollection; use crate::handle_unwind::handle_unwind; use crate::lockable::{ Lockable, LockableGetMut, LockableIntoInner, OwnedLockable, RawLock, Sharable, }; -use crate::{Keyable, ThreadKey}; +use crate::{Keyable, LockContext, ThreadKey}; use super::{ PoisonError, PoisonFlag, PoisonGuard, PoisonRef, PoisonResult, Poisonable, @@ -853,5 +854,24 @@ impl<L: LockableGetMut + RawLock> Poisonable<L> { } } +impl<L: OwnedLockable> Poisonable<OwnedLockCollection<L>> { + /// Creates a context that can be used to iterate over the items in order. + /// + /// For more information, see [`OwnedLockCollection::context`]. + /// + /// # Errors + /// + /// If another user of this lock panicked while holding the lock, then + /// this call will return an error instead. A `Poisonable` is poisoned + /// whenever a thread panics while holding a lock. + pub fn context(&self) -> PoisonResult<LockContext<'_, L>> { + if self.is_poisoned() { + Ok(self.inner.context()) + } else { + Err(PoisonError::new(self.inner.context())) + } + } +} + impl<L: UnwindSafe> RefUnwindSafe for Poisonable<L> {} impl<L: UnwindSafe> UnwindSafe for Poisonable<L> {} |