linera-service: enable gRPC reflection

Cargo.toml

@@ -114,6 +114,7 @@ thiserror = "1.0.57"
 tonic = { version = "0.11", default-features = false }
 tonic-build = { version = "0.11", default-features = false }
 tonic-health = "0.11"
+tonic-reflection = "0.11"
 tonic-web = "0.11"
 tonic-web-wasm-client = "0.5.1"
 tokio = "1.36.0"

flake.nix

@@ -35,6 +35,7 @@
           kubectl
 
           # for Wasm testing
+          chromium
           chromedriver
           wasm-pack
         ];

linera-rpc/Cargo.toml

@@ -29,7 +29,7 @@ metrics = [
     "linera-views/metrics",
 ]
 
-server = ["tonic-health"]
+server = ["tonic-health", "tonic-reflection"]
 simple-network = ["tokio-util/net"]
 
 web = [
@@ -65,6 +65,7 @@ thiserror.workspace = true
 tokio.workspace = true
 tokio-util = { workspace = true, optional = true, features = ["codec"] }
 tonic-health = { workspace = true, optional = true }
+tonic-reflection = { workspace = true, optional = true }
 tower.workspace = true
 tracing.workspace = true
 

linera-rpc/build.rs

@@ -2,8 +2,11 @@
 // SPDX-License-Identifier: Apache-2.0
 
 fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let out_dir: std::path::PathBuf = std::env::var("OUT_DIR")?.into();
+
     let no_includes: &[&str] = &[];
     tonic_build::configure()
+        .file_descriptor_set_path(out_dir.join("file_descriptor_set.bin"))
         .protoc_arg("--experimental_allow_proto3_optional")
         .compile(&["proto/rpc.proto"], no_includes)?;
 

linera-rpc/src/grpc/mod.rs

@@ -37,6 +37,10 @@ pub enum GrpcError {
 
     #[error(transparent)]
     InvalidUri(#[from] tonic::codegen::http::uri::InvalidUri),
+
+    #[cfg(with_server)]
+    #[error(transparent)]
+    Reflection(#[from] tonic_reflection::server::Error),
 }
 
 const MEBIBYTE: usize = 1024 * 1024;

linera-rpc/src/grpc/server.rs

@@ -252,6 +252,10 @@ where
             .max_encoding_message_size(GRPC_MAX_MESSAGE_SIZE)
             .max_decoding_message_size(GRPC_MAX_MESSAGE_SIZE);
 
+        let reflection_service = tonic_reflection::server::Builder::configure()
+            .register_encoded_file_descriptor_set(crate::FILE_DESCRIPTOR_SET)
+            .build()?;
+
         let handle = tokio::spawn(
             tonic::transport::Server::builder()
                 .layer(
@@ -260,6 +264,7 @@ where
                         .into_inner(),
                 )
                 .add_service(health_service)
+                .add_service(reflection_service)
                 .add_service(worker_node)
                 .serve_with_shutdown(server_address, receiver.map(|_| ())),
         );

linera-rpc/src/lib.rs

@@ -37,3 +37,5 @@ pub struct HandleCertificateRequest {
     pub wait_for_outgoing_messages: bool,
     pub blobs: Vec<linera_chain::data_types::HashedValue>,
 }
+
+pub const FILE_DESCRIPTOR_SET: &[u8] = tonic::include_file_descriptor_set!("file_descriptor_set");

linera-service/Cargo.toml

@@ -79,6 +79,7 @@ tokio-stream.workspace = true
 toml.workspace = true
 tonic = { workspace = true, features = ["transport", "tls", "tls-roots"] }
 tonic-health.workspace = true
+tonic-reflection.workspace = true
 tonic-web.workspace = true
 tower.workspace = true
 tower-http = { workspace = true, features = ["cors"] }

linera-service/src/grpc_proxy.rs

@@ -214,14 +214,19 @@ impl GrpcProxy {
         let internal_server = Server::builder()
             .add_service(tonic_web::enable(self.as_notifier_service()))
             .serve(self.internal_address());
+        let reflection_service = tonic_reflection::server::Builder::configure()
+            .register_encoded_file_descriptor_set(linera_rpc::FILE_DESCRIPTOR_SET)
+            .build()?;
         let public_server = self
             .public_server()?
             .layer(
                 ServiceBuilder::new()
                     .layer(PrometheusMetricsMiddlewareLayer)
                     .into_inner(),
             )
+            .accept_http1(true)
             .add_service(health_service)
+            .add_service(reflection_service)
             .add_service(self.as_validator_node())
             .serve(self.public_address());