Merge pull request #3866 from actiontech/3.21.02.99/lts_2382

[inner-2340] fix: explain' lost connection....
actiontech · Feb 26, 2024 · ace71d8 · ace71d8
2 parents 604bfc4 + 7689a73
commit ace71d8
Showing 1 changed file with 18 additions and 10 deletions.
diff --git 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<RowDataPacket> 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<RowDataPacket> getRows(ShardingService service, RouteResultset rrs) {
+        LinkedList<RowDataPacket> 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);
@@ -232,13 +244,9 @@ public static void writeOutHeadAndEof(ShardingService service, RouteResultset rr
                 row.add(StringUtil.encode(result.getName(), service.getCharset().getResults()));
                 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;
     }
 }