fix test

Signed-off-by: disksing <[email protected]>
pingcap · Oct 30, 2023 · c8c69ce · c8c69ce
1 parent b353ff0
commit c8c69ce
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/pkg/proxy/backend/authenticator.go b/pkg/proxy/backend/authenticator.go
@@ -226,19 +226,22 @@ loop:
 			}
 			return err
 		}
+		var packetErr error
 		if serverPkt[0] == pnet.ErrHeader.Byte() {
-			err = pnet.ParseErrorPacket(serverPkt)
-			if handshakeHandler.HandleHandshakeErr(cctx, err.(*gomysql.MyError)) {
+			packetErr = pnet.ParseErrorPacket(serverPkt)
+			if handshakeHandler.HandleHandshakeErr(cctx, packetErr.(*gomysql.MyError)) {
 				logger.Warn("handle handshake error, start reconnect", zap.Error(err))
 				backendIO.Close()
 				goto RECONNECT
 			}
-			return err
 		}
 		err = clientIO.WritePacket(serverPkt, true)
 		if err != nil {
 			return err
 		}
+		if packetErr != nil {
+			return packetErr
+		}
 
 		pktIdx++
 		switch serverPkt[0] {