From 0e17d5b2ae93eea5fef18bdaee61eb2dd1b34b42 Mon Sep 17 00:00:00 2001
From: Casper Beyer <caspervonb@icloud.com>
Date: Wed, 17 May 2023 05:19:08 -0400
Subject: [PATCH] Benchmark should not panic on server close

---
 async-nats/benches/core_nats.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/async-nats/benches/core_nats.rs b/async-nats/benches/core_nats.rs
index ca3a275e0..a5d698ef6 100644
--- a/async-nats/benches/core_nats.rs
+++ b/async-nats/benches/core_nats.rs
@@ -60,7 +60,6 @@ pub fn publish(c: &mut Criterion) {
 
 pub fn subscribe(c: &mut Criterion) {
     let server = nats_server::run_basic_server();
-
     let mut subscribe_amount_group = c.benchmark_group("subscribe amount");
     subscribe_amount_group.sample_size(30);
     subscribe_amount_group.warm_up_time(std::time::Duration::from_secs(1));
@@ -82,9 +81,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;
+                                }
                             }
                         }
                     });