ScopeGuard.h

Go to the documentation of this file.

#pragma once
 
#include <functional>
 
namespace Gorgon {namespace Utils {
 
    class ScopeGuard {
    public:
        template<class F_>
        ScopeGuard(F_ &&fn) : fn(std::forward<F_>(fn)) {}
 
        ScopeGuard() = default;
 
        ScopeGuard(ScopeGuard &&other) : fn(std::move(other.fn)) {
        }
 
        ScopeGuard(const ScopeGuard&) = delete;
 
        ScopeGuard &operator = (ScopeGuard &&other) {
            fn = std::move(other.fn);
 
            return *this;
        }
 
        ScopeGuard &operator = (const ScopeGuard&) = delete;
 
        void Disarm() {
            fn = nullptr;
        }
 
        template<class F_>
        void Arm(F_ &&value) {
            fn = std::forward<F_>(value);
        }
 
        ~ScopeGuard() {
            if(fn)
                fn();
        }
 
    private:
        std::function<void()> fn;
    };
 
}}