AuroraRuntime/Source/Async/AsyncRunnable.hpp

/***
    Copyright (C) 2021 J Reece Wilson (a/k/a "Reece"). All rights reserved.

    File: AsyncRunnable.hpp
    Date: 2021-11-2
    Author: Reece
***/
#pragma once

namespace Aurora::Async
{
    class IAsyncRunnable
    {
    public:
        virtual float GetPrio()  { return 0.5f; };
        virtual void RunAsync() = 0;

        virtual void CancelAsync() {}
    };

    class AsyncFuncRunnable : public IAsyncRunnable
    {
    public:
        
        std::function<void()> callback;
        std::function<void()> fail;
        AuThreadPrimitives::SpinLock lock;
        
        AsyncFuncRunnable(std::function<void()> &&callback) : callback(std::move(callback))
        {}
        
        AsyncFuncRunnable(std::function<void()> &&callback, std::function<void()> &&fail) : callback(std::move(callback)), fail(std::move(fail))
        {}
        
        AsyncFuncRunnable(const std::function<void()> &callback) : callback(callback)
        {}

        AsyncFuncRunnable(const std::function<void()> &callback, const std::function<void()> &fail) : callback(callback), fail(fail)
        {}

        void RunAsync() override
        {
            AU_LOCK_GUARD(lock);
            SysAssertDbgExp(callback, "Missing callback std::function");
            try
            {
                callback();
            } 
            catch (...)
            {
                Debug::PrintError();
            }
            fail = {};
            callback = {};
        }

        void CancelAsync() override
        {
            AU_LOCK_GUARD(lock);
            if (fail)
            {
                try
                {
                    fail();
                } 
                catch (...)
                {
                    Debug::PrintError();
                }
            }
            fail = {};
            callback = {};
        }
    };
}