Skip to content

Commit 6ef148a

Browse files
authored
feat: improve error msg when state streaming API is unimplemented (#598)
1 parent 7c1f9f2 commit 6ef148a

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

tokio-console/src/conn.rs

+12-1
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,18 @@ impl Connection {
122122
let update_stream =
123123
Box::new(client.watch_updates(update_request).await?.into_inner());
124124
let state_request = tonic::Request::new(StateRequest {});
125-
let state_stream = Box::new(client.watch_state(state_request).await?.into_inner());
125+
let state_stream = match client.watch_state(state_request).await {
126+
Ok(stream) => Box::new(stream.into_inner()),
127+
Err(e) => {
128+
if e.code() == tonic::Code::Unimplemented {
129+
tracing::error!(
130+
"The server at {} does not support state streaming. Please update the console-subscriber to v0.5.0 or later version.",
131+
self.target
132+
);
133+
}
134+
return Err(e.into());
135+
}
136+
};
126137
Ok::<State, Box<dyn Error + Send + Sync>>(State::Connected {
127138
client,
128139
update_stream,

0 commit comments

Comments
 (0)