mirror of
https://github.com/XRPLF/clio.git
synced 2025-11-20 03:35:55 +00:00
890 lines
28 KiB
C++
890 lines
28 KiB
C++
|
|
|
|
//------------------------------------------------------------------------------
|
|
/*
|
|
This file is part of rippled: https://github.com/ripple/rippled
|
|
Copyright (c) 2020 Ripple Labs Inc.
|
|
|
|
Permission to use, copy, modify, and/or distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
ANY SPECIAL , DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
//==============================================================================
|
|
|
|
#include <ripple/protocol/STLedgerEntry.h>
|
|
#include <boost/asio/strand.hpp>
|
|
#include <boost/json.hpp>
|
|
#include <boost/json/src.hpp>
|
|
#include <boost/log/trivial.hpp>
|
|
#include <reporting/ETLSource.h>
|
|
|
|
// Create ETL source without grpc endpoint
|
|
// Fetch ledger and load initial ledger will fail for this source
|
|
// Primarly used in read-only mode, to monitor when ledgers are validated
|
|
ETLSource::ETLSource(
|
|
boost::json::object const& config,
|
|
BackendInterface& backend,
|
|
NetworkValidatedLedgers& networkValidatedLedgers,
|
|
boost::asio::io_context& ioContext)
|
|
: ioc_(ioContext)
|
|
, ws_(std::make_unique<
|
|
boost::beast::websocket::stream<boost::beast::tcp_stream>>(
|
|
boost::asio::make_strand(ioc_)))
|
|
, resolver_(boost::asio::make_strand(ioc_))
|
|
, timer_(ioc_)
|
|
, networkValidatedLedgers_(networkValidatedLedgers)
|
|
, backend_(backend)
|
|
{
|
|
if (config.contains("ip"))
|
|
{
|
|
auto ipJs = config.at("ip").as_string();
|
|
ip_ = {ipJs.c_str(), ipJs.size()};
|
|
}
|
|
if (config.contains("ws_port"))
|
|
{
|
|
auto portjs = config.at("ws_port").as_string();
|
|
wsPort_ = {portjs.c_str(), portjs.size()};
|
|
}
|
|
if (config.contains("grpc_port"))
|
|
{
|
|
auto portjs = config.at("grpc_port").as_string();
|
|
grpcPort_ = {portjs.c_str(), portjs.size()};
|
|
try
|
|
{
|
|
boost::asio::ip::tcp::endpoint endpoint{
|
|
boost::asio::ip::make_address(ip_), std::stoi(grpcPort_)};
|
|
std::stringstream ss;
|
|
ss << endpoint;
|
|
stub_ = org::xrpl::rpc::v1::XRPLedgerAPIService::NewStub(
|
|
grpc::CreateChannel(
|
|
ss.str(), grpc::InsecureChannelCredentials()));
|
|
BOOST_LOG_TRIVIAL(debug) << "Made stub for remote = " << toString();
|
|
}
|
|
catch (std::exception const& e)
|
|
{
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< "Exception while creating stub = " << e.what()
|
|
<< " . Remote = " << toString();
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLSource::reconnect(boost::beast::error_code ec)
|
|
{
|
|
connected_ = false;
|
|
// These are somewhat normal errors. operation_aborted occurs on shutdown,
|
|
// when the timer is cancelled. connection_refused will occur repeatedly
|
|
// if we cannot connect to the transaction processing process
|
|
if (ec != boost::asio::error::operation_aborted &&
|
|
ec != boost::asio::error::connection_refused)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error)
|
|
<< __func__ << " : "
|
|
<< "error code = " << ec << " - " << toString();
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< __func__ << " : "
|
|
<< "error code = " << ec << " - " << toString();
|
|
}
|
|
|
|
// exponentially increasing timeouts, with a max of 30 seconds
|
|
size_t waitTime = std::min(pow(2, numFailures_), 30.0);
|
|
numFailures_++;
|
|
timer_.expires_after(boost::asio::chrono::seconds(waitTime));
|
|
timer_.async_wait([this](auto ec) {
|
|
bool startAgain = (ec != boost::asio::error::operation_aborted);
|
|
BOOST_LOG_TRIVIAL(trace) << __func__ << " async_wait : ec = " << ec;
|
|
close(startAgain);
|
|
});
|
|
}
|
|
|
|
void
|
|
ETLSource::close(bool startAgain)
|
|
{
|
|
timer_.cancel();
|
|
ioc_.post([this, startAgain]() {
|
|
if (closing_)
|
|
return;
|
|
|
|
if (ws_->is_open())
|
|
{
|
|
// onStop() also calls close(). If the async_close is called twice,
|
|
// an assertion fails. Using closing_ makes sure async_close is only
|
|
// called once
|
|
closing_ = true;
|
|
ws_->async_close(
|
|
boost::beast::websocket::close_code::normal,
|
|
[this, startAgain](auto ec) {
|
|
if (ec)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error)
|
|
<< __func__ << " async_close : "
|
|
<< "error code = " << ec << " - " << toString();
|
|
}
|
|
closing_ = false;
|
|
if (startAgain)
|
|
start();
|
|
});
|
|
}
|
|
else if (startAgain)
|
|
{
|
|
start();
|
|
}
|
|
});
|
|
}
|
|
|
|
void
|
|
ETLSource::start()
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace) << __func__ << " : " << toString();
|
|
|
|
auto const host = ip_;
|
|
auto const port = wsPort_;
|
|
|
|
resolver_.async_resolve(
|
|
host, port, [this](auto ec, auto results) { onResolve(ec, results); });
|
|
}
|
|
|
|
void
|
|
ETLSource::onResolve(
|
|
boost::beast::error_code ec,
|
|
boost::asio::ip::tcp::resolver::results_type results)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : ec = " << ec << " - " << toString();
|
|
if (ec)
|
|
{
|
|
// try again
|
|
reconnect(ec);
|
|
}
|
|
else
|
|
{
|
|
boost::beast::get_lowest_layer(*ws_).expires_after(
|
|
std::chrono::seconds(30));
|
|
boost::beast::get_lowest_layer(*ws_).async_connect(
|
|
results, [this](auto ec, auto ep) { onConnect(ec, ep); });
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLSource::onConnect(
|
|
boost::beast::error_code ec,
|
|
boost::asio::ip::tcp::resolver::results_type::endpoint_type endpoint)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : ec = " << ec << " - " << toString();
|
|
if (ec)
|
|
{
|
|
// start over
|
|
reconnect(ec);
|
|
}
|
|
else
|
|
{
|
|
numFailures_ = 0;
|
|
// Turn off timeout on the tcp stream, because websocket stream has it's
|
|
// own timeout system
|
|
boost::beast::get_lowest_layer(*ws_).expires_never();
|
|
|
|
// Set suggested timeout settings for the websocket
|
|
ws_->set_option(
|
|
boost::beast::websocket::stream_base::timeout::suggested(
|
|
boost::beast::role_type::client));
|
|
|
|
// Set a decorator to change the User-Agent of the handshake
|
|
ws_->set_option(boost::beast::websocket::stream_base::decorator(
|
|
[](boost::beast::websocket::request_type& req) {
|
|
req.set(
|
|
boost::beast::http::field::user_agent,
|
|
std::string(BOOST_BEAST_VERSION_STRING) +
|
|
" websocket-client-async");
|
|
}));
|
|
|
|
// Update the host_ string. This will provide the value of the
|
|
// Host HTTP header during the WebSocket handshake.
|
|
// See https://tools.ietf.org/html/rfc7230#section-5.4
|
|
auto host = ip_ + ':' + std::to_string(endpoint.port());
|
|
// Perform the websocket handshake
|
|
ws_->async_handshake(host, "/", [this](auto ec) { onHandshake(ec); });
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLSource::onHandshake(boost::beast::error_code ec)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : ec = " << ec << " - " << toString();
|
|
if (ec)
|
|
{
|
|
// start over
|
|
reconnect(ec);
|
|
}
|
|
else
|
|
{
|
|
boost::json::object jv{
|
|
{"command", "subscribe"},
|
|
{"streams", {"ledger", "transactions_proposed"}}};
|
|
std::string s = boost::json::serialize(jv);
|
|
BOOST_LOG_TRIVIAL(trace) << "Sending subscribe stream message";
|
|
// Send the message
|
|
ws_->async_write(boost::asio::buffer(s), [this](auto ec, size_t size) {
|
|
onWrite(ec, size);
|
|
});
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLSource::onWrite(boost::beast::error_code ec, size_t bytesWritten)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : ec = " << ec << " - " << toString();
|
|
if (ec)
|
|
{
|
|
// start over
|
|
reconnect(ec);
|
|
}
|
|
else
|
|
{
|
|
ws_->async_read(
|
|
readBuffer_, [this](auto ec, size_t size) { onRead(ec, size); });
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLSource::onRead(boost::beast::error_code ec, size_t size)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : ec = " << ec << " - " << toString();
|
|
// if error or error reading message, start over
|
|
if (ec)
|
|
{
|
|
reconnect(ec);
|
|
}
|
|
else
|
|
{
|
|
handleMessage();
|
|
boost::beast::flat_buffer buffer;
|
|
swap(readBuffer_, buffer);
|
|
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : calling async_read - " << toString();
|
|
ws_->async_read(
|
|
readBuffer_, [this](auto ec, size_t size) { onRead(ec, size); });
|
|
}
|
|
}
|
|
|
|
bool
|
|
ETLSource::handleMessage()
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace) << __func__ << " : " << toString();
|
|
|
|
setLastMsgTime();
|
|
connected_ = true;
|
|
try
|
|
{
|
|
std::string msg{
|
|
static_cast<char const*>(readBuffer_.data().data()),
|
|
readBuffer_.size()};
|
|
// BOOST_LOG_TRIVIAL(debug) << __func__ << msg;
|
|
boost::json::value raw = boost::json::parse(msg);
|
|
// BOOST_LOG_TRIVIAL(debug) << __func__ << " parsed";
|
|
boost::json::object response = raw.as_object();
|
|
|
|
uint32_t ledgerIndex = 0;
|
|
if (response.contains("result"))
|
|
{
|
|
boost::json::object result = response["result"].as_object();
|
|
if (result.contains("ledger_index"))
|
|
{
|
|
ledgerIndex = result["ledger_index"].as_int64();
|
|
}
|
|
if (result.contains("validated_ledgers"))
|
|
{
|
|
boost::json::string const& validatedLedgers =
|
|
result["validated_ledgers"].as_string();
|
|
setValidatedRange(
|
|
{validatedLedgers.c_str(), validatedLedgers.size()});
|
|
}
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< __func__ << " : "
|
|
<< "Received a message on ledger "
|
|
<< " subscription stream. Message : " << response << " - "
|
|
<< toString();
|
|
}
|
|
else
|
|
{
|
|
if (response.contains("transaction"))
|
|
{
|
|
/*
|
|
if
|
|
(etl_.getETLLoadBalancer().shouldPropagateTxnStream(this))
|
|
{
|
|
etl_.getApplication().getOPs().forwardProposedTransaction(
|
|
response);
|
|
}
|
|
*/
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< __func__ << " : "
|
|
<< "Received a message on ledger "
|
|
<< " subscription stream. Message : " << response << " - "
|
|
<< toString();
|
|
if (response.contains("ledger_index"))
|
|
{
|
|
ledgerIndex = response["ledger_index"].as_int64();
|
|
}
|
|
if (response.contains("validated_ledgers"))
|
|
{
|
|
boost::json::string const& validatedLedgers =
|
|
response["validated_ledgers"].as_string();
|
|
setValidatedRange(
|
|
{validatedLedgers.c_str(), validatedLedgers.size()});
|
|
}
|
|
}
|
|
}
|
|
|
|
if (ledgerIndex != 0)
|
|
{
|
|
BOOST_LOG_TRIVIAL(trace)
|
|
<< __func__ << " : "
|
|
<< "Pushing ledger sequence = " << ledgerIndex << " - "
|
|
<< toString();
|
|
networkValidatedLedgers_.push(ledgerIndex);
|
|
}
|
|
return true;
|
|
}
|
|
catch (std::exception const& e)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "Exception in handleMessage : " << e.what();
|
|
return false;
|
|
}
|
|
}
|
|
|
|
class AsyncCallData
|
|
{
|
|
std::unique_ptr<org::xrpl::rpc::v1::GetLedgerDataResponse> cur_;
|
|
std::unique_ptr<org::xrpl::rpc::v1::GetLedgerDataResponse> next_;
|
|
|
|
org::xrpl::rpc::v1::GetLedgerDataRequest request_;
|
|
std::unique_ptr<grpc::ClientContext> context_;
|
|
|
|
grpc::Status status_;
|
|
|
|
public:
|
|
AsyncCallData(uint32_t seq)
|
|
{
|
|
request_.mutable_ledger()->set_sequence(seq);
|
|
request_.set_user("ETL");
|
|
|
|
cur_ = std::make_unique<org::xrpl::rpc::v1::GetLedgerDataResponse>();
|
|
|
|
next_ = std::make_unique<org::xrpl::rpc::v1::GetLedgerDataResponse>();
|
|
|
|
context_ = std::make_unique<grpc::ClientContext>();
|
|
}
|
|
|
|
enum class CallStatus { MORE, DONE, ERRORED };
|
|
CallStatus
|
|
process(
|
|
std::unique_ptr<org::xrpl::rpc::v1::XRPLedgerAPIService::Stub>& stub,
|
|
grpc::CompletionQueue& cq,
|
|
BackendInterface& backend,
|
|
bool abort = false)
|
|
{
|
|
BOOST_LOG_TRIVIAL(info) << "Processing response. "
|
|
<< "Marker prefix = " << getMarkerPrefix();
|
|
if (abort)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "AsyncCallData aborted";
|
|
return CallStatus::ERRORED;
|
|
}
|
|
if (!status_.ok())
|
|
{
|
|
BOOST_LOG_TRIVIAL(error)
|
|
<< "AsyncCallData status_ not ok: "
|
|
<< " code = " << status_.error_code()
|
|
<< " message = " << status_.error_message();
|
|
return CallStatus::ERRORED;
|
|
}
|
|
if (!next_->is_unlimited())
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< "AsyncCallData is_unlimited is false. Make sure "
|
|
"secure_gateway is set correctly at the ETL source";
|
|
assert(false);
|
|
}
|
|
|
|
std::swap(cur_, next_);
|
|
|
|
bool more = true;
|
|
|
|
// if no marker returned, we are done
|
|
if (cur_->marker().size() == 0)
|
|
more = false;
|
|
|
|
// if we are not done, make the next async call
|
|
if (more)
|
|
{
|
|
request_.set_marker(std::move(cur_->marker()));
|
|
call(stub, cq);
|
|
}
|
|
|
|
BOOST_LOG_TRIVIAL(trace) << "Writing objects";
|
|
for (auto& obj : *(cur_->mutable_ledger_objects()->mutable_objects()))
|
|
{
|
|
std::optional<ripple::uint256> book;
|
|
|
|
short offer_bytes = (obj.data()[1] << 8) | obj.data()[2];
|
|
if (offer_bytes == 0x006f)
|
|
{
|
|
ripple::SerialIter it{obj.data().data(), obj.data().size()};
|
|
ripple::SLE sle{it, {}};
|
|
book = sle.getFieldH256(ripple::sfBookDirectory);
|
|
for (size_t i = 0; i < 8; ++i)
|
|
{
|
|
book->data()[book->size() - 1 - i] = 0x00;
|
|
}
|
|
}
|
|
backend.writeLedgerObject(
|
|
std::move(*obj.mutable_key()),
|
|
request_.ledger().sequence(),
|
|
std::move(*obj.mutable_data()),
|
|
true,
|
|
false,
|
|
std::move(book));
|
|
}
|
|
BOOST_LOG_TRIVIAL(trace) << "Wrote objects";
|
|
|
|
return more ? CallStatus::MORE : CallStatus::DONE;
|
|
}
|
|
|
|
void
|
|
call(
|
|
std::unique_ptr<org::xrpl::rpc::v1::XRPLedgerAPIService::Stub>& stub,
|
|
grpc::CompletionQueue& cq)
|
|
{
|
|
BOOST_LOG_TRIVIAL(info) << "Making next request. " << getMarkerPrefix();
|
|
context_ = std::make_unique<grpc::ClientContext>();
|
|
|
|
std::unique_ptr<grpc::ClientAsyncResponseReader<
|
|
org::xrpl::rpc::v1::GetLedgerDataResponse>>
|
|
rpc(stub->PrepareAsyncGetLedgerData(context_.get(), request_, &cq));
|
|
|
|
rpc->StartCall();
|
|
|
|
rpc->Finish(next_.get(), &status_, this);
|
|
}
|
|
|
|
std::string
|
|
getMarkerPrefix()
|
|
{
|
|
if (next_->marker().size() == 0)
|
|
return "";
|
|
else
|
|
return ripple::strHex(std::string{next_->marker().data()[0]});
|
|
}
|
|
};
|
|
|
|
bool
|
|
ETLSource::loadInitialLedger(uint32_t sequence)
|
|
{
|
|
if (!stub_)
|
|
return false;
|
|
|
|
grpc::CompletionQueue cq;
|
|
|
|
void* tag;
|
|
|
|
bool ok = false;
|
|
|
|
std::vector<AsyncCallData> calls;
|
|
calls.emplace_back(sequence);
|
|
|
|
BOOST_LOG_TRIVIAL(info) << "Starting data download for ledger " << sequence
|
|
<< ". Using source = " << toString();
|
|
|
|
for (auto& c : calls)
|
|
c.call(stub_, cq);
|
|
|
|
size_t numFinished = 0;
|
|
bool abort = false;
|
|
while (numFinished < calls.size() && cq.Next(&tag, &ok))
|
|
{
|
|
assert(tag);
|
|
|
|
auto ptr = static_cast<AsyncCallData*>(tag);
|
|
|
|
if (!ok)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "loadInitialLedger - ok is false";
|
|
return false;
|
|
// handle cancelled
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(info)
|
|
<< "Marker prefix = " << ptr->getMarkerPrefix();
|
|
auto result = ptr->process(stub_, cq, backend_, abort);
|
|
if (result != AsyncCallData::CallStatus::MORE)
|
|
{
|
|
numFinished++;
|
|
BOOST_LOG_TRIVIAL(info)
|
|
<< "Finished a marker. "
|
|
<< "Current number of finished = " << numFinished;
|
|
}
|
|
if (result == AsyncCallData::CallStatus::ERRORED)
|
|
{
|
|
abort = true;
|
|
}
|
|
}
|
|
}
|
|
return !abort;
|
|
}
|
|
|
|
std::pair<grpc::Status, org::xrpl::rpc::v1::GetLedgerResponse>
|
|
ETLSource::fetchLedger(uint32_t ledgerSequence, bool getObjects)
|
|
{
|
|
org::xrpl::rpc::v1::GetLedgerResponse response;
|
|
if (!stub_)
|
|
return {{grpc::StatusCode::INTERNAL, "No Stub"}, response};
|
|
|
|
// ledger header with txns and metadata
|
|
org::xrpl::rpc::v1::GetLedgerRequest request;
|
|
grpc::ClientContext context;
|
|
request.mutable_ledger()->set_sequence(ledgerSequence);
|
|
request.set_transactions(true);
|
|
request.set_expand(true);
|
|
request.set_get_objects(getObjects);
|
|
request.set_user("ETL");
|
|
grpc::Status status = stub_->GetLedger(&context, request, &response);
|
|
if (status.ok() && !response.is_unlimited())
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< "ETLSource::fetchLedger - is_unlimited is "
|
|
"false. Make sure secure_gateway is set "
|
|
"correctly on the ETL source. source = "
|
|
<< toString() << " status = " << status.error_message();
|
|
assert(false);
|
|
}
|
|
return {status, std::move(response)};
|
|
}
|
|
ETLLoadBalancer::ETLLoadBalancer(
|
|
boost::json::array const& config,
|
|
BackendInterface& backend,
|
|
NetworkValidatedLedgers& nwvl,
|
|
boost::asio::io_context& ioContext)
|
|
{
|
|
for (auto& entry : config)
|
|
{
|
|
std::unique_ptr<ETLSource> source = std::make_unique<ETLSource>(
|
|
entry.as_object(), backend, nwvl, ioContext);
|
|
sources_.push_back(std::move(source));
|
|
BOOST_LOG_TRIVIAL(info) << __func__ << " : added etl source - "
|
|
<< sources_.back()->toString();
|
|
}
|
|
}
|
|
|
|
void
|
|
ETLLoadBalancer::loadInitialLedger(uint32_t sequence)
|
|
{
|
|
execute(
|
|
[this, &sequence](auto& source) {
|
|
bool res = source->loadInitialLedger(sequence);
|
|
if (!res)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "Failed to download initial ledger."
|
|
<< " Sequence = " << sequence
|
|
<< " source = " << source->toString();
|
|
}
|
|
return res;
|
|
},
|
|
sequence);
|
|
}
|
|
|
|
std::optional<org::xrpl::rpc::v1::GetLedgerResponse>
|
|
ETLLoadBalancer::fetchLedger(uint32_t ledgerSequence, bool getObjects)
|
|
{
|
|
org::xrpl::rpc::v1::GetLedgerResponse response;
|
|
bool success = execute(
|
|
[&response, ledgerSequence, getObjects, this](auto& source) {
|
|
auto [status, data] =
|
|
source->fetchLedger(ledgerSequence, getObjects);
|
|
response = std::move(data);
|
|
if (status.ok() && (response.validated()|| true))
|
|
{
|
|
BOOST_LOG_TRIVIAL(info)
|
|
<< "Successfully fetched ledger = " << ledgerSequence
|
|
<< " from source = " << source->toString();
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< "Error getting ledger = " << ledgerSequence
|
|
<< " Reply : " << response.DebugString()
|
|
<< " error_code : " << status.error_code()
|
|
<< " error_msg : " << status.error_message()
|
|
<< " source = " << source->toString();
|
|
return false;
|
|
}
|
|
},
|
|
ledgerSequence);
|
|
if (success)
|
|
return response;
|
|
else
|
|
return {};
|
|
}
|
|
|
|
/*
|
|
std::unique_ptr<org::xrpl::rpc::v1::XRPLedgerAPIService::Stub>
|
|
ETLLoadBalancer::getP2pForwardingStub() const
|
|
{
|
|
if (sources_.size() == 0)
|
|
return nullptr;
|
|
srand((unsigned)time(0));
|
|
auto sourceIdx = rand() % sources_.size();
|
|
auto numAttempts = 0;
|
|
while (numAttempts < sources_.size())
|
|
{
|
|
auto stub = sources_[sourceIdx]->getP2pForwardingStub();
|
|
if (!stub)
|
|
{
|
|
sourceIdx = (sourceIdx + 1) % sources_.size();
|
|
++numAttempts;
|
|
continue;
|
|
}
|
|
return stub;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
Json::Value
|
|
ETLLoadBalancer::forwardToP2p(RPC::JsonContext& context) const
|
|
{
|
|
Json::Value res;
|
|
if (sources_.size() == 0)
|
|
return res;
|
|
srand((unsigned)time(0));
|
|
auto sourceIdx = rand() % sources_.size();
|
|
auto numAttempts = 0;
|
|
while (numAttempts < sources_.size())
|
|
{
|
|
res = sources_[sourceIdx]->forwardToP2p(context);
|
|
if (!res.isMember("forwarded") || res["forwarded"] != true)
|
|
{
|
|
sourceIdx = (sourceIdx + 1) % sources_.size();
|
|
++numAttempts;
|
|
continue;
|
|
}
|
|
return res;
|
|
}
|
|
RPC::Status err = {rpcFAILED_TO_FORWARD};
|
|
err.inject(res);
|
|
return res;
|
|
}
|
|
|
|
std::unique_ptr<org::xrpl::rpc::v1::XRPLedgerAPIService::Stub>
|
|
ETLSource::getP2pForwardingStub() const
|
|
{
|
|
if (!connected_)
|
|
return nullptr;
|
|
try
|
|
{
|
|
return org::xrpl::rpc::v1::XRPLedgerAPIService::NewStub(
|
|
grpc::CreateChannel(
|
|
beast::IP::Endpoint(
|
|
boost::asio::ip::make_address(ip_), std::stoi(grpcPort_))
|
|
.to_string(),
|
|
grpc::InsecureChannelCredentials()));
|
|
}
|
|
catch (std::exception const&)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "Failed to create grpc stub";
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
Json::Value
|
|
ETLSource::forwardToP2p(RPC::JsonContext& context) const
|
|
{
|
|
BOOST_LOG_TRIVIAL(debug) << "Attempting to forward request to tx. "
|
|
<< "request = " << context.params.toStyledString();
|
|
|
|
Json::Value response;
|
|
if (!connected_)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error)
|
|
<< "Attempted to proxy but failed to connect to tx";
|
|
return response;
|
|
}
|
|
namespace beast = boost::beast; // from <boost/beast.hpp>
|
|
namespace http = beast::http; // from <boost/beast/http.hpp>
|
|
namespace websocket = beast::websocket; // from
|
|
<boost / beast / websocket.hpp> namespace net = boost::asio; // from
|
|
<boost / asio.hpp> using tcp = boost::asio::ip::tcp; // from
|
|
<boost / asio / ip / tcp.hpp> Json::Value& request = context.params;
|
|
try
|
|
{
|
|
// The io_context is required for all I/O
|
|
net::io_context ioc;
|
|
|
|
// These objects perform our I/O
|
|
tcp::resolver resolver{ioc};
|
|
|
|
BOOST_LOG_TRIVIAL(debug) << "Creating websocket";
|
|
auto ws = std::make_unique<websocket::stream<tcp::socket>>(ioc);
|
|
|
|
// Look up the domain name
|
|
auto const results = resolver.resolve(ip_, wsPort_);
|
|
|
|
BOOST_LOG_TRIVIAL(debug) << "Connecting websocket";
|
|
// Make the connection on the IP address we get from a lookup
|
|
net::connect(ws->next_layer(), results.begin(), results.end());
|
|
|
|
// Set a decorator to change the User-Agent of the handshake
|
|
// and to tell rippled to charge the client IP for RPC
|
|
// resources. See "secure_gateway" in
|
|
//
|
|
https: // github.com/ripple/rippled/blob/develop/cfg/rippled-example.cfg
|
|
ws->set_option(websocket::stream_base::decorator(
|
|
[&context](websocket::request_type& req) {
|
|
req.set(
|
|
http::field::user_agent,
|
|
std::string(BOOST_BEAST_VERSION_STRING) +
|
|
" websocket-client-coro");
|
|
req.set(
|
|
http::field::forwarded,
|
|
"for=" + context.consumer.to_string());
|
|
}));
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< "client ip: " << context.consumer.to_string();
|
|
|
|
BOOST_LOG_TRIVIAL(debug) << "Performing websocket handshake";
|
|
// Perform the websocket handshake
|
|
ws->handshake(ip_, "/");
|
|
|
|
Json::FastWriter fastWriter;
|
|
|
|
BOOST_LOG_TRIVIAL(debug) << "Sending request";
|
|
// Send the message
|
|
ws->write(net::buffer(fastWriter.write(request)));
|
|
|
|
beast::flat_buffer buffer;
|
|
ws->read(buffer);
|
|
|
|
Json::Reader reader;
|
|
if (!reader.parse(
|
|
static_cast<char const*>(buffer.data().data()), response))
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "Error parsing response";
|
|
response[jss::error] = "Error parsing response from tx";
|
|
}
|
|
BOOST_LOG_TRIVIAL(debug) << "Successfully forward request";
|
|
|
|
response["forwarded"] = true;
|
|
return response;
|
|
}
|
|
catch (std::exception const& e)
|
|
{
|
|
BOOST_LOG_TRIVIAL(error) << "Encountered exception : " << e.what();
|
|
return response;
|
|
}
|
|
}
|
|
*/
|
|
template <class Func>
|
|
bool
|
|
ETLLoadBalancer::execute(Func f, uint32_t ledgerSequence)
|
|
{
|
|
srand((unsigned)time(0));
|
|
auto sourceIdx = rand() % sources_.size();
|
|
auto numAttempts = 0;
|
|
|
|
while (true)
|
|
{
|
|
auto& source = sources_[sourceIdx];
|
|
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< __func__ << " : "
|
|
<< "Attempting to execute func. ledger sequence = "
|
|
<< ledgerSequence << " - source = " << source->toString();
|
|
if (source->hasLedger(ledgerSequence)|| true)
|
|
{
|
|
bool res = f(source);
|
|
if (res)
|
|
{
|
|
BOOST_LOG_TRIVIAL(debug)
|
|
<< __func__ << " : "
|
|
<< "Successfully executed func at source = "
|
|
<< source->toString()
|
|
<< " - ledger sequence = " << ledgerSequence;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< __func__ << " : "
|
|
<< "Failed to execute func at source = "
|
|
<< source->toString()
|
|
<< " - ledger sequence = " << ledgerSequence;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< __func__ << " : "
|
|
<< "Ledger not present at source = " << source->toString()
|
|
<< " - ledger sequence = " << ledgerSequence;
|
|
}
|
|
sourceIdx = (sourceIdx + 1) % sources_.size();
|
|
numAttempts++;
|
|
if (numAttempts % sources_.size() == 0)
|
|
{
|
|
/*
|
|
if (etl_.getApplication().getLedgerMaster().getLedgerBySeq(
|
|
ledgerSequence))
|
|
{
|
|
BOOST_LOG_TRIVIAL(warning)
|
|
<< __func__ << " : "
|
|
<< "Error executing function. "
|
|
<< " Tried all sources, but ledger was found in db."
|
|
<< " Sequence = " << ledgerSequence;
|
|
break;
|
|
}
|
|
*/
|
|
BOOST_LOG_TRIVIAL(error)
|
|
<< __func__ << " : "
|
|
<< "Error executing function "
|
|
<< " - ledger sequence = " << ledgerSequence
|
|
<< " - Tried all sources. Sleeping and trying again";
|
|
std::this_thread::sleep_for(std::chrono::seconds(2));
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
void
|
|
ETLLoadBalancer::start()
|
|
{
|
|
for (auto& source : sources_)
|
|
source->start();
|
|
}
|
|
|
|
void
|
|
ETLLoadBalancer::stop()
|
|
{
|
|
for (auto& source : sources_)
|
|
source->stop();
|
|
}
|