feat(rpc): gw_get_mem_pool_state_ready

crates/benches/benches/benchmarks/smt.rs

@@ -156,7 +156,7 @@ impl BenchExecutionEnvironment {
  let generator = Generator::new(backend_manage, account_lock_manage, rollup_context);
 
  Self::init_genesis(&store, &genesis_config, accounts);
- let mem_pool_state = MemPoolState::new(Arc::new(MemStore::new(store.get_snapshot())));
+ let mem_pool_state = MemPoolState::new(Arc::new(MemStore::new(store.get_snapshot())), true);
 
  BenchExecutionEnvironment {
  generator,

crates/chain/src/chain.rs

@@ -257,7 +257,9 @@ impl Chain {
 
  log::debug!("[complete_initial_syncing] acquire mem-pool",);
  let t = Instant::now();
- mem_pool.lock().await.notify_new_tip(tip_block_hash).await?;
+ let mut mem_pool = mem_pool.lock().await;
+ mem_pool.notify_new_tip(tip_block_hash).await?;
+ mem_pool.mem_pool_state().set_complete_initial_syncing();
  log::debug!(
  "[complete_initial_syncing] unlock mem-pool {}ms",
  t.elapsed().as_millis()

crates/mem-pool/src/pool.rs

@@ -174,7 +174,7 @@ impl MemPool {
 
  let mem_pool_state = {
  let mem_store = MemStore::new(store.get_snapshot());
- Arc::new(MemPoolState::new(Arc::new(mem_store)))
+ Arc::new(MemPoolState::new(Arc::new(mem_store), false))
  };
 
  let mut mem_pool = MemPool {

crates/rpc-server/src/registry.rs

@@ -183,9 +183,10 @@ impl Registry {
  let mem_pool = pool.lock().await;
  mem_pool.mem_pool_state()
  }
- None => Arc::new(MemPoolState::new(Arc::new(MemStore::new(
- store.get_snapshot(),
- )))),
+ None => Arc::new(MemPoolState::new(
+ Arc::new(MemStore::new(store.get_snapshot())),
+ true,
+ )),
  };
  let (submit_tx, submit_rx) = async_channel::bounded(RequestSubmitter::MAX_CHANNEL_SIZE);
  if let Some(mem_pool) = mem_pool.as_ref().to_owned() {
@@ -280,6 +281,7 @@ impl Registry {
  )
  .with_method("gw_get_fee_config", get_fee_config)
  .with_method("gw_get_mem_pool_state_root", get_mem_pool_state_root)
+ .with_method("gw_get_mem_pool_state_ready", get_mem_pool_state_ready)
  .with_method("gw_get_node_info", get_node_info)
  .with_method("gw_reload_config", reload_config);
 
@@ -1365,6 +1367,12 @@ async fn get_mem_pool_state_root(
  Ok(to_jsonh256(root))
 }
 
+async fn get_mem_pool_state_ready(
+ mem_pool_state: Data<Arc<MemPoolState>>,
+) -> Result<bool, RpcError> {
+ Ok(mem_pool_state.complete_initial_syncing())
+}
+
 async fn tests_produce_block(
  Params((payload,)): Params<(TestModePayload,)>,
  tests_rpc_impl: Data<BoxedTestsRPCImpl>,

crates/store/src/mem_pool_state.rs

@@ -1,4 +1,7 @@
-use std::sync::Arc;
+use std::sync::{
+ atomic::{AtomicBool, Ordering},
+ Arc,
+};
 
 use crate::snapshot::StoreSnapshot;
 use arc_swap::{ArcSwap, Guard};
@@ -11,7 +14,7 @@ use gw_db::{
  schema::{Col, COLUMNS, COLUMN_ACCOUNT_SMT_BRANCH, COLUMN_ACCOUNT_SMT_LEAF, COLUMN_META},
 };
 use gw_types::{
- packed::{self},
+ packed,
  prelude::{Entity, FromSliceShouldBeOk, Pack, Reader, Unpack},
 };
 
@@ -32,12 +35,14 @@ pub const META_MEM_SMT_COUNT_KEY: &[u8] = b"MEM_ACCOUNT_SMT_COUNT_KEY";
 
 pub struct MemPoolState {
  store: ArcSwap<MemStore>,
+ complete_initial_syncing: AtomicBool,
 }
 
 impl MemPoolState {
- pub fn new(store: Arc<MemStore>) -> Self {
+ pub fn new(store: Arc<MemStore>, complete_initial_syncing: bool) -> Self {
  Self {
  store: ArcSwap::new(store),
+ complete_initial_syncing: AtomicBool::new(complete_initial_syncing),
  }
  }
 
@@ -50,6 +55,14 @@ impl MemPoolState {
  pub fn store(&self, mem_store: Arc<MemStore>) {
  self.store.store(mem_store);
  }
+
+ pub fn complete_initial_syncing(&self) -> bool {
+ self.complete_initial_syncing.load(Ordering::SeqCst)
+ }
+
+ pub fn set_complete_initial_syncing(&self) {
+ self.complete_initial_syncing.store(true, Ordering::SeqCst);
+ }
 }
 
 enum Value<T> {