ExceptionContext.h

Go to the documentation of this file.

 
#pragma once
 
#include "ncengine/type/StableAddress.h"
 
#include <concepts>
#include <exception>
#include <mutex>
#include <utility>
 
namespace nc::task
{
class ExceptionContext : StableAddress
{
    public:
        void StoreException(std::exception_ptr e)
        {
            auto lock = std::lock_guard{m_mutex};
            if(m_exception)
            {
                return;
            }
 
            m_exception = e;
        }
 
        void ThrowIfExceptionStored()
        {
            if(m_exception)
            {
                auto e = std::exchange(m_exception, nullptr);
                std::rethrow_exception(e);
            }
        }
 
    private:
        std::exception_ptr m_exception;
        std::mutex m_mutex;
};
 
template<std::invocable<> F>
auto Guard(ExceptionContext& context, F&& func)
{
    if constexpr(std::is_nothrow_invocable_v<F>)
    {
        return std::forward<F>(func);
    }
    else
    {
        return [func = std::forward<F>(func), &context]
        {
            try
            {
                func();
            }
            catch(const std::exception&)
            {
                context.StoreException(std::current_exception());
            }
        };
    }
}
} // namespace nc::task