penumbra_sdk_app/
rpc.rs

1mod query;
2
3// TODO: Once we migrate to Tonic 0.10.0, we'll be able to use the `Routes` structure to have each
4// component define a method that returns a `Routes` with all of its query services bundled inside.
5//
6// This means we won't have to import all this shit and recite every single service -- we can e.g.,
7// have the app crate assemble all of its components' query services into a single `Routes` and
8// then just add that to the gRPC server.
9use {
10    self::query::AppQueryServer,
11    crate::PenumbraHost,
12    anyhow::Context,
13    cnidarium::proto::v1::query_service_server::QueryServiceServer as StorageQueryServiceServer,
14    cnidarium::rpc::Server as StorageServer,
15    ibc_proto::{
16        cosmos::bank::v1beta1::query_server::QueryServer as TransferQueryServer,
17        ibc::{
18            applications::transfer::v1::query_server::QueryServer as BankQueryServer,
19            core::{
20                channel::v1::query_server::QueryServer as ChannelQueryServer,
21                client::v1::query_server::QueryServer as ClientQueryServer,
22                connection::v1::query_server::QueryServer as ConnectionQueryServer,
23            },
24        },
25    },
26    penumbra_sdk_auction::component::rpc::Server as AuctionServer,
27    penumbra_sdk_compact_block::component::rpc::Server as CompactBlockServer,
28    penumbra_sdk_dex::component::rpc::Server as DexServer,
29    penumbra_sdk_fee::component::rpc::Server as FeeServer,
30    penumbra_sdk_governance::component::rpc::Server as GovernanceServer,
31    penumbra_sdk_proto::{
32        core::{
33            app::v1::query_service_server::QueryServiceServer as AppQueryServiceServer,
34            component::{
35                auction::v1::query_service_server::QueryServiceServer as AuctionQueryServiceServer,
36                compact_block::v1::query_service_server::QueryServiceServer as CompactBlockQueryServiceServer,
37                dex::v1::{
38                    query_service_server::QueryServiceServer as DexQueryServiceServer,
39                    simulation_service_server::SimulationServiceServer,
40                },
41                fee::v1::query_service_server::QueryServiceServer as FeeQueryServiceServer,
42                governance::v1::query_service_server::QueryServiceServer as GovernanceQueryServiceServer,
43                sct::v1::query_service_server::QueryServiceServer as SctQueryServiceServer,
44                shielded_pool::v1::query_service_server::QueryServiceServer as ShieldedPoolQueryServiceServer,
45                stake::v1::query_service_server::QueryServiceServer as StakeQueryServiceServer,
46            },
47        },
48        util::tendermint_proxy::v1::tendermint_proxy_service_server::{
49            TendermintProxyService, TendermintProxyServiceServer,
50        },
51    },
52    penumbra_sdk_sct::component::rpc::Server as SctServer,
53    penumbra_sdk_shielded_pool::component::rpc::Server as ShieldedPoolServer,
54    penumbra_sdk_stake::component::rpc::Server as StakeServer,
55    tonic::service::Routes,
56    tonic_web::enable as we,
57};
58
59pub fn routes(
60    storage: &cnidarium::Storage,
61    tm_proxy: impl TendermintProxyService,
62    _enable_expensive_rpc: bool,
63) -> anyhow::Result<tonic::service::Routes> {
64    let ibc = penumbra_sdk_ibc::component::rpc::IbcQuery::<PenumbraHost>::new(storage.clone());
65
66    let mut builder = Routes::builder();
67    builder
68        // As part of #2932, we are disabling all timeouts until we circle back to our
69        // performance story.
70        // Sets a timeout for all gRPC requests, but note that in the case of streaming
71        // requests, the timeout is only applied to the initial request. This means that
72        // this does not prevent long lived streams, for example to allow clients to obtain
73        // new blocks.
74        // .timeout(std::time::Duration::from_secs(7))
75        // Wrap each of the gRPC services in a tonic-web proxy:
76        .add_service(we(StorageQueryServiceServer::new(StorageServer::new(
77            storage.clone(),
78        ))))
79        .add_service(we(AuctionQueryServiceServer::new(AuctionServer::new(
80            storage.clone(),
81        ))))
82        .add_service(we(AppQueryServiceServer::new(AppQueryServer::new(
83            storage.clone(),
84        ))))
85        .add_service(we(CompactBlockQueryServiceServer::new(
86            CompactBlockServer::new(storage.clone()),
87        )))
88        .add_service(we(DexQueryServiceServer::new(DexServer::new(
89            storage.clone(),
90        ))))
91        .add_service(we(FeeQueryServiceServer::new(FeeServer::new(
92            storage.clone(),
93        ))))
94        .add_service(we(GovernanceQueryServiceServer::new(
95            GovernanceServer::new(storage.clone()),
96        )))
97        .add_service(we(SctQueryServiceServer::new(SctServer::new(
98            storage.clone(),
99        ))))
100        .add_service(we(ShieldedPoolQueryServiceServer::new(
101            ShieldedPoolServer::new(storage.clone()),
102        )))
103        .add_service(we(TransferQueryServer::new(ShieldedPoolServer::new(
104            storage.clone(),
105        ))))
106        .add_service(we(BankQueryServer::new(ShieldedPoolServer::new(
107            storage.clone(),
108        ))))
109        .add_service(we(StakeQueryServiceServer::new(StakeServer::new(
110            storage.clone(),
111        ))))
112        .add_service(we(ClientQueryServer::new(ibc.clone())))
113        .add_service(we(ChannelQueryServer::new(ibc.clone())))
114        .add_service(we(ConnectionQueryServer::new(ibc.clone())))
115        .add_service(we(TendermintProxyServiceServer::new(tm_proxy)))
116        .add_service(we(SimulationServiceServer::new(DexServer::new(
117            storage.clone(),
118        ))))
119        .add_service(we(tonic_reflection::server::Builder::configure()
120            .register_encoded_file_descriptor_set(penumbra_sdk_proto::FILE_DESCRIPTOR_SET)
121            .build_v1()
122            .with_context(|| "could not configure grpc reflection service")?));
123    Ok(builder.routes().prepare())
124}
penumbra_sdk_app/rpc.rs

penumbra_sdk_app/
rpc.rs
