rejit_work_offloader.h

/*
 * Copyright The OpenTelemetry Authors
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef OTEL_CLR_PROFILER_REJIT_WORK_OFFLOADER_H_
#define OTEL_CLR_PROFILER_REJIT_WORK_OFFLOADER_H_

#include <atomic>
#include <future>
#include <mutex>
#include <shared_mutex>
#include <string>
#include <unordered_map>
#include <vector>

#include "cor.h"
#include "corprof.h"
#include "module_metadata.h"

namespace trace
{

struct RejitWorkItem
{
    const bool terminating = false;
    const std::function<void()> func = nullptr;

    RejitWorkItem();

    RejitWorkItem(std::function<void()>&& func);

    static std::unique_ptr<RejitWorkItem> CreateTerminatingWorkItem();
};

class RejitWorkOffloader
{

private:
    ICorProfilerInfo7* m_profilerInfo;

    std::unique_ptr<UniqueBlockingQueue<RejitWorkItem>> m_offloader_queue;
    std::unique_ptr<std::thread> m_offloader_queue_thread;

    static void EnqueueThreadLoop(RejitWorkOffloader* offloader);

public:
    RejitWorkOffloader(ICorProfilerInfo7* pInfo);

    void Enqueue(std::unique_ptr<RejitWorkItem>&& item);
    bool WaitForTermination();
};

} // namespace trace

#endif // OTEL_CLR_PROFILER_REJIT_WORK_OFFLOADER_H_