Handle SocketException on writes

Those exceptions are skipped when automatic connection recovery is on.

[#159130745]

References #106

Author: acogoluegnes

src/main/java/com/rabbitmq/perf/AgentBase.java

@@ -21,6 +21,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.io.IOException;
+import java.net.SocketException;
 import java.util.function.Predicate;
 
 /**
@@ -83,7 +84,16 @@ protected void dealWithWriteOperation(WriteOperation writeOperation, Recovery.Re
             writeOperation.call();
         } catch (ShutdownSignalException e) {
             handleShutdownSignalExceptionOnWrite(recoveryProcess, e);
-            // connection recovery is in progress, we ignore the exception if this point is reached
+        } catch (SocketException e) {
+            if (recoveryProcess.isEnabled()) {
+                if (LOGGER.isDebugEnabled()) {
+                    LOGGER.debug(
+                        "Socket exception in write, recovery process is enabled, ignoring to let connection recovery carry on"
+                    );
+                }
+            } else {
+                throw e;
+            }
         }
     }