Gas Profiling Implementation

Cargo.toml

@@ -8,6 +8,7 @@ members = [
     "moveos/moveos-commons/bcs_ext",
     "moveos/moveos-commons/moveos-common",
     "moveos/moveos-commons/timeout-join-handler",
+    "moveos/moveos-commons/moveos-common",
     "moveos/moveos-compiler",
     "moveos/moveos-config",
     "moveos/moveos-object-runtime",
@@ -18,6 +19,7 @@ members = [
     "moveos/raw-store",
     "moveos/smt",
     "moveos/moveos-eventbus",
+    "moveos/moveos-gas-profiling",
     "crates/data_verify",
     "crates/rooch",
     "crates/rooch-benchmarks",
@@ -103,6 +105,7 @@ moveos-object-runtime = { path = "moveos/moveos-object-runtime" }
 moveos-compiler = { path = "moveos/moveos-compiler" }
 moveos-eventbus = { path = "moveos/moveos-eventbus" }
 accumulator = { path = "moveos/moveos-commons/accumulator" }
+moveos-gas-profiling = { path = "moveos/moveos-gas-profiling" }
 
 # crates for Rooch
 rooch = { path = "crates/rooch" }
@@ -347,6 +350,9 @@ vergen-git2 = { version = "1.0.0", features = ["build", "cargo", "rustc"] }
 vergen-pretty = "0.3.5"
 crossbeam-channel = "0.5.13"
 
+inferno = "0.11.14"
+handlebars = "4.2.2"
+
 # Note: the BEGIN and END comments below are required for external tooling. Do not remove.
 # BEGIN MOVE DEPENDENCIES
 move-abigen = { git = "https://github.com/rooch-network/move", rev = "c05b5f71a24beb498eb743d73be4f69d8d325e62" }

crates/rooch-executor/Cargo.toml

@@ -35,3 +35,4 @@ rooch-types = { workspace = true }
 rooch-genesis = { workspace = true }
 rooch-event = { workspace = true }
 rooch-store = { workspace = true }
+hex = "0.4.3"

crates/rooch-executor/src/actor/messages.rs

@@ -99,6 +99,7 @@ impl Message for ExecuteViewFunctionMessage {
 
 #[derive(Debug, Serialize, Deserialize)]
 pub struct StatesMessage {
+    pub state_root: Option<H256>,
     pub access_path: AccessPath,
 }
 
@@ -126,6 +127,7 @@ impl Message for AnnotatedStatesMessage {
 
 #[derive(Debug, Serialize, Deserialize)]
 pub struct ListStatesMessage {
+    pub state_root: Option<H256>,
     pub access_path: AccessPath,
     pub cursor: Option<FieldKey>,
     pub limit: usize,

crates/rooch-executor/src/actor/reader_executor.rs

@@ -146,7 +146,12 @@ impl Handler<StatesMessage> for ReaderExecutorActor {
         msg: StatesMessage,
         _ctx: &mut ActorContext,
     ) -> Result<Vec<Option<ObjectState>>, anyhow::Error> {
-        let resolver = RootObjectResolver::new(self.root.clone(), &self.moveos_store);
+        let resolver = if let Some(state_root) = msg.state_root {
+            let root_object_meta = ObjectMeta::root_metadata(state_root, 0);
+            RootObjectResolver::new(root_object_meta, &self.moveos_store)
+        } else {
+            RootObjectResolver::new(self.root.clone(), &self.moveos_store)
+        };
         resolver.get_states(msg.access_path)
     }
 }
@@ -170,7 +175,12 @@ impl Handler<ListStatesMessage> for ReaderExecutorActor {
         msg: ListStatesMessage,
         _ctx: &mut ActorContext,
     ) -> Result<Vec<StateKV>, anyhow::Error> {
-        let resolver = RootObjectResolver::new(self.root.clone(), &self.moveos_store);
+        let resolver = if let Some(state_root) = msg.state_root {
+            let root_object_meta = ObjectMeta::root_metadata(state_root, 0);
+            RootObjectResolver::new(root_object_meta, &self.moveos_store)
+        } else {
+            RootObjectResolver::new(self.root.clone(), &self.moveos_store)
+        };
         resolver.list_states(msg.access_path, msg.cursor, msg.limit)
     }
 }

crates/rooch-executor/src/proxy/mod.rs

@@ -116,9 +116,16 @@ impl ExecutorProxy {
             .await?
     }
 
-    pub async fn get_states(&self, access_path: AccessPath) -> Result<Vec<Option<ObjectState>>> {
+    pub async fn get_states(
+        &self,
+        access_path: AccessPath,
+        state_root: Option<H256>,
+    ) -> Result<Vec<Option<ObjectState>>> {
         self.reader_actor
-            .send(StatesMessage { access_path })
+            .send(StatesMessage {
+                state_root,
+                access_path,
+            })
             .await?
     }
 
@@ -133,12 +140,14 @@ impl ExecutorProxy {
 
     pub async fn list_states(
         &self,
+        state_root: Option<H256>,
         access_path: AccessPath,
         cursor: Option<FieldKey>,
         limit: usize,
     ) -> Result<Vec<StateKV>> {
         self.reader_actor
             .send(ListStatesMessage {
+                state_root,
                 access_path,
                 cursor,
                 limit,
@@ -261,7 +270,7 @@ impl ExecutorProxy {
     }
 
     pub async fn chain_id(&self) -> Result<ChainID> {
-        self.get_states(AccessPath::object(ChainID::chain_id_object_id()))
+        self.get_states(AccessPath::object(ChainID::chain_id_object_id()), None)
             .await?
             .into_iter()
             .next()
@@ -271,7 +280,7 @@ impl ExecutorProxy {
     }
 
     pub async fn bitcoin_network(&self) -> Result<BitcoinNetwork> {
-        self.get_states(AccessPath::object(BitcoinNetwork::object_id()))
+        self.get_states(AccessPath::object(BitcoinNetwork::object_id()), None)
             .await?
             .into_iter()
             .next()
@@ -283,7 +292,10 @@ impl ExecutorProxy {
     //TODO provide a trait to abstract the async state reader, elemiate the duplicated code bwteen RpcService and Client
     pub async fn get_sequence_number(&self, address: AccountAddress) -> Result<u64> {
         Ok(self
-            .get_states(AccessPath::object(Account::account_object_id(address)))
+            .get_states(
+                AccessPath::object(Account::account_object_id(address)),
+                None,
+            )
             .await?
             .pop()
             .flatten()

crates/rooch-open-rpc-spec/schemas/openrpc.json

@@ -2827,6 +2827,18 @@
             "description": "If true, result with display rendered is returned",
             "default": false,
             "type": "boolean"
+          },
+          "stateRoot": {
+            "description": "The state root of remote stateDB",
+            "default": null,
+            "anyOf": [
+              {
+                "$ref": "#/components/schemas/primitive_types::H256"
+              },
+              {
+                "type": "null"
+              }
+            ]
           }
         }
       },

crates/rooch-rpc-api/src/jsonrpc_types/rpc_options.rs

@@ -1,6 +1,8 @@
 // Copyright (c) RoochNetwork
 // SPDX-License-Identifier: Apache-2.0
 
+use crate::jsonrpc_types::H256View;
+use moveos_types::h256::H256;
 use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 
@@ -11,6 +13,8 @@ pub struct StateOptions {
     pub decode: bool,
     /// If true, result with display rendered is returned
     pub show_display: bool,
+    /// The state root of remote stateDB
+    pub state_root: Option<H256View>,
 }
 
 impl StateOptions {
@@ -27,6 +31,16 @@ impl StateOptions {
         self.show_display = show_display;
         self
     }
+
+    pub fn state_root(mut self, state_root: Option<H256>) -> Self {
+        match state_root {
+            None => {}
+            Some(h256) => {
+                self.state_root = Some(H256View::from(h256));
+            }
+        }
+        self
+    }
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, JsonSchema, Eq, PartialEq, Default)]