add set concurrency level and get concurrency level, and

get height blocks

Author: nkysg

chain/api/src/chain.rs

@@ -43,6 +43,10 @@ pub trait ChainReader {
         reverse: bool,
         count: u64,
     ) -> Result<Vec<Block>>;
+    fn get_num_blocks(
+        &self,
+        number: BlockNumber
+    ) -> Result<Vec<Block>>;
     fn get_block(&self, hash: HashValue) -> Result<Option<Block>>;
     /// Get block hash by block number, if not exist, return None
     fn get_hash_by_number(&self, number: BlockNumber) -> Result<Option<HashValue>>;

chain/src/chain.rs

@@ -605,6 +605,11 @@ impl ChainReader for BlockChain {
         Ok(blocks)
     }
 
+    fn get_num_blocks(&self, number: BlockNumber) -> Result<Vec<Block>> {
+        // XXX FIXME YSG
+        todo!()
+    }
+
     fn get_block(&self, hash: HashValue) -> Result<Option<Block>> {
         self.storage
             .get_block_by_hash(hash)

cmd/starcoin/src/chain/get_num_blocks_cmd.rs

@@ -0,0 +1,51 @@
+// Copyright (c) The Starcoin Core Contributors
+// SPDX-License-Identifier: Apache-2.0
+
+use crate::cli_state::CliState;
+use crate::StarcoinOpt;
+use anyhow::Result;
+use clap::Parser;
+use scmd::{CommandAction, ExecContext};
+use starcoin_crypto::HashValue;
+use starcoin_rpc_api::chain::GetBlockOption;
+use starcoin_rpc_api::types::BlockView;
+use std::str::FromStr;
+use starcoin_types::block::BlockNumber;
+
+/// Get block by hash or number.
+#[derive(Debug, Parser)]
+#[clap(name = "get-block", alias = "get_block")]
+pub struct GetNumBlocksOpt {
+    #[clap(name = "number")]
+    number: BlockNumber,
+
+    #[clap(name = "contains-raw-block")]
+    raw: bool,
+}
+
+pub struct GetNumBlocksCommand;
+
+impl CommandAction for GetNumBlocksCommand {
+    type State = CliState;
+    type GlobalOpt = StarcoinOpt;
+    type Opt = GetNumBlocksOpt;
+    type ReturnItem = Vec<BlockView>;
+
+    fn run(
+        &self,
+        ctx: &ExecContext<Self::State, Self::GlobalOpt, Self::Opt>,
+    ) -> Result<Self::ReturnItem> {
+        let client = ctx.state().client();
+        let opt = ctx.opt();
+        let blocks = client
+                .chain_get_num_blocks(
+                    number,
+                    Some(GetBlockOption {
+                        decode: true,
+                        raw: opt.raw,
+                    }),
+                )?
+                .ok_or_else(|| anyhow::format_err!("block of height {} not found", number))?;
+        Ok(blocks)
+    }
+}

cmd/starcoin/src/chain/mod.rs

@@ -3,6 +3,7 @@
 
 mod epoch_info;
 mod get_block_cmd;
+mod get_num_blocks_cmd;
 mod get_block_info_cmd;
 mod get_events_cmd;
 mod get_txn_cmd;

rpc/api/src/chain/mod.rs

@@ -47,6 +47,14 @@ pub trait ChainApi {
         count: u64,
         option: Option<GetBlocksOption>,
     ) -> FutureResult<Vec<BlockView>>;
+
+    /// Get height `number` blocks.
+    #[rpc(name = "chain.get_num_blocks")]
+    fn get_num_blocks(
+        &self,
+        number: BlockNumber,
+        option: Option<GetBlocksOption>,
+    ) -> FutureResult<Vec<BlockView>>;
     #[rpc(name = "chain.get_block_info_by_number")]
     fn get_block_info_by_number(&self, number: BlockNumber) -> FutureResult<Option<BlockInfoView>>;
     /// Get chain transactions

rpc/client/src/lib.rs

@@ -799,6 +799,14 @@ impl RpcClient {
         .map_err(map_err)
     }
 
+    pub fn chain_get_num_blocks(&self,
+    number: BlockNumber,
+    option: Option<GetBlocksOption>,) -> anyhow::Result<Vec<BlockView>> {
+        self.call_rpc_blocking(|inner| {
+            inner.chain_client.get_num_blocks(number, option)
+        }).map_err(map_err)
+    }
+
     pub fn chain_get_transaction(
         &self,
         txn_id: HashValue,

rpc/server/src/module/chain_rpc.rs

@@ -171,6 +171,11 @@ where
         Box::pin(fut.boxed())
     }
 
+    fn get_num_blocks(&self, number: BlockNumber, option: Option<GetBlocksOption>) -> FutureResult<Vec<BlockView>> {
+        // XXX FIXME YSG
+        todo!()
+    }
+
     fn get_block_info_by_number(&self, number: u64) -> FutureResult<Option<BlockInfoView>> {
         let service = self.service.clone();
 

vm/starcoin-transactional-test-harness/src/fork_chain.rs

@@ -307,6 +307,13 @@ impl ChainApi for MockChainApi {
         Box::pin(fut.boxed().map_err(map_err))
     }
 
+    fn get_num_blocks(&self, number: BlockNumber, option: Option<GetBlocksOption>) -> FutureResult<Vec<BlockView>> {
+        let fut = async move {
+            bail!("not implemented.");
+        };
+        Box::pin(fut.boxed().map_err(map_err))
+    }
+
     fn get_block_info_by_number(&self, number: BlockNumber) -> FutureResult<Option<BlockInfoView>> {
         let chain = self.chain.lock().unwrap();
         let client = chain.remote_chain_client();