fast-able 1.20.2

/// Guard structure for defer
/// defer 的守卫结构体
pub struct Guard<F: FnMut()>(pub Option<F>);

impl<F: FnMut()> Drop for Guard<F> {
    fn drop(&mut self) {
        if let Some(mut f) = (self.0).take() {
            f()
        }
    }
}

/// Defers evaluation of a block of code until the end of the scope.
/// Sort of LIFO(last-in, first-out queue)
/// If you encounter references to resources that cannot mut more than 2 times, try nesting RefCell and then use.borrow() and.borrow_mut().
///
/// 延迟代码块的执行直到作用域结束。
/// 类似于 LIFO（后进先出队列）
/// 如果遇到不能 mut 超过 2 次的资源引用，尝试嵌套 RefCell 然后使用 .borrow() 和 .borrow_mut()。
///
/// for example:
/// ```
///  use dark_std::defer;
///  //LIFO, so it will print: guard: 3  guard: 2   guard: 1
///  fn main(){
///     defer!({
///        println!("guard: 1");
///        });
///     defer!(||{
///        println!("guard: 2");
///        });
///     defer!{
///        println!("guard: 3");
///     }
/// }
/// ```
///
///
#[macro_export]
macro_rules! defer {
    ($func:block) => {
       let _guard = $crate::defer::Guard(Some( ||$func));
    };
    ($func:expr) => {
        let _guard = $crate::defer::Guard(Some($func));
    };
    { $($func:expr$(;)?)+ } => {
       let _guard = $crate::defer::Guard(Some( ||{$($func;)+}));
    }
}

#[cfg(test)]
mod test {
    #[test]
    fn test_defer() {
        defer!(|| {
            println!("defer");
        });
    }
}