From d6373d540751c06b7fb96f0e7e26ad56f50f25bd Mon Sep 17 00:00:00 2001
From: satoren <satoreyo@hotmail.com>
Date: Tue, 16 Apr 2024 21:12:26 +0900
Subject: [PATCH] Rust: fix panic with normal errors (#1377)

---
 rust/src/router/consumer.rs | 4 ++--
 rust/src/router/producer.rs | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/rust/src/router/consumer.rs b/rust/src/router/consumer.rs
index 3914d40b2d..8d58811aea 100644
--- a/rust/src/router/consumer.rs
+++ b/rust/src/router/consumer.rs
@@ -1020,9 +1020,9 @@ impl Consumer {
             .inner
             .channel
             .request(self.id(), ConsumerGetStatsRequest {})
-            .await;
+            .await?;
 
-        if let Ok(response::Body::ConsumerGetStatsResponse(data)) = response {
+        if let response::Body::ConsumerGetStatsResponse(data) = response {
             match data.stats.len() {
                 0 => panic!("Empty stats response from worker"),
                 1 => {
diff --git a/rust/src/router/producer.rs b/rust/src/router/producer.rs
index 8b1a952e37..ec60283239 100644
--- a/rust/src/router/producer.rs
+++ b/rust/src/router/producer.rs
@@ -899,9 +899,9 @@ impl Producer {
             .inner()
             .channel
             .request(self.id(), ProducerGetStatsRequest {})
-            .await;
+            .await?;
 
-        if let Ok(response::Body::ProducerGetStatsResponse(data)) = response {
+        if let response::Body::ProducerGetStatsResponse(data) = response {
             Ok(data.stats.iter().map(ProducerStat::from_fbs).collect())
         } else {
             panic!("Wrong message from worker");