use std::ops::Deref;
pub struct Dirty<T> {
value: T,
dirty: bool,
}
impl<T> Dirty<T> {
pub fn new(val: T) -> Dirty<T> {
Dirty {
value: val,
dirty: true,
}
}
pub fn dirty(&self) -> bool {
self.dirty
}
pub fn write(&mut self) -> &mut T {
self.dirty = true;
&mut self.value
}
pub fn read(&self) -> &T {
&self.value
}
pub fn clear(&mut self) {
self.dirty = false;
}
pub fn read_dirty(&self) -> Option<&T> {
match self.dirty {
true => {
Some(&self.value)
},
false => None,
}
}
}
impl<T> Deref for Dirty<T> {
type Target = T;
fn deref(&self) -> &T {
&self.value
}
}
impl<T> Default for Dirty<T> where T: Default {
fn default() -> Self {
Dirty::new(T::default())
}
}
#[cfg(test)]
mod tests {
use super::Dirty;
#[test]
fn new_dirty() {
let dirty = Dirty::new(0);
assert!(dirty.dirty());
}
#[test]
fn read_doesnt_clear_flag() {
let dirty = Dirty::new(0);
assert!(dirty.dirty());
assert!(*dirty.read() == 0);
assert!(dirty.dirty());
}
#[test]
fn write_sets_flag() {
let mut dirty = Dirty::new(0);
assert!(*dirty.read() == 0);
dirty.clear();
assert!(!dirty.dirty());
*dirty.write() += 1;
assert!(dirty.dirty());
}
#[test]
fn read_dirty() {
let mut dirty = Dirty::new(0);
assert!(dirty.read_dirty().is_some());
dirty.clear();
assert!(!dirty.dirty());
assert!(dirty.read_dirty() == None);
assert!(!dirty.dirty());
*dirty.write() += 1;
assert!(dirty.dirty());
assert!(dirty.read_dirty().is_some());
dirty.clear();
assert!(!dirty.dirty());
assert!(dirty.read_dirty() == None);
}
#[test]
fn access_inner_deref() {
let dirty = Dirty::new(0);
assert!(*dirty == 0);
}
#[test]
fn default_value() {
let dirty = Dirty::<i32>::default();
assert!(*dirty == 0);
}
}