external/ton
1
0
Fork 0
mirror of https://github.com/ton-blockchain/ton synced 2025-03-09 15:40:10 +00:00
Code Issues Releases Wiki Activity

Fix QueueSizeCounter and ValidatorGroup after merge

Browse source
This commit is contained in:
SpyCheese 2024-02-19 13:10:47 +03:00
parent a2eb3f3631
commit 747b24aa18
8 changed files with 31 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

4
validator/full-node.cpp
View file

@ -654,7 +654,7 @@ void FullNodeImpl::update_private_block_overlays() {
} }
void FullNodeImpl::create_private_block_overlay(PublicKeyHash key) { void FullNodeImpl::create_private_block_overlay(PublicKeyHash key) {
CHECK(local_keys_.count(key)); /*CHECK(local_keys_.count(key));
if (current_validators_.count(key)) { if (current_validators_.count(key)) {
std::vector<adnl::AdnlNodeIdShort> nodes; std::vector<adnl::AdnlNodeIdShort> nodes;
for (const auto &p : current_validators_) { for (const auto &p : current_validators_) {
@ -663,7 +663,7 @@ void FullNodeImpl::create_private_block_overlay(PublicKeyHash key) {
private_block_overlays_[key] = td::actor::create_actor<FullNodePrivateOverlay>( private_block_overlays_[key] = td::actor::create_actor<FullNodePrivateOverlay>(
"BlocksPrivateOverlay", current_validators_[key], std::move(nodes), zero_state_file_hash_, config_, keyring_, "BlocksPrivateOverlay", current_validators_[key], std::move(nodes), zero_state_file_hash_, config_, keyring_,
adnl_, rldp_, rldp2_, overlays_, validator_manager_); adnl_, rldp_, rldp2_, overlays_, validator_manager_);
} }*/
} }
FullNodeImpl::FullNodeImpl(PublicKeyHash local_id, adnl::AdnlNodeIdShort adnl_id, FileHash zero_state_file_hash, FullNodeImpl::FullNodeImpl(PublicKeyHash local_id, adnl::AdnlNodeIdShort adnl_id, FileHash zero_state_file_hash,

4
validator/manager-disk.hpp
View file

@ -388,8 +388,8 @@ class ValidatorManagerImpl : public ValidatorManager {
} }
void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override { void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override {
if (queue_size_counter_.empty()) { if (queue_size_counter_.empty()) {
queue_size_counter_ = queue_size_counter_ = td::actor::create_actor<QueueSizeCounter>("queuesizecounter", td::Ref<MasterchainState>{},
td::actor::create_actor<QueueSizeCounter>("queuesizecounter", td::Ref<MasterchainState>{}, actor_id(this)); opts_, actor_id(this));
} }
td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise)); td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise));
} }

4
validator/manager-hardfork.hpp
View file

@ -450,8 +450,8 @@ class ValidatorManagerImpl : public ValidatorManager {
} }
void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override { void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override {
if (queue_size_counter_.empty()) { if (queue_size_counter_.empty()) {
queue_size_counter_ = queue_size_counter_ = td::actor::create_actor<QueueSizeCounter>("queuesizecounter", td::Ref<MasterchainState>{},
td::actor::create_actor<QueueSizeCounter>("queuesizecounter", td::Ref<MasterchainState>{}, actor_id(this)); opts_, actor_id(this));
} }
td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise)); td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise));
} }

3
validator/manager.cpp
View file

@ -3115,6 +3115,9 @@ void ValidatorManagerImpl::update_options(td::Ref<ValidatorManagerOptions> opts)
if (!out_msg_queue_importer_.empty()) { if (!out_msg_queue_importer_.empty()) {
td::actor::send_closure(out_msg_queue_importer_, &OutMsgQueueImporter::update_options, opts); td::actor::send_closure(out_msg_queue_importer_, &OutMsgQueueImporter::update_options, opts);
} }
if (!queue_size_counter_.empty()) {
td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::update_options, opts);
}
opts_ = std::move(opts); opts_ = std::move(opts);
} }

16
validator/manager.hpp
View file

