Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

rpcdaemon: ots_searchTransactionsBefore for e3 #2520

Merged
merged 6 commits into from
Nov 21, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/workflows/rpc-integration-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,9 @@ jobs:
- name: Checkout RPC Tests Repository & Install Requirements
run: |
rm -rf ${{runner.workspace}}/rpc-tests
git -c advice.detachedHead=false clone --depth 1 --branch v1.15.0 https://github.com/erigontech/rpc-tests ${{runner.workspace}}/rpc-tests
git -c advice.detachedHead=false clone --depth 1 --branch v1.16.1 https://github.com/erigontech/rpc-tests ${{runner.workspace}}/rpc-tests
cd ${{runner.workspace}}/rpc-tests
pip3 install -r requirements.txt
pip3 install -r requirements.txt --break-system-packages

- name: Clean Build Directory
run: rm -rf ${{runner.workspace}}/silkworm/build
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/rpc-performance-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ jobs:
rm -rf ${{runner.workspace}}/rpc-tests
git -c advice.detachedHead=false clone --depth 1 --branch v0.26.0 https://github.com/erigontech/rpc-tests ${{runner.workspace}}/rpc-tests
cd ${{runner.workspace}}/rpc-tests
pip3 install -r requirements.txt
pip3 install -r requirements.txt --break-system-packages

- name: Clean Build Directory
run: rm -rf ${{runner.workspace}}/silkworm/build
Expand Down
3 changes: 1 addition & 2 deletions .github/workflows/run_integration_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -50,8 +50,7 @@ engine_,\
erigon_getLatestLogs,\
eth_getLogs,\
ots_getContractCreator,\
ots_searchTransactionsAfter,\
ots_searchTransactionsBefore
ots_searchTransactionsAfter

failed_test=$?

Expand Down
2 changes: 2 additions & 0 deletions silkworm/db/tables.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -440,5 +440,7 @@ inline constexpr const char* kStorageHistoryIdx{"StorageHistoryIdx"};
inline constexpr const char* kCodeHistoryIdx{"CodeHistoryIdx"};
inline constexpr const char* kCommitmentHistoryIdx{"CommitmentHistoryIdx"};
inline constexpr const char* kReceiptHistoryIdx{"ReceiptHistoryIdx"};
inline constexpr const char* kTracesFromIdx{"TracesFromIdx"};
inline constexpr const char* kTracesToIdx{"TracesToIdx"};

} // namespace silkworm::db::table
169 changes: 129 additions & 40 deletions silkworm/rpc/commands/ots_api.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -681,6 +681,11 @@ Task<void> OtsRpcApi::handle_ots_get_internal_operations(const nlohmann::json& r
co_await tx->close(); // RAII not (yet) available with coroutines
}

struct BlockInfo {
uint64_t number{0};
BlockDetails details;
};

