forked from hyperium/tonic
-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.rs
44 lines (36 loc) · 1.38 KB
/
client.rs
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
use tokio_stream::StreamExt;
use tonic_reflection::pb::{
server_reflection_client::ServerReflectionClient, server_reflection_request::MessageRequest,
server_reflection_response::MessageResponse, ServerReflectionRequest, ServerReflectionResponse,
};
fn parse_response(resp: ServerReflectionResponse) {
let message_response = resp.message_response.expect("message response");
if let MessageResponse::ListServicesResponse(list_response) = message_response {
for svc in list_response.service {
println!("\tfound service: `{}`", svc.name);
}
}
}
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let conn = tonic::transport::Endpoint::new("http://[::1]:50052")?
.connect()
.await?;
let mut client = ServerReflectionClient::new(conn);
let list_services_request = ServerReflectionRequest {
host: "host".into(),
message_request: Some(MessageRequest::ListServices("list".into())),
};
let request_stream = tokio_stream::once(list_services_request);
let mut inbound = client
.server_reflection_info(request_stream)
.await?
.into_inner();
while let Some(recv) = inbound.next().await {
match recv {
Ok(resp) => parse_response(resp),
Err(e) => println!("\tdid not receive response due to error: `{}`", e),
}
}
Ok(())
}