miden-node-rpc 0.17.0-rc.4

Miden node's front-end RPC server
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use miden_node_block_producer::{BlockProducerStatus, MempoolStats};
use miden_node_proto::generated as proto;
use miden_node_tracing::{debug, miden_instrument};

use super::{ProtoMempoolStats, Request, RpcBackend, RpcService};
use crate::{COMPONENT, LOG_TARGET};

#[tonic::async_trait]
impl proto::server::miden_node_v1_node_service::Status for RpcService {
    type Input = ();
    type Output = proto::miden::node::v1::StatusResponse;

    fn decode(_request: proto::miden::node::v1::StatusRequest) -> tonic::Result<Self::Input> {
        Ok(())
    }

    fn encode(output: Self::Output) -> tonic::Result<proto::miden::node::v1::StatusResponse> {
        Ok(output)
    }

    #[miden_instrument(
        target = COMPONENT,
        name = "status",
        err,
    )]
    async fn handle(
        &self,
        _input: Self::Input,
        _metadata: &tonic::metadata::MetadataMap,
        _extensions: &tonic::codegen::http::Extensions,
    ) -> tonic::Result<Self::Output> {
        let block_producer_status = match &self.backend {
            RpcBackend::Sequencer { block_producer, .. } => {
                Some(block_producer_status_to_proto(block_producer.status().await))
            },
            RpcBackend::FullNode { source_rpc, .. } => source_rpc
                .as_ref()
                .clone()
                .status(Request::new(proto::miden::node::v1::StatusRequest {}))
                .await
                .ok()
                .and_then(|response| response.into_inner().block_producer),
        };

        debug!(target: LOG_TARGET, "Getting status");

        Ok(proto::miden::node::v1::StatusResponse {
            version: env!("CARGO_PKG_VERSION").to_string(),
            chain_tip: self.state.committed_tip().as_u32(),
            block_producer: block_producer_status.or(Some(
                proto::miden::node::v1::BlockProducerStatus {
                    status: "unreachable".to_string(),
                    version: "-".to_string(),
                    chain_tip: 0,
                    mempool_stats: Some(ProtoMempoolStats::default()),
                },
            )),
            genesis_commitment: self.genesis_commitment.map(Into::into),
        })
    }
}

// HELPERS
// ================================================================================================

fn block_producer_status_to_proto(
    status: BlockProducerStatus,
) -> proto::miden::node::v1::BlockProducerStatus {
    proto::miden::node::v1::BlockProducerStatus {
        version: status.version,
        status: status.status,
        chain_tip: status.chain_tip.as_u32(),
        mempool_stats: Some(block_producer_mempool_stats_to_proto(status.mempool_stats)),
    }
}

fn block_producer_mempool_stats_to_proto(
    stats: MempoolStats,
) -> proto::miden::node::v1::MempoolStats {
    proto::miden::node::v1::MempoolStats {
        unbatched_transactions: stats.unbatched_transactions,
        proposed_batches: stats.proposed_batches,
        proven_batches: stats.proven_batches,
        uncommitted_transactions: stats.uncommitted_transactions,
    }
}