initial commit

validator/shard-client.hpp

@@ -0,0 +1,95 @@
+/*
+    This file is part of TON Blockchain Library.
+
+    TON Blockchain Library is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, either version 2 of the License, or
+    (at your option) any later version.
+
+    TON Blockchain Library is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with TON Blockchain Library.  If not, see <http://www.gnu.org/licenses/>.
+
+    Copyright 2017-2019 Telegram Systems LLP
+*/
+#pragma once
+
+#include "interfaces/validator-manager.h"
+#include <set>
+
+namespace ton {
+
+namespace validator {
+
+class ShardClient : public td::actor::Actor {
+ private:
+  td::Ref<ValidatorManagerOptions> opts_;
+
+  BlockHandle masterchain_block_handle_;
+  td::Ref<MasterchainState> masterchain_state_;
+
+  std::vector<td::actor::ActorOwn<ShardClient>> children_;
+
+  bool waiting_ = false;
+  bool init_mode_ = false;
+
+  td::actor::ActorId<ValidatorManager> manager_;
+
+  td::Promise<td::Unit> promise_;
+
+  std::set<ShardIdFull> created_overlays_;
+
+ public:
+  ShardClient(td::Ref<ValidatorManagerOptions> opts, BlockHandle masterchain_block_handle,
+              td::Ref<MasterchainState> masterchain_state, td::actor::ActorId<ValidatorManager> manager,
+              td::Promise<td::Unit> promise)
+      : opts_(std::move(opts))
+      , masterchain_block_handle_(masterchain_block_handle)
+      , masterchain_state_(std::move(masterchain_state))
+      , manager_(manager)
+      , promise_(std::move(promise)) {
+    init_mode_ = true;
+  }
+  ShardClient(td::Ref<ValidatorManagerOptions> opts, td::actor::ActorId<ValidatorManager> manager)
+      : opts_(std::move(opts)), manager_(manager) {
+  }
+
+  static constexpr td::uint32 shard_client_priority() {
+    return 2;
+  }
+
+  void build_shard_overlays();
+
+  void start_up() override;
+  void start_up_init_mode();
+  void start_up_init_mode_finished();
+  void got_state_from_db(BlockIdExt masterchain_block_id);
+
+  void im_download_shard_state(BlockIdExt block_id, td::Promise<td::Unit> promise);
+  void im_downloaded_zero_state(BlockIdExt block_id, td::BufferSlice data, td::Promise<td::Unit> promise);
+  void im_downloaded_proof_link(BlockIdExt block_id, td::BufferSlice data, td::Promise<td::Unit> promise);
+  void im_checked_proof_link(BlockIdExt block_id, td::Promise<td::Unit> promise);
+  void im_downloaded_shard_state(BlockIdExt block_id, td::Promise<td::Unit> promise);
+  void im_got_shard_handle(BlockHandle handle, td::Promise<td::Unit> promise);
+
+  void new_masterchain_block_id(BlockIdExt masterchain_block_id);
+  void got_masterchain_block_handle(BlockHandle handle);
+  void download_masterchain_state();
+  void got_masterchain_block_state(td::Ref<MasterchainState> state);
+  void apply_all_shards();
+  void downloaded_shard_state(td::Ref<ShardState> state, td::Promise<td::Unit> promise);
+  void applied_all_shards();
+  void saved_to_db();
+
+  void new_masterchain_block_notification(BlockHandle handle, td::Ref<MasterchainState> state);
+
+  void get_processed_masterchain_block(td::Promise<BlockSeqno> promise);
+};
+
+}  // namespace validator
+
+}  // namespace ton