@ -591,13 +591,17 @@ class ValidatorManagerImpl : public ValidatorManager {
void update_options(td::Ref<ValidatorManagerOptions> opts) override; void update_options(td::Ref<ValidatorManagerOptions> opts) override;
void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override { void get_out_msg_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise) override {
if (last_masterchain_state_.is_null()) {
promise.set_error(td::Status::Error(ErrorCode::notready, "not ready"));
return;
}
if (queue_size_counter_.empty()) { if (queue_size_counter_.empty()) {
if (last_masterchain_state_.is_null()) { queue_size_counter_ =
promise.set_error(td::Status::Error(ErrorCode::notready, "not ready")); td::actor::create_actor<QueueSizeCounter>("queuesizecounter", last_masterchain_state_, opts_, actor_id(this));
return; }
} if (!opts_->need_monitor(block_id.shard_full(), last_masterchain_state_)) {
queue_size_counter_ = td::actor::create_actor<QueueSizeCounter>("queuesizecounter", return promise.set_error(
last_masterchain_state_, actor_id(this)); td::Status::Error(PSTRING() << "not monitoring shard " << block_id.shard_full().to_str()));
} }
td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise)); td::actor::send_closure(queue_size_counter_, &QueueSizeCounter::get_queue_size, block_id, std::move(promise));
} }

4
validator/queue-size-counter.cpp
View file

@ -234,7 +234,9 @@ void QueueSizeCounter::process_top_shard_blocks_cont(td::Ref<MasterchainState> s
last_top_blocks_.clear(); last_top_blocks_.clear();
last_top_blocks_.push_back(state->get_block_id()); last_top_blocks_.push_back(state->get_block_id());
for (auto &shard : state->get_shards()) { for (auto &shard : state->get_shards()) {
last_top_blocks_.push_back(shard->top_block_id()); if (opts_->need_monitor(shard->shard(), state)) {
last_top_blocks_.push_back(shard->top_block_id());
}
} }
for (const BlockIdExt &block_id : last_top_blocks_) { for (const BlockIdExt &block_id : last_top_blocks_) {
get_queue_size_ex_retry(block_id, init, ig.get_promise()); get_queue_size_ex_retry(block_id, init, ig.get_promise());

10
validator/queue-size-counter.hpp
View file

@ -21,16 +21,22 @@ namespace ton::validator {
class QueueSizeCounter : public td::actor::Actor { class QueueSizeCounter : public td::actor::Actor {
public: public:
QueueSizeCounter(td::Ref<MasterchainState> last_masterchain_state, td::actor::ActorId<ValidatorManager> manager) QueueSizeCounter(td::Ref<MasterchainState> last_masterchain_state, td::Ref<ValidatorManagerOptions> opts,
: init_masterchain_state_(last_masterchain_state), manager_(std::move(manager)) { td::actor::ActorId<ValidatorManager> manager)
: init_masterchain_state_(last_masterchain_state), opts_(std::move(opts)), manager_(std::move(manager)) {
} }
void start_up() override; void start_up() override;
void get_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise); void get_queue_size(BlockIdExt block_id, td::Promise<td::uint32> promise);
void alarm() override; void alarm() override;
void update_options(td::Ref<ValidatorManagerOptions> opts) {
opts_ = std::move(opts);
}
private: private:
td::Ref<MasterchainState> init_masterchain_state_; td::Ref<MasterchainState> init_masterchain_state_;
td::Ref<ValidatorManagerOptions> opts_;
td::actor::ActorId<ValidatorManager> manager_; td::actor::ActorId<ValidatorManager> manager_;
bool simple_mode_ = false; bool simple_mode_ = false;

2
validator/validator-group.cpp
View file

@ -195,7 +195,7 @@ void ValidatorGroup::accept_block_query(BlockIdExt block_id, td::Ref<BlockData>
} }
}); });
run_accept_block_query(block_id, std::move(block), prev_block_ids_, validator_set_, std::move(sig_set), run_accept_block_query(block_id, std::move(block), std::move(prev), validator_set_, std::move(sig_set),
std::move(approve_sig_set), send_broadcast, std::move(approve_sig_set), send_broadcast,
shard_.is_masterchain() || mode_ == ValidatorManagerOptions::validator_normal, manager_, shard_.is_masterchain() || mode_ == ValidatorManagerOptions::validator_normal, manager_,
std::move(P)); std::move(P));