Benchmark should not panic on server close

nats-io · May 17, 2023 · 3b8d39f · 3b8d39f
1 parent 3c71104
commit 3b8d39f
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/async-nats/benches/core_nats.rs b/async-nats/benches/core_nats.rs
@@ -82,9 +82,12 @@ pub fn subscribe(c: &mut Criterion) {
                             let msg = &bmsg[0..*size].to_vec();
 
                             loop {
-                                nc.publish("bench".to_string(), msg.clone().into())
-                                    .await
-                                    .unwrap();
+                                let result = nc.publish("bench".to_string(), msg.clone().into())
+                                    .await;
+
+                                if result.is_err() {
+                                    break
+                                }
                             }
                         }
                     });