[inner-2321] fix: explain statement contained in multiple-sql appears…

… lost connection... (cherry picked from commit 02f9ee6)
actiontech · Jan 14, 2024 · 00349ea · 00349ea
1 parent 5eb3652
commit 00349ea
Showing 1 changed file with 6 additions and 6 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
@@ -203,25 +203,25 @@ public static void writeOutHeadAndEof(ShardingService service, RouteResultset rr
         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 {
@@ -232,13 +232,13 @@ 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(++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);
     }
 }