[inner-2321] fix: explain statement contained in multiple-sql appears lost connection...

(cherry picked from commit 02f9ee6b60)
This commit is contained in:
wenyh
2023-08-08 10:54:58 +08:00
committed by wenyh1
parent bf01cf6703
commit 0d5a5bc550

View File

@@ -273,25 +273,25 @@ public final class ExplainHandler {
ByteBuffer buffer = service.allocate();
// writeDirectly header
ResultSetHeaderPacket header = PacketUtil.getHeader(FIELD_COUNT);
byte packetId = header.getPacketId();
header.setPacketId(service.nextPacketId());
buffer = header.write(buffer, service, true);
// writeDirectly fields
for (FieldPacket field : FIELDS) {
field.setPacketId(++packetId);
field.setPacketId(service.nextPacketId());
buffer = field.write(buffer, service, true);
}
// writeDirectly eof
EOFPacket eof = new EOFPacket();
eof.setPacketId(++packetId);
eof.setPacketId(service.nextPacketId());
buffer = eof.write(buffer, service, true);
if (!rrs.isNeedOptimizer()) {
// writeDirectly rows
for (RouteResultsetNode node : rrs.getNodes()) {
RowDataPacket row = getRow(node, service.getCharset().getResults());
row.setPacketId(++packetId);
row.setPacketId(service.nextPacketId());
buffer = row.write(buffer, service, true);
}
} else {
@@ -299,7 +299,7 @@ public final class ExplainHandler {
RouteResultsetNode routeSingleNode = getTryRouteSingleNode(builder, rrs);
if (routeSingleNode != null) {
RowDataPacket row = getRow(routeSingleNode, service.getCharset().getResults());
row.setPacketId(++packetId);
row.setPacketId(service.nextPacketId());
buffer = row.write(buffer, service, true);
} else {
List<ReferenceHandlerInfo> results = ComplexQueryPlanUtil.getComplexQueryResult(builder);
@@ -308,14 +308,14 @@ public final class ExplainHandler {
row.add(StringUtil.encode(getRowStr(result.getName(), result.isIndentation()), service.getCharset().getResults()));
row.add(StringUtil.encode(getRowStr(result.getType(), result.isIndentation()), service.getCharset().getResults()));
row.add(StringUtil.encode(getRowStr(result.getRefOrSQL(), result.isIndentation()), service.getCharset().getResults()));
row.setPacketId(++packetId);
row.setPacketId(service.nextPacketId());
buffer = row.write(buffer, service, true);
}
}
}
// writeDirectly last eof
EOFRowPacket lastEof = new EOFRowPacket();
lastEof.setPacketId(++packetId);
lastEof.setPacketId(service.nextPacketId());
lastEof.write(buffer, service);
}