mirror/dble
1
0
Fork 0
mirror of https://github.com/actiontech/dble.git synced 2025-12-31 09:50:12 -06:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3866 from actiontech/3.21.02.99/lts_2382

Browse Source 
[inner-2340] fix: explain' lost connection....
This commit is contained in:
wenyh
2024-02-26 10:54:05 +08:00
committed by GitHub
parent 604bfc480e 7689a732e8
commit ace71d864e
1 changed files with 18 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/main/java/com/actiontech/dble/server/handler/ExplainHandler.java
View File

@@ -43,6 +43,7 @@ import java.nio.ByteBuffer;
import java.sql.SQLException;
import java.sql.SQLNonTransientException;
import java.sql.SQLSyntaxErrorException;
import java.util.LinkedList;
import java.util.List;
/**
@@ -200,6 +201,7 @@ public final class ExplainHandler {
}
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 final class ExplainHandler {
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 final class ExplainHandler {
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;
}
}