Merge

near · Apr 13, 2023 · 0855002 · 0855002
1 parent 00f70d4
commit 0855002
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/chain/client/src/client.rs b/chain/client/src/client.rs
@@ -32,6 +32,7 @@ use near_chunks::logic::{
 use near_chunks::ShardsManager;
 use near_client_primitives::debug::ChunkProduction;
 use near_client_primitives::types::{Error, ShardSyncDownload, ShardSyncStatus};
+use near_epoch_manager::shard_tracker::ShardTracker;
 use near_network::types::{AccountKeys, ChainInfo, PeerManagerMessageRequest, SetChainInfo};
 use near_network::types::{
     HighestHeightPeerInfo, NetworkRequests, PeerManagerAdapter, ReasonForBan,
@@ -359,7 +360,7 @@ impl Client {
                     block.header().prev_hash(),
                     shard_id,
                     false,
-                    self.runtime_adapter.as_ref(),
+                    &self.shard_tracker,
                 ) {
                     self.sharded_tx_pool.reintroduce_transactions(
                         shard_id,
@@ -1596,7 +1597,8 @@ impl Client {
             &encoded_chunk,
             merkle_paths.clone(),
             Some(&validator_id),
-            self.runtime_adapter.as_ref(),
+            self.runtime_adapter.epoch_manager_adapter(),
+            &self.shard_tracker,
         )?;
         persist_chunk(partial_chunk.clone(), Some(shard_chunk), self.chain.mut_store())?;
         self.on_chunk_header_ready_for_inclusion(encoded_chunk.cloned_header(), validator_id);

diff --git a/chain/client/src/client_actor.rs b/chain/client/src/client_actor.rs
@@ -1657,7 +1657,7 @@ impl ClientActor {
                                     &prev_hash,
                                     *x,
                                     true,
-                                    self.client.runtime_adapter.as_ref(),
+                                    &self.client.shard_tracker,
                                 )
                             })
                             .collect();