GrpcSource.hpp

#pragma once
 
#include "etl/InitialLoadObserverInterface.hpp"
#include "etl/LoadBalancerInterface.hpp"
#include "util/log/Logger.hpp"
 
#include <boost/asio/spawn.hpp>
#include <grpcpp/support/status.h>
#include <org/xrpl/rpc/v1/get_ledger.pb.h>
#include <xrpl/proto/org/xrpl/rpc/v1/xrp_ledger.grpc.pb.h>
 
#include <atomic>
#include <chrono>
#include <cstdint>
#include <memory>
#include <string>
#include <utility>
 
namespace etl::impl {
 
class GrpcSource {
    util::Logger log_;
    std::unique_ptr<org::xrpl::rpc::v1::XRPLedgerAPIService::Stub> stub_;
    std::unique_ptr<std::atomic_bool> initialLoadShouldStop_;
    std::chrono::system_clock::duration deadline_;
 
    static constexpr auto kKeepalivePingIntervalMs = 10000;
    static constexpr auto kKeepaliveTimeoutMs = 5000;
    static constexpr auto kKeepalivePermitWithoutCalls =
        true;                                        // Allow keepalive pings when no calls
    static constexpr auto kMaxPingsWithoutData = 0;  // No limit
    static constexpr auto kDeadline = std::chrono::seconds(30);
 
public:
    GrpcSource(
        std::string const& ip,
        std::string const& grpcPort,
        std::chrono::system_clock::duration deadline = kDeadline
    );

    std::pair<grpc::Status, org::xrpl::rpc::v1::GetLedgerResponse>
    fetchLedger(uint32_t sequence, bool getObjects = true, bool getObjectNeighbors = false);

    InitialLedgerLoadResult
    loadInitialLedger(
        uint32_t sequence,
        uint32_t numMarkers,
        InitialLoadObserverInterface& observer
    );

    void
    stop(boost::asio::yield_context yield);
};
 
}  // namespace etl::impl