add reflection

Author: 0g-peterzhb

.gitignore

@@ -7,3 +7,5 @@ tests/tmp/**
 .vscode/*.json
 /0g-storage-contracts-dev
 /run/.env
+
+**.bin

Cargo.lock

@@ -32,6 +32,7 @@ parking_lot = "0.12.3"
 tonic = { version = "0.9.2", features = ["transport"] }
 prost = "0.11.9"
 prost-types = "0.11.9"
+tonic-reflection = "0.9.2"
 
 [build-dependencies]
 tonic-build = "0.9.2"

node/rpc/Cargo.toml

@@ -1,5 +1,7 @@
 fn main() -> Result<(), Box<dyn std::error::Error>> {
     // Compile proto/my_service.proto
-    tonic_build::compile_protos("proto/zgs_grpc.proto")?;
+    tonic_build::configure()
+        .file_descriptor_set_path("proto/zgs_grpc_descriptor.bin")
+        .compile(&["proto/zgs_grpc.proto"], &["proto"])?;
     Ok(())
 }

node/rpc/build.rs

@@ -34,6 +34,9 @@ pub use admin::RpcClient as ZgsAdminRpcClient;
 pub use config::Config as RPCConfig;
 pub use miner::RpcClient as ZgsMinerRpcClient;
 pub use zgs::RpcClient as ZgsRPCClient;
+// bring in the reflection-builder
+use tonic_reflection::server::Builder as ReflectionBuilder;
+
 
 pub mod zgs_grpc_proto {
     tonic::include_proto!("zgs_grpc");
@@ -43,6 +46,8 @@ mod zgs_grpc;
 
 use tonic::transport::Server;
 
+const DESCRIPTOR_SET: &[u8] = include_bytes!("../proto/zgs_grpc_descriptor.bin");
+
 /// A wrapper around all the items required to spawn the HTTP server.
 ///
 /// The server will gracefully handle the case where any fields are `None`.
@@ -146,9 +151,14 @@ async fn run_server_public_private(
 
 pub async fn run_grpc_server(ctx: Context) -> Result<(), Box<dyn Error>> {
     let grpc_addr = ctx.config.listen_address_grpc;
+    let reflection = ReflectionBuilder::configure()
+        .register_encoded_file_descriptor_set(DESCRIPTOR_SET)
+        .build()?;
+
     let server = ZgsGrpcServiceServer::new(ZgsGrpcServiceImpl { ctx });
     Server::builder()
         .add_service(server)
+        .add_service(reflection)
         .serve(grpc_addr)
         .await?;
     Ok(())