NodeTask.hpp

// SPDX-FileCopyrightText: 2025 Vincent Leroy
// SPDX-License-Identifier: MIT
//
// This file is part of dfx.
//
// Licensed under the MIT License. See the LICENSE file in the project root
// for full license information.
 
#pragma once
 
// Standard includes
#include <concepts>
 
// Project includes
#include "Task.hpp"
#include <dfx-core/Node.hpp>
 
namespace dfx::Runtime
{
template<typename Func>
requires std::invocable<Func, Core::NodePtr>
class NodeTask : public Task
{
public:
    NodeTask(Func func, Core::NodeWPtr node, int priority = 0)
        : Task(priority)
        , _node { node }
        , _func { std::move(func) }
    {}

    void run() override
    {
        auto node = _node.lock();
        if (node == nullptr)
            return ;
 
        // Ensure only 1 thread enter the node at the same time
        std::lock_guard lock(*node);
        _func(node);
    }
 
private:
    Core::NodeWPtr _node;
    Func           _func;
};

template<typename Func, typename ... Args>
auto makeNodeTask(Func && func, Args && ... args)
{
    return std::make_unique<NodeTask<std::decay_t<Func>>>(std::forward<Func>(func), std::forward<Args>(args)...);
}
} // !namespace dfx::Runtime