CNodeExecutor.hpp

// SPDX-FileCopyrightText: 2026 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 <mutex>
#include <vector>
 
// Project includes
#include "../../Plugin.hpp"
#include <dfx-plugins-interface/PluginInterface.h>
#include <dfx-runtime-api/NodeTaskExecutor.hpp>
#include <dfx-utilities/sync-queues/PriorityStableQueue.hpp>
 
namespace dfx::Plugins
{
class CNodeExecutor : public Runtime::Api::NodeTaskExecutor
{
    struct TaskContext
    {
        Runtime::Api::TaskPtr task;
        std::atomic_flag hasBeenPushed;
        CNodeExecutor * self;
    };
 
public:
    CNodeExecutor(Plugin const & plugin, dfx_task_executor_interface_t interface);

    ~CNodeExecutor();
 
public:
    bool pushTask(Runtime::Api::TaskPtr task) override;

    std::size_t pushTasks(std::span<Runtime::Api::TaskPtr> tasks) override;
 
private:
    std::unique_ptr<TaskContext> _createContext(Runtime::Api::TaskPtr task);
    dfx_task_api_t _createTaskApi(TaskContext * ctx) noexcept;
    void _onTaskFinished(TaskContext const & ctx) noexcept;
 
private:
    Plugin const & _plugin;
    dfx_task_executor_interface_t _interface{};
    dfx_task_executor_handle_t _handle{};
 
private:
    std::vector<std::unique_ptr<TaskContext>> _tasks;
    std::mutex                                _mutex;
};
} // !namespace dfx::Plugins