50 lines
771 B
Rust
50 lines
771 B
Rust
// Extracted from the scopeguard crate
|
|
use core::ops::{Deref, DerefMut};
|
|
|
|
pub struct ScopeGuard<T, F>
|
|
where
|
|
F: FnMut(&mut T),
|
|
{
|
|
dropfn: F,
|
|
value: T,
|
|
}
|
|
|
|
#[inline]
|
|
pub fn guard<T, F>(value: T, dropfn: F) -> ScopeGuard<T, F>
|
|
where
|
|
F: FnMut(&mut T),
|
|
{
|
|
ScopeGuard { dropfn, value }
|
|
}
|
|
|
|
impl<T, F> Deref for ScopeGuard<T, F>
|
|
where
|
|
F: FnMut(&mut T),
|
|
{
|
|
type Target = T;
|
|
#[inline]
|
|
fn deref(&self) -> &T {
|
|
&self.value
|
|
}
|
|
}
|
|
|
|
impl<T, F> DerefMut for ScopeGuard<T, F>
|
|
where
|
|
F: FnMut(&mut T),
|
|
{
|
|
#[inline]
|
|
fn deref_mut(&mut self) -> &mut T {
|
|
&mut self.value
|
|
}
|
|
}
|
|
|
|
impl<T, F> Drop for ScopeGuard<T, F>
|
|
where
|
|
F: FnMut(&mut T),
|
|
{
|
|
#[inline]
|
|
fn drop(&mut self) {
|
|
(self.dropfn)(&mut self.value)
|
|
}
|
|
}
|