Fix: exception on simple consume when disconnected immediately (#1253)

nats-io · Nov 15, 2024 · 2175501 · 2175501
1 parent 1bf06d5
commit 2175501
Showing 1 changed file with 6 additions and 15 deletions.
diff --git a/src/main/java/io/nats/client/impl/NatsMessageConsumer.java b/src/main/java/io/nats/client/impl/NatsMessageConsumer.java
@@ -58,7 +58,8 @@ public void heartbeatError() {
             doSub();
         }
         catch (JetStreamApiException | IOException e) {
-            setupHbAlarmToTrigger();
+            pmm.resetTracking();
+            pmm.initOrResetHeartbeatTimer();
         }
     }
 
@@ -69,20 +70,10 @@ void doSub() throws JetStreamApiException, IOException {
                 finished.set(true);
             }
         };
-        try {
-            super.initSub(subscriptionMaker.subscribe(mh, userDispatcher, pmm, null));
-            repull();
-            stopped.set(false);
-            finished.set(false);
-        }
-        catch (JetStreamApiException | IOException e) {
-            setupHbAlarmToTrigger();
-        }
-    }
-
-    private void setupHbAlarmToTrigger() {
-        pmm.resetTracking();
-        pmm.initOrResetHeartbeatTimer();
+        super.initSub(subscriptionMaker.subscribe(mh, userDispatcher, pmm, null));
+        repull();
+        stopped.set(false);
+        finished.set(false);
     }
 
     @Override