#include <xrpld/app/main/NodeStoreScheduler.h>

#include <xrpl/core/Job.h>
#include <xrpl/core/JobQueue.h>
#include <xrpl/nodestore/Scheduler.h>
#include <xrpl/nodestore/Task.h>

#include <chrono>

namespace xrpl {

NodeStoreScheduler::NodeStoreScheduler(JobQueue& jobQueue) : jobQueue_(jobQueue)
{
}

void
NodeStoreScheduler::scheduleTask(node_store::Task& task)
{
    if (jobQueue_.isStopped())
        return;

    if (!jobQueue_.addJob(JtWrite, "NObjStore", [&task]() { task.performScheduledTask(); }))
    {
        // Job not added, presumably because we're shutting down.
        // Recover by executing the task synchronously.
        task.performScheduledTask();
    }
}

void
NodeStoreScheduler::onFetch(node_store::FetchReport const& report)
{
    if (jobQueue_.isStopped())
        return;

    // The report is in microseconds but addLoadEvents takes milliseconds, so
    // cast explicitly. The load monitor only tracks whole-millisecond load,
    // so the sub-millisecond detail is deliberately dropped here; telemetry
    // reads the microsecond value from the nodestore instead.
    jobQueue_.addLoadEvents(
        report.fetchType == node_store::FetchType::Async ? JtNsAsyncRead : JtNsSyncRead,
        1,
        std::chrono::duration_cast<std::chrono::milliseconds>(report.elapsed));
}

void
NodeStoreScheduler::onBatchWrite(node_store::BatchWriteReport const& report)
{
    if (jobQueue_.isStopped())
        return;

    jobQueue_.addLoadEvents(JtNsWrite, report.writeCount, report.elapsed);
}

}  // namespace xrpl