BatchWriter.h

#pragma once
 
#include <xrpl/nodestore/NodeObject.h>
#include <xrpl/nodestore/Scheduler.h>
#include <xrpl/nodestore/Task.h>
#include <xrpl/nodestore/Types.h>
 
#include <condition_variable>
#include <memory>
#include <mutex>
 
namespace xrpl::node_store {

class BatchWriter : private Task
{
public:
    struct Callback
    {
        virtual ~Callback() = default;
        Callback() = default;
        Callback(Callback const&) = delete;
        Callback&
        operator=(Callback const&) = delete;
 
        virtual void
        writeBatch(Batch const& batch) = 0;
    };

    BatchWriter(Callback& callback, Scheduler& scheduler);

    ~BatchWriter() override;

    void
    store(std::shared_ptr<NodeObject> const& object);

    int
    getWriteLoad();
 
private:
    void
    performScheduledTask() override;
    void
    writeBatch();
    void
    waitForWriting();
 
private:
    using LockType = std::recursive_mutex;
    using CondvarType = std::condition_variable_any;
 
    Callback& callback_;
    Scheduler& scheduler_;
    LockType writeMutex_;
    CondvarType writeCondition_;
    int writeLoad_{0};
    bool writePending_{false};
    Batch writeSet_;
};
 
}  // namespace xrpl::node_store