refactor MySQLResponseHandler.channelRead (28b18d9d) · Commits · SUMMER2020 / students / proj-2004142

sharding-proxy/src/main/java/io/shardingsphere/proxy/backend/netty/client/response/mysql/MySQLResponseHandler.java

+38 −23

Original line number	Diff line number	Diff line
		@@ -66,29 +66,21 @@ public final class MySQLResponseHandler extends ResponseHandler {
		public void channelRead(final ChannelHandlerContext context, final Object message) {
		MySQLPacketPayload payload = new MySQLPacketPayload((ByteBuf) message);
		int header = getHeader(payload);
		if (AuthType.UN_AUTH == authType) {
		switch (authType) {
		case UN_AUTH:
		auth(context, payload);
		authType = AuthType.AUTHING;
		} else if (AuthType.AUTHING == authType) {
		if (OKPacket.HEADER == header) {
		okPacket(context, payload);
		authType = AuthType.AUTH_SUCCESS;
		} else {
		errPacket(context, payload);
		authType = AuthType.AUTH_FAILED;
		}
		} else if (AuthType.AUTH_FAILED == authType) {
		return;
		case AUTHING:
		authing(context, payload, header);
		return;
		case AUTH_SUCCESS:
		authSuccess(context, payload, header);
		return;
		case AUTH_FAILED:
		log.error("mysql auth failed, cannot handle channel read message");
		} else {
		if (EofPacket.HEADER == header) {
		eofPacket(context, payload);
		} else if (OKPacket.HEADER == header) {
		okPacket(context, payload);
		} else if (ErrPacket.HEADER == header) {
		errPacket(context, payload);
		} else {
		commonPacket(context, payload);
		}
		return;
		default:
		throw new UnsupportedOperationException(authType.name());
		}
		}

		@@ -115,6 +107,29 @@ public final class MySQLResponseHandler extends ResponseHandler {
		} finally {
		payload.close();
		}
		authType = AuthType.AUTHING;
		}

		private void authing(final ChannelHandlerContext context, final MySQLPacketPayload payload, final int header) {
		if (OKPacket.HEADER == header) {
		okPacket(context, payload);
		authType = AuthType.AUTH_SUCCESS;
		} else {
		errPacket(context, payload);
		authType = AuthType.AUTH_FAILED;
		}
		}

		private void authSuccess(final ChannelHandlerContext context, final MySQLPacketPayload payload, final int header) {
		if (EofPacket.HEADER == header) {
		eofPacket(context, payload);
		} else if (OKPacket.HEADER == header) {
		okPacket(context, payload);
		} else if (ErrPacket.HEADER == header) {
		errPacket(context, payload);
		} else {
		commonPacket(context, payload);
		}
		}

		@Override