Added case for XAER_RMFAIL (#2132)

Author: tkyc

src/main/java/com/microsoft/sqlserver/jdbc/SQLServerXAResource.java

@@ -5,6 +5,7 @@
 
 package com.microsoft.sqlserver.jdbc;
 
+import java.net.SocketException;
 import java.sql.CallableStatement;
 import java.sql.ResultSet;
 import java.sql.SQLException;
@@ -26,6 +27,7 @@
 
 import com.microsoft.sqlserver.jdbc.SQLServerError.TransientError;
 
+
 /**
  * Implements Transaction id used to recover transactions.
  */
@@ -764,7 +766,7 @@ private XAReturnValue dtc_XA_interface(int nType, Xid xid, int xaFlags) throws X
                 xaLogger.finer(toString() + " exception:" + ex);
 
             if (ex.getMessage().equals(SQLServerException.getErrString("R_noServerResponse"))
-                    || TransientError.isTransientError(ex.getSQLServerError())) {
+                    || TransientError.isTransientError(ex.getSQLServerError()) || isResourceManagerFailure(ex)) {
                 XAException e = new XAException(ex.toString());
                 e.errorCode = XAException.XAER_RMFAIL;
                 throw e;
@@ -943,4 +945,62 @@ private static int nextResourceID() {
         return baseResourceID.incrementAndGet();
     }
 
+    private enum ResourceManagerFailure {
+        CONN_RESET("Connection reset");
+
+        private final String errString;
+
+        ResourceManagerFailure(String errString) {
+            this.errString = errString;
+        }
+
+        @Override
+        public String toString() {
+            return errString;
+        }
+
+        static ResourceManagerFailure fromString(String errString) {
+            for (ResourceManagerFailure resourceManagerFailure : ResourceManagerFailure.values()) {
+                if (errString.equalsIgnoreCase(resourceManagerFailure.toString())) {
+                    return resourceManagerFailure;
+                }
+            }
+            return null;
+        }
+    }
+
+    /**
+     * Check if the root exception of the throwable should be a XAER_RMFAIL exception
+     *
+     * @param throwable
+     *        The exception to check if the root cause should be a XAER_RMFAIL
+     *
+     * @return True if XAER_RMFAIL, otherwise false
+     */
+    private boolean isResourceManagerFailure(Throwable throwable) {
+        Throwable root = Util.getRootCause(throwable);
+
+        if (null == root) {
+            return false;
+        }
+
+        ResourceManagerFailure err = ResourceManagerFailure.fromString(root.getMessage());
+
+        if (null == err) {
+            return false;
+        }
+
+        // Add as needed here for future XAER_RMFAIL exceptions
+        if (root instanceof SocketException) {
+            switch (err) {
+                case CONN_RESET:
+                    return true;
+                default:
+                    return false;
+            }
+        }
+
+        return false;
+    }
+
 }

src/main/java/com/microsoft/sqlserver/jdbc/Util.java

@@ -13,6 +13,8 @@
 import java.net.UnknownHostException;
 import java.text.DecimalFormat;
 import java.text.MessageFormat;
+import java.util.List;
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.Locale;
 import java.util.Properties;
@@ -1043,6 +1045,23 @@ static char[] bytesToChars(byte[] bytes) {
         }
         return chars;
     }
+
+    /**
+     * @param throwable
+     *        The exception to find root cause for.
+     *
+     * @return The root cause of the exception, otherwise null if null throwable input
+     */
+    static Throwable getRootCause(Throwable throwable) {
+        final List<Throwable> list = new ArrayList<>();
+
+        while (throwable != null && !list.contains(throwable)) {
+            list.add(throwable);
+            throwable = throwable.getCause();
+        }
+
+        return list.isEmpty() ? null : list.get(list.size() - 1);
+    }
 }