Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: new grpc method #6742

Merged
merged 6 commits into from
Jan 16, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions applications/minotari_app_grpc/proto/base_node.proto
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ service BaseNode {
// Get templates
rpc GetTemplateRegistrations(GetTemplateRegistrationsRequest) returns (stream GetTemplateRegistrationResponse);
rpc GetSideChainUtxos(GetSideChainUtxosRequest) returns (stream GetSideChainUtxosResponse);
rpc GetNetworkState(GetNetworkStateRequest) returns (GetNetworkStateResponse);
}

message GetAssetMetadataRequest {
Expand Down Expand Up @@ -521,3 +522,23 @@ message GetSideChainUtxosResponse {
repeated TransactionOutput outputs = 2;
}

message GetNetworkStateRequest {
}

message GetNetworkStateResponse {
// metadata
MetaData metadata = 1;
// has the base node synced
bool initial_sync_achieved = 2;
//current state of the base node
BaseNodeState base_node_state = 3;
// do we have failed checkpoints
bool failed_checkpoints = 4;
// The block reward of the next tip
uint64 reward = 5;
// estimate sha3x hash rate
uint64 sha3x_estimated_hash_rate = 6;
// estimate randomx hash rate
uint64 randomx_estimated_hash_rate = 7;
}

169 changes: 158 additions & 11 deletions applications/minotari_node/src/grpc/base_node_grpc_server.rs
Original file line number Diff line number Diff line change
Expand Up @@ -79,6 +79,7 @@ use crate::{
builder::BaseNodeContext,
grpc::{
blocks::{block_fees, block_heights, block_size, GET_BLOCKS_MAX_HEIGHTS, GET_BLOCKS_PAGE_SIZE},
data_cache::DataCache,
hash_rate::HashRateMovingAverage,
helpers::{mean, median},
},
Expand Down Expand Up @@ -117,6 +118,7 @@ pub struct BaseNodeGrpcServer {
report_grpc_error: bool,
tari_pulse: TariPulseHandle,
config: BaseNodeConfig,
data_cache: DataCache,
}

impl BaseNodeGrpcServer {
Expand All @@ -133,6 +135,7 @@ impl BaseNodeGrpcServer {
report_grpc_error: ctx.get_report_grpc_error(),
tari_pulse: ctx.tari_pulse(),
config,
data_cache: DataCache::new(),
}
}

Expand Down Expand Up @@ -364,6 +367,96 @@ impl tari_rpc::base_node_server::BaseNode for BaseNodeGrpcServer {
Ok(Response::new(rx))
}

async fn get_network_state(
&self,
_request: Request<tari_rpc::GetNetworkStateRequest>,
) -> Result<Response<tari_rpc::GetNetworkStateResponse>, Status> {
trace!(target: LOG_TARGET, "Incoming GRPC request for get network hash rate");
let report_error_flag = self.report_error_flag();
let mut handler = self.node_service.clone();
let metadata = handler.get_metadata().await.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get node tip: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
let reward = self
.consensus_rules
.get_block_reward_at(metadata.best_block_height())
.as_u64();
let constants = self.consensus_rules.consensus_constants(metadata.best_block_height());
let sha3x_estimated_hash_rate = match self
.data_cache
.get_sha3x_estimated_hash_rate(metadata.best_block_hash())
.await
{
Some(hash_rate) => hash_rate,
None => {
let target_difficulty = handler
.get_target_difficulty_for_next_block(PowAlgorithm::Sha3x)
.await
.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get target difficulty for Sha3x: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
let target_time = constants.pow_target_block_interval(PowAlgorithm::Sha3x);
let estimated_hash_rate = target_difficulty.as_u64() / target_time;
self.data_cache
.set_sha3x_estimated_hash_rate(estimated_hash_rate, *metadata.best_block_hash())
.await;
estimated_hash_rate
},
};
let randomx_estimated_hash_rate = match self
.data_cache
.get_randomx_estimated_hash_rate(metadata.best_block_hash())
.await
{
Some(hash_rate) => hash_rate,
None => {
let target_difficulty = handler
.get_target_difficulty_for_next_block(PowAlgorithm::RandomX)
.await
.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get target difficulty for RandomX: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
let target_time = constants.pow_target_block_interval(PowAlgorithm::RandomX);
let estimated_hash_rate = target_difficulty.as_u64() / target_time;
self.data_cache
.set_randomx_estimated_hash_rate(estimated_hash_rate, *metadata.best_block_hash())
.await;
estimated_hash_rate
},
};

let failed_checkpoints = *self.tari_pulse.get_failed_checkpoints_notifier();
let status_watch = self.state_machine_handle.get_status_info_watch();
let state: tari_rpc::BaseNodeState = (&status_watch.borrow().state_info).into();

let response = tari_rpc::GetNetworkStateResponse {
metadata: Some(metadata.into()),
initial_sync_achieved: status_watch.borrow().bootstrapped,
base_node_state: state.into(),
failed_checkpoints,
reward,
sha3x_estimated_hash_rate,
randomx_estimated_hash_rate,
};
trace!(target: LOG_TARGET, "Sending GetNetworkState response to client");
Ok(Response::new(response))
}

async fn get_mempool_transactions(
&self,
request: Request<tari_rpc::GetMempoolTransactionsRequest>,
Expand Down Expand Up @@ -596,6 +689,7 @@ impl tari_rpc::base_node_server::BaseNode for BaseNodeGrpcServer {
Ok(Response::new(rx))
}

#[allow(clippy::too_many_lines)]
async fn get_new_block_template(
&self,
request: Request<tari_rpc::NewBlockTemplateRequest>,
Expand Down Expand Up @@ -624,18 +718,71 @@ impl tari_rpc::base_node_server::BaseNode for BaseNodeGrpcServer {
})?;

let mut handler = self.node_service.clone();
let metadata = handler.get_metadata().await.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get node tip: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;

let new_template = handler
.get_new_block_template(algo, request.max_weight)
.await
.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get new block template: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
let new_template = match algo {
PowAlgorithm::Sha3x => {
match self
.data_cache
.get_sha3x_new_block_template(metadata.best_block_hash())
.await
{
Some(template) => template,
None => {
let new_template =
handler
.get_new_block_template(algo, request.max_weight)
.await
.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get new block template: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
self.data_cache
.set_sha3x_new_block_template(new_template.clone(), *metadata.best_block_hash())
.await;
new_template
},
}
},
PowAlgorithm::RandomX => {
match self
.data_cache
.get_randomx_new_block_template(metadata.best_block_hash())
.await
{
Some(template) => template,
None => {
let new_template =
handler
.get_new_block_template(algo, request.max_weight)
.await
.map_err(|e| {
warn!(
target: LOG_TARGET,
"Could not get new block template: {}",
e.to_string()
);
obscure_error_if_true(report_error_flag, Status::internal(e.to_string()))
})?;
self.data_cache
.set_randomx_new_block_template(new_template.clone(), *metadata.best_block_hash())
.await;
new_template
},
}
},
};

let status_watch = self.state_machine_handle.get_status_info_watch();
let pow = algo as i32;
Expand Down
128 changes: 128 additions & 0 deletions applications/minotari_node/src/grpc/data_cache.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,128 @@
// Copyright 2025. The Tari Project
//
// Redistribution and use in source and binary forms, with or without modification, are permitted provided that the
// following conditions are met:
//
// 1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following
// disclaimer.
//
// 2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the
// following disclaimer in the documentation and/or other materials provided with the distribution.
//
// 3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote
// products derived from this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
// INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
// SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
// WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
// USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use std::sync::Arc;

use tari_common_types::types::FixedHash;
use tari_core::blocks::NewBlockTemplate;
use tokio::sync::RwLock;

pub struct DataCache {
inner_data_cache: Arc<RwLock<InnerDataCache>>,
}

impl DataCache {
pub fn new() -> Self {
Self {
inner_data_cache: Arc::new(RwLock::new(InnerDataCache::default())),
}
}

pub async fn get_randomx_estimated_hash_rate(&self, current_tip: &FixedHash) -> Option<u64> {
let res = &self.inner_data_cache.read().await.randomx_estimated_hash_rate;
if res.tip == *current_tip {
Some(res.data)
} else {
None
}
}

pub async fn get_sha3x_estimated_hash_rate(&self, current_tip: &FixedHash) -> Option<u64> {
let res = &self.inner_data_cache.read().await.sha3x_estimated_hash_rate;
if res.tip == *current_tip {
Some(res.data)
} else {
None
}
}

pub async fn set_randomx_estimated_hash_rate(&self, hash_rate: u64, current_tip: FixedHash) {
self.inner_data_cache.write().await.randomx_estimated_hash_rate = DataCacheData::new(hash_rate, current_tip);
}

pub async fn set_sha3x_estimated_hash_rate(&self, hash_rate: u64, current_tip: FixedHash) {
self.inner_data_cache.write().await.sha3x_estimated_hash_rate = DataCacheData::new(hash_rate, current_tip);
}

pub async fn get_randomx_new_block_template(&self, current_tip: &FixedHash) -> Option<NewBlockTemplate> {
let res = &self.inner_data_cache.read().await.randomx_new_block_template;
if res.tip == *current_tip {
Some(res.data.clone())
} else {
None
}
}

pub async fn get_sha3x_new_block_template(&self, current_tip: &FixedHash) -> Option<NewBlockTemplate> {
let res = &self.inner_data_cache.read().await.sha3x_new_block_template;
if res.tip == *current_tip {
Some(res.data.clone())
} else {
None
}
}

pub async fn set_randomx_new_block_template(&self, new_block_template: NewBlockTemplate, current_tip: FixedHash) {
self.inner_data_cache.write().await.randomx_new_block_template =
DataCacheData::new(new_block_template, current_tip);
}

pub async fn set_sha3x_new_block_template(&self, new_block_template: NewBlockTemplate, current_tip: FixedHash) {
self.inner_data_cache.write().await.sha3x_new_block_template =
DataCacheData::new(new_block_template, current_tip);
}
}

struct InnerDataCache {
pub randomx_estimated_hash_rate: DataCacheData<u64>,
pub sha3x_estimated_hash_rate: DataCacheData<u64>,
pub sha3x_new_block_template: DataCacheData<NewBlockTemplate>,
pub randomx_new_block_template: DataCacheData<NewBlockTemplate>,
}
impl Default for InnerDataCache {
fn default() -> Self {
Self {
randomx_estimated_hash_rate: DataCacheData::new_empty(0),
sha3x_estimated_hash_rate: DataCacheData::new_empty(0),
sha3x_new_block_template: DataCacheData::new_empty(NewBlockTemplate::empty()),
randomx_new_block_template: DataCacheData::new_empty(NewBlockTemplate::empty()),
}
}
}

struct DataCacheData<T> {
pub data: T,
pub tip: FixedHash,
}

impl<T> DataCacheData<T> {
pub fn new(data: T, tip: FixedHash) -> Self {
Self { data, tip }
}

pub fn new_empty(data: T) -> Self {
Self {
data,
tip: FixedHash::default(),
}
}
}
1 change: 1 addition & 0 deletions applications/minotari_node/src/grpc/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,5 +22,6 @@

pub mod base_node_grpc_server;
pub mod blocks;
pub mod data_cache;
pub mod hash_rate;
pub mod helpers;
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ pub struct MmrStateRequest {
#[derive(Debug, Serialize, Deserialize)]
pub enum NodeCommsRequest {
GetChainMetadata,
GetTargetDifficultyNextBlock(PowAlgorithm),
FetchHeaders(RangeInclusive<u64>),
FetchHeadersByHashes(Vec<HashOutput>),
FetchMatchingUtxos(Vec<HashOutput>),
Expand Down Expand Up @@ -74,6 +75,7 @@ impl Display for NodeCommsRequest {
use NodeCommsRequest::*;
match self {
GetChainMetadata => write!(f, "GetChainMetadata"),
GetTargetDifficultyNextBlock(algo) => write!(f, "GetTargetDifficultyNextBlock ({:?})", algo),
FetchHeaders(range) => {
write!(f, "FetchHeaders ({:?})", range)
},
Expand Down
Loading
Loading