You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
1.0 KiB
55 lines
1.0 KiB
#![allow(clippy::module_name_repetitions)]
|
|
|
|
use std::error::Error as StdError;
|
|
use std::fmt::{self, Display};
|
|
use std::sync::atomic::AtomicBool;
|
|
use std::sync::atomic::Ordering::SeqCst;
|
|
use std::sync::Arc;
|
|
|
|
#[derive(Debug)]
|
|
pub struct Flag {
|
|
atomic: Arc<AtomicBool>,
|
|
}
|
|
|
|
impl Flag {
|
|
pub fn new() -> Self {
|
|
Flag {
|
|
atomic: Arc::new(AtomicBool::new(false)),
|
|
}
|
|
}
|
|
|
|
pub fn get(&self) -> bool {
|
|
self.atomic.load(SeqCst)
|
|
}
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct DetectDrop {
|
|
has_dropped: Flag,
|
|
}
|
|
|
|
impl DetectDrop {
|
|
pub fn new(has_dropped: &Flag) -> Self {
|
|
DetectDrop {
|
|
has_dropped: Flag {
|
|
atomic: Arc::clone(&has_dropped.atomic),
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
impl StdError for DetectDrop {}
|
|
|
|
impl Display for DetectDrop {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
write!(f, "oh no!")
|
|
}
|
|
}
|
|
|
|
impl Drop for DetectDrop {
|
|
fn drop(&mut self) {
|
|
let already_dropped = self.has_dropped.atomic.swap(true, SeqCst);
|
|
assert!(!already_dropped);
|
|
}
|
|
}
|