Task<void> OtsRpcApi::handle_ots_search_transactions_before(const nlohmann::json& request, nlohmann::json& reply) {
const auto& params = request["params"];
if (params.size() != 3) {
Expand All @@ -694,7 +699,7 @@ Task<void> OtsRpcApi::handle_ots_search_transactions_before(const nlohmann::json
auto block_number = params[1].get<BlockNum>();
const auto page_size = params[2].get<uint64_t>();

SILK_DEBUG << "address: " << address << " block_number: " << block_number << " page_size: " << page_size;
SILK_DEBUG << "address: " << address << ", block_number: " << block_number << ", page_size: " << page_size;

if (page_size > kMaxPageSize) {
auto error_msg = "max allowed page size: " + std::to_string(kMaxPageSize);
Expand All @@ -703,52 +708,23 @@ Task<void> OtsRpcApi::handle_ots_search_transactions_before(const nlohmann::json
co_return;
}

if (block_number > 0) {
--block_number;
}
auto tx = co_await database_->begin();

try {
auto call_from_cursor = co_await tx->cursor(table::kCallFromIndexName);
auto call_to_cursor = co_await tx->cursor(table::kCallToIndexName);

bool is_first_page = false;
auto provider = ethdb::kv::canonical_body_for_storage_provider(backend_);

if (block_number == 0) {
is_first_page = true;
} else {
// Internal search code considers blockNum [including], so adjust the value
--block_number;
db::kv::api::Timestamp from_timestamp{-1};
if (block_number > 0) {
const auto max_tx_num = co_await db::txn::max_tx_num(*tx, block_number, provider);
from_timestamp = static_cast<db::kv::api::Timestamp>(max_tx_num);
SILK_DEBUG << "block_number: " << block_number << " max_tx_num: " << max_tx_num;
}

BackwardBlockProvider from_provider{call_from_cursor.get(), address, block_number};
BackwardBlockProvider to_provider{call_to_cursor.get(), address, block_number};
FromToBlockProvider from_to_provider{false, &from_provider, &to_provider};

uint64_t result_count = 0;
bool has_more = true;

TransactionsWithReceipts results{
.first_page = is_first_page};

while (result_count < page_size && has_more) {
std::vector<TransactionsWithReceipts> transactions_with_receipts_vec;

has_more = co_await trace_blocks(from_to_provider, *tx, address, page_size, result_count, transactions_with_receipts_vec);

for (const auto& item : transactions_with_receipts_vec) {
results.receipts.insert(results.receipts.end(), item.receipts.rbegin(), item.receipts.rend());
results.transactions.insert(results.transactions.end(), item.transactions.rbegin(), item.transactions.rend());
results.blocks.insert(results.blocks.end(), item.blocks.rbegin(), item.blocks.rend());

result_count += item.transactions.size();

if (result_count >= page_size) {
break;
}
}
}
const auto results = co_await collect_transactions_with_receipts(*tx, provider, block_number, address, from_timestamp, false, page_size);

results.last_page = !has_more;
reply = make_json_content(request, results);

} catch (const std::invalid_argument& iv) {
SILK_WARN << "invalid_argument: " << iv.what() << " processing request: " << request.dump();
reply = make_json_content(request, nlohmann::detail::value_t::null);
Expand Down Expand Up @@ -850,6 +826,119 @@ Task<void> OtsRpcApi::handle_ots_search_transactions_after(const nlohmann::json&
co_await tx->close(); // RAII not (yet) available with coroutines
}

Task<TransactionsWithReceipts> OtsRpcApi::collect_transactions_with_receipts(
kv::api::Transaction& tx,
db::chain::CanonicalBodyForStorageProvider& provider,
BlockNum block_number,
const evmc::address& address,
db::kv::api::Timestamp from_timestamp,
bool ascending, uint64_t page_size) {
const auto key = db::code_domain_key(address);
db::kv::api::IndexRangeQuery query_to{
.table = db::table::kTracesToIdx,
.key = key,
.from_timestamp = from_timestamp,
.to_timestamp = -1,
.ascending_order = ascending};
auto paginated_result_to = co_await tx.index_range(std::move(query_to));

db::kv::api::IndexRangeQuery query_from{
.table = db::table::kTracesFromIdx,
.key = key,
.from_timestamp = from_timestamp,
.to_timestamp = -1,
.ascending_order = ascending};
auto paginated_result_from = co_await tx.index_range(std::move(query_from));

auto it_from = co_await paginated_result_from.begin();
auto it_to = co_await paginated_result_to.begin();

TransactionsWithReceipts results{
.first_page = block_number == 0,
.last_page = true};

std::map<std::string, Receipt> receipts;
std::optional<BlockInfo> block_info;
auto block_num_changed = false;

auto it = db::kv::api::set_union(it_from, it_to, ascending);
const auto chain_storage = tx.create_storage();

while (const auto value = co_await it.next()) {
const auto txn_id = static_cast<TxnId>(*value);
const auto block_number_opt = co_await db::txn::block_num_from_tx_num(tx, txn_id, provider);
if (!block_number_opt) {
SILK_DEBUG << "No block found for txn_id " << txn_id;
break;
}
const auto bn = block_number_opt.value();
const auto max_txn_id = co_await db::txn::max_tx_num(tx, bn, provider);
const auto min_txn_id = co_await db::txn::min_tx_num(tx, bn, provider);
const auto txn_index = txn_id - min_txn_id - 1;
SILK_DEBUG
<< "txn_id: " << txn_id
<< " block_number: " << bn
<< ", txn_index: " << txn_index
<< ", max_txn_id: " << max_txn_id
<< ", min_txn_id: " << min_txn_id
<< ", final txn: " << (txn_id == max_txn_id)
<< ", ascending: " << ascending;

if (txn_id == max_txn_id) {
continue;
}

block_num_changed = false;
if (block_info && block_info.value().number != bn) {
block_info.reset();
}

if (!block_info) {
const auto block_with_hash = co_await rpc::core::read_block_by_number(*block_cache_, *chain_storage, bn);
if (!block_with_hash) {
SILK_DEBUG << "Not found block no. " << bn;
co_return results;
}

auto rr = co_await core::get_receipts(tx, *block_with_hash, *chain_storage, workers_);
SILK_DEBUG << "Read #" << rr.size() << " receipts from block " << bn;

std::for_each(rr.begin(), rr.end(), [&receipts](const auto& item) {
receipts[silkworm::to_hex(item.tx_hash, false)] = std::move(item);
});

const Block extended_block{block_with_hash, false};
const auto block_size = extended_block.get_block_size();
const BlockDetails block_details{block_size, block_with_hash->hash, block_with_hash->block.header,
block_with_hash->block.transactions.size(), block_with_hash->block.ommers,
block_with_hash->block.withdrawals};
block_info = BlockInfo{block_with_hash->block.header.number, block_details};
block_num_changed = true;
}

if (results.transactions.size() >= page_size && block_num_changed) {
results.last_page = false;
break;
}

auto transaction = co_await chain_storage->read_transaction_by_idx_in_block(bn, txn_index);
if (!transaction) {
SILK_DEBUG << "No transaction found in block " << bn << " for index " << txn_index;
co_return results;
}
results.receipts.push_back(std::move(receipts.at(silkworm::to_hex(transaction.value().hash(), false))));
results.transactions.push_back(std::move(transaction.value()));
results.blocks.push_back(block_info.value().details);
}

SILK_DEBUG << "Results"
<< " transactions size: " << results.transactions.size()
<< " receipts size: " << results.receipts.size()
<< " block details size: " << results.blocks.size();

co_return results;
}

Task<bool> OtsRpcApi::trace_blocks(
FromToBlockProvider& from_to_provider,
kv::api::Transaction& tx,
Expand Down
11 changes: 11 additions & 0 deletions silkworm/rpc/commands/ots_api.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,12 @@
#include <silkworm/core/common/base.hpp>
#include <silkworm/core/common/block_cache.hpp>
#include <silkworm/core/common/bytes.hpp>
#include <silkworm/db/chain/providers.hpp>
#include <silkworm/db/datastore/mdbx/bitmap.hpp>
#include <silkworm/db/kv/api/cursor.hpp>
#include <silkworm/db/kv/api/endpoint/key_value.hpp>
#include <silkworm/db/kv/api/state_cache.hpp>
#include <silkworm/db/kv/api/transaction.hpp>
#include <silkworm/infra/concurrency/private_service.hpp>
#include <silkworm/infra/concurrency/shared_service.hpp>
#include <silkworm/rpc/common/worker_pool.hpp>
Expand Down Expand Up @@ -228,6 +230,15 @@ class OtsRpcApi {
friend class silkworm::rpc::json_rpc::RequestHandler;

private:
Task<TransactionsWithReceipts> collect_transactions_with_receipts(
db::kv::api::Transaction& tx,
db::chain::CanonicalBodyForStorageProvider& provider,
BlockNum block_number,
const evmc::address& address,
db::kv::api::Timestamp from_timestamp,
bool ascending,
uint64_t page_size);

Task<bool> trace_blocks(
FromToBlockProvider& from_to_provider,
db::kv::api::Transaction& tx,
Expand Down
Loading