diff --git a/src/main/java/com/actiontech/dble/server/handler/ExplainHandler.java b/src/main/java/com/actiontech/dble/server/handler/ExplainHandler.java index 3f5b8ccadc..5157216854 100644 --- a/src/main/java/com/actiontech/dble/server/handler/ExplainHandler.java +++ b/src/main/java/com/actiontech/dble/server/handler/ExplainHandler.java @@ -43,6 +43,7 @@ import java.sql.SQLException; import java.sql.SQLNonTransientException; import java.sql.SQLSyntaxErrorException; +import java.util.LinkedList; import java.util.List; /** @@ -200,6 +201,7 @@ private static boolean isInsertSeq(ShardingService service, String stmt, SchemaC } public static void writeOutHeadAndEof(ShardingService service, RouteResultset rrs) { + final List rows = getRows(service, rrs); ByteBuffer buffer = service.allocate(); // writeDirectly header ResultSetHeaderPacket header = PacketUtil.getHeader(FIELD_COUNT); @@ -217,12 +219,22 @@ public static void writeOutHeadAndEof(ShardingService service, RouteResultset rr eof.setPacketId(service.nextPacketId()); buffer = eof.write(buffer, service, true); + // writeDirectly rows + for (RowDataPacket row : rows) { + row.setPacketId(service.nextPacketId()); + buffer = row.write(buffer, service, true); + } + + // writeDirectly last eof + EOFRowPacket lastEof = new EOFRowPacket(); + lastEof.setPacketId(service.nextPacketId()); + lastEof.write(buffer, service); + } + private static List getRows(ShardingService service, RouteResultset rrs) { + LinkedList rows = new LinkedList<>(); if (!rrs.isNeedOptimizer()) { - // writeDirectly rows for (RouteResultsetNode node : rrs.getNodes()) { - RowDataPacket row = getRow(node, service.getCharset().getResults()); - row.setPacketId(service.nextPacketId()); - buffer = row.write(buffer, service, true); + rows.add(getRow(node, service.getCharset().getResults())); } } else { BaseHandlerBuilder builder = buildNodes(rrs, service); @@ -233,12 +245,9 @@ public static void writeOutHeadAndEof(ShardingService service, RouteResultset rr row.add(StringUtil.encode(result.getType(), service.getCharset().getResults())); row.add(StringUtil.encode(result.getRefOrSQL(), service.getCharset().getResults())); row.setPacketId(service.nextPacketId()); - buffer = row.write(buffer, service, true); + rows.add(row); } } - // writeDirectly last eof - EOFRowPacket lastEof = new EOFRowPacket(); - lastEof.setPacketId(service.nextPacketId()); - lastEof.write(buffer, service); + return rows; } }