testConn #527

actiontech · May 3, 2018 · 5b40da6 · 5b40da6
1 parent 8422ab5
commit 5b40da6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/actiontech/dble/backend/mysql/nio/MySQLDataSource.java b/src/main/java/com/actiontech/dble/backend/mysql/nio/MySQLDataSource.java
@@ -17,6 +17,7 @@
 
 import java.io.*;
 import java.net.Socket;
+import java.nio.charset.StandardCharsets;
 import java.security.NoSuchAlgorithmException;
 
 /**
@@ -101,6 +102,11 @@ public boolean testConnection(String schema) throws IOException {
             BinaryPacket bin1 = new BinaryPacket();
             bin1.read(in);
 
+            if (bin1.getData()[0] == ErrorPacket.FIELD_COUNT) {
+                ErrorPacket err = new ErrorPacket();
+                err.read(bin1);
+                throw new RuntimeException(new String(err.getMessage(), StandardCharsets.UTF_8));
+            }
             HandshakeV10Packet handshake = new HandshakeV10Packet();
             handshake.read(bin1);