From 562552bf303d55a566e18b94a70c751fbac52345 Mon Sep 17 00:00:00 2001
From: sheyanjie-qq <249478495@qq.com>
Date: Fri, 16 Aug 2024 17:46:55 +0800
Subject: [PATCH] update rust sample code

---
 .../rust/nativeexample/examples/tmq.rs         | 18 +++++++++++++++---
 docs/examples/rust/restexample/examples/tmq.rs | 18 +++++++++++++++---
 2 files changed, 30 insertions(+), 6 deletions(-)

diff --git a/docs/examples/rust/nativeexample/examples/tmq.rs b/docs/examples/rust/nativeexample/examples/tmq.rs
index 2f8a9e5995..49580f81b7 100644
--- a/docs/examples/rust/nativeexample/examples/tmq.rs
+++ b/docs/examples/rust/nativeexample/examples/tmq.rs
@@ -86,7 +86,7 @@ async fn main() -> anyhow::Result<()> {
     match consumer.subscribe([topic]).await{
         Ok(_) => println!("Subscribe topics successfully."),
         Err(err) => {
-            eprintln!("Failed to subscribe topic_meters, ErrMessage: {:?}", err);
+            eprintln!("Failed to subscribe topic: {}, groupId: {}, clientId: {}, ErrMessage: {:?}", topic, group_id, client_id, err);
             return Err(err.into());
         }
     }
@@ -164,7 +164,13 @@ async fn main() -> anyhow::Result<()> {
 
 
     // ANCHOR: seek_offset
-    let assignments = consumer.assignments().await.unwrap();
+    let assignments = match consumer.assignments().await{
+        Some(assignments) => assignments,
+        None => {
+            eprintln!("Failed to get assignments.");
+            return Err(anyhow::anyhow!("Failed to get assignments. topic: {}, groupId: {}, clientId: {}", topic, group_id, client_id));
+        }
+    };
     println!("assignments: {:?}", assignments);
 
     // seek offset
@@ -200,7 +206,13 @@ async fn main() -> anyhow::Result<()> {
     }
     println!("Assignment seek to beginning successfully.");
     // after seek offset
-    let assignments = consumer.assignments().await.unwrap();
+    let assignments = match consumer.assignments().await{
+        Some(assignments) => assignments,
+        None => {
+            eprintln!("Failed to get assignments.");
+            return Err(anyhow::anyhow!("Failed to get assignments. topic: {}, groupId: {}, clientId: {}", topic, group_id, client_id));
+        }
+    };
     println!("After seek offset assignments: {:?}", assignments);
     // ANCHOR_END: seek_offset
 
diff --git a/docs/examples/rust/restexample/examples/tmq.rs b/docs/examples/rust/restexample/examples/tmq.rs
index 670ae0ca3c..86715d57cb 100644
--- a/docs/examples/rust/restexample/examples/tmq.rs
+++ b/docs/examples/rust/restexample/examples/tmq.rs
@@ -86,7 +86,7 @@ async fn main() -> anyhow::Result<()> {
     match consumer.subscribe([topic]).await{
         Ok(_) => println!("Subscribe topics successfully."),
         Err(err) => {
-            eprintln!("Failed to subscribe topic_meters, ErrMessage: {:?}", err);
+            eprintln!("Failed to subscribe topic: {}, groupId: {}, clientId: {}, ErrMessage: {:?}", topic, group_id, client_id, err);
             return Err(err.into());
         }
     }
@@ -164,7 +164,13 @@ async fn main() -> anyhow::Result<()> {
 
 
     // ANCHOR: seek_offset
-    let assignments = consumer.assignments().await.unwrap();
+    let assignments = match consumer.assignments().await{
+        Some(assignments) => assignments,
+        None => {
+            eprintln!("Failed to get assignments.");
+            return Err(anyhow::anyhow!("Failed to get assignments. topic: {}, groupId: {}, clientId: {}", topic, group_id, client_id));
+        }
+    };
     println!("assignments: {:?}", assignments);
 
     // seek offset
@@ -200,7 +206,13 @@ async fn main() -> anyhow::Result<()> {
     }
     println!("Assignment seek to beginning successfully.");
     // after seek offset
-    let assignments = consumer.assignments().await.unwrap();
+    let assignments = match consumer.assignments().await{
+        Some(assignments) => assignments,
+        None => {
+            eprintln!("Failed to get assignments.");
+            return Err(anyhow::anyhow!("Failed to get assignments. topic: {}, groupId: {}, clientId: {}", topic, group_id, client_id));
+        }
+    };
     println!("After seek offset assignments: {:?}", assignments);
     // ANCHOR_END: seek_offset