Set SQL state when there is an exception #1385 (#1386)

Author: DavideD

vertx-mssql-client/src/main/java/io/vertx/mssqlclient/MSSQLException.java

@@ -30,7 +30,7 @@ public class MSSQLException extends DatabaseException {
   private List<MSSQLException> additional;
 
   public MSSQLException(int number, byte state, byte severity, String errorMessage, String serverName, String procedureName, int lineNumber) {
-    super(formatMessage(number, state, severity, errorMessage, serverName, procedureName, lineNumber), number, null);
+    super(formatMessage(number, state, severity, errorMessage, serverName, procedureName, lineNumber), number, generateStateCode(number, state));
     this.state = state;
     this.severity = severity;
     this.errorMessage = errorMessage;
@@ -39,6 +39,41 @@ public MSSQLException(int number, byte state, byte severity, String errorMessage
     this.lineNumber = lineNumber;
   }
 
+
+  /**
+   * Generates the JDBC state code based on the error number returned from the database.
+   *
+   * This method is derived from the method with the same name in the JDBC driver
+   * in com.microsoft.sqlserver.jdbc.SQLServerException
+   *
+   * @param errNum
+   *        the vendor error number
+   * @param databaseState
+   *        the database state
+   * @return the SQL state code (XOPEN or SQL:2003 conventions)
+   */
+  static String generateStateCode(int errNum, int databaseState) {
+      switch (errNum) {
+        // case 18456: return EXCEPTION_XOPEN_CONNECTION_CANT_ESTABLISH; //username password wrong at login
+        case 8152:
+          return "22001"; // String data right truncation
+        case 515: // 2.2705
+        case 547:
+        case 2601:
+        case 2627:
+          return "23000"; // Integrity constraint violation
+        case 2714:
+          return "S0001"; // table already exists
+        case 208:
+          return "S0002"; // table not found
+        case 1205:
+          return "40001"; // deadlock detected
+        default: {
+          return "S" + String.format( "%4s", databaseState ).replaceAll( " ", "0" );
+        }
+      }
+  }
+
   public void add(MSSQLException e) {
     if (additional == null) {
       additional = new ArrayList<>(3);

vertx-mssql-client/src/test/java/io/vertx/mssqlclient/MSSQLExceptionTest.java

@@ -0,0 +1,57 @@
+package io.vertx.mssqlclient;
+
+import java.util.Arrays;
+import java.util.Collection;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+import static org.junit.Assert.assertEquals;
+
+/**
+ * Test the generation of a SQL state code given a vendor error code
+ */
+@RunWith(Parameterized.class)
+public class MSSQLExceptionTest {
+
+  @Parameterized.Parameters
+  public static Collection<Object[]> data() {
+    return Arrays.asList( new Object[][] {
+      // 22001: String data right truncation
+      {8152, 1, "22001"},
+      // 23000: Integrity constraint violation
+      {515, 1, "23000"},
+      {547, 1, "23000"},
+      {2601, 1, "23000"},
+      {2627, 1, "23000"},
+      // S0001: Integrity constraint violation
+      {2714, 1, "S0001"},
+      // S0002: table not found
+      {208, 1, "S0002"},
+      // deadlock detected
+      {1205, 1, "40001"},
+      {-100, 54, "S0054"},
+      {0, 5, "S0005"},
+      {0, 127, "S0127"},
+      {0, 0, "S0000"}
+    } );
+  }
+
+  private int number;
+  private byte databaseState;
+  private String expectedSqlCode;
+
+  public MSSQLExceptionTest(int vendorCode, int state, String sqlStateCode) {
+    this.number = vendorCode;
+    this.databaseState = (byte) state;
+    this.expectedSqlCode = sqlStateCode;
+  }
+
+  @Test
+  public void testSqlStateCodes() {
+    MSSQLException mssqlException = new MSSQLException( number, databaseState, (byte) 0, null, null, null, 1 );
+    assertEquals( expectedSqlCode, mssqlException.getSqlState() );
+    assertEquals( number, mssqlException.getErrorCode() );
+  }
+}