FunctionGuard.h

#pragma once
 
 
//--- User configuration header ---//
#include "TasksConfig.h"
 
NAMESPACE_SQUID_BEGIN
 
template <typename tFn = std::function<void()>>
class FunctionGuard
{
public:
    FunctionGuard() = default; 
    FunctionGuard(nullptr_t) 
    {
    }
    FunctionGuard(tFn in_fn) 
        : m_fn(std::move(in_fn))
    {
    }
    ~FunctionGuard() 
    {
        Execute();
    }
    FunctionGuard(FunctionGuard&& in_other) noexcept 
        : m_fn(std::move(in_other.m_fn))
    {
        in_other.Forget();
    }
    FunctionGuard& operator=(FunctionGuard<tFn>&& in_other) noexcept 
    {
        m_fn = std::move(in_other.m_fn);
        in_other.Forget();
        return *this;
    }
    FunctionGuard& operator=(nullptr_t) noexcept 
    {
        Forget();
        return *this;
    }
    operator bool() const 
    {
        return IsBound();
    }
    bool IsBound() noexcept 
    {
        return m_fn;
    }
    void Execute() 
    {
        if(m_fn)
        {
            m_fn.value()();
            Forget();
        }
    }
    void Forget() noexcept 
    {
        m_fn.reset();
    }
 
private:
    std::optional<tFn> m_fn; // The function to call when this scope guard is destroyed
};
 
template <typename tFn>
FunctionGuard<tFn> MakeFnGuard(tFn in_fn)
{
    return FunctionGuard<tFn>(std::move(in_fn));
}
 
inline FunctionGuard<> MakeGenericFnGuard(std::function<void()> in_fn)
{
    return FunctionGuard<>(std::move(in_fn));
}
 
NAMESPACE_SQUID_END