Make sure missed [server] heartbeats trigger connection recovery

michaelklishin · michaelklishin · commit ffb25593153f · 2015-09-23T23:54:06.000+03:00
Fixes rabbitmq#57.
diff --git a/src/com/rabbitmq/client/impl/recovery/AutorecoveringConnection.java b/src/com/rabbitmq/client/impl/recovery/AutorecoveringConnection.java
@@ -5,6 +5,7 @@
 import com.rabbitmq.client.BlockedListener;
 import com.rabbitmq.client.Channel;
 import com.rabbitmq.client.Connection;
+import com.rabbitmq.client.MissedHeartbeatException;
 import com.rabbitmq.client.Recoverable;
 import com.rabbitmq.client.RecoveryListener;
 import com.rabbitmq.client.ShutdownListener;
@@ -378,7 +379,7 @@ private void addAutomaticRecoveryListener() {
         ShutdownListener automaticRecoveryListener = new ShutdownListener() {
             public void shutdownCompleted(ShutdownSignalException cause) {
                 try {
-                    if (!cause.isInitiatedByApplication()) {
+                    if (shouldTriggerConnectionRecovery(cause)) {
                         c.beginAutomaticRecovery();
                     }
                 } catch (Exception e) {
@@ -394,6 +395,10 @@ public void shutdownCompleted(ShutdownSignalException cause) {
         }
     }
 
+    protected boolean shouldTriggerConnectionRecovery(ShutdownSignalException cause) {
+        return !cause.isInitiatedByApplication() || (cause.getCause() instanceof MissedHeartbeatException);
+    }
+
     /**
      * Not part of the public API. Mean to be used by JVM RabbitMQ clients that build on
      * top of the Java client and need to be notified when server-named queue name changes
