[inner-1762] fix:no handler (#3257)

(cherry picked from commit 16c317b9f7)
This commit is contained in:
wenyh
2022-05-24 11:45:12 +08:00
committed by wd2365151147
parent bf4b5a6f90
commit 5276c9408d
3 changed files with 12 additions and 9 deletions
@@ -138,9 +138,10 @@ public class DefaultResponseHandler implements ProtocolResponseHandler {
service.getSession().setBackendResponseEndTime(this.service);
}
service.getLogResponse().set(false);
service.backendSpecialCleanUp();
if (service.getResponseHandler() != null) {
service.getResponseHandler().rowEofResponse(data, false, service);
ResponseHandler respHand = service.getResponseHandler();
if (respHand != null) {
service.backendSpecialCleanUp();
respHand.rowEofResponse(data, false, service);
} else {
closeNoHandler();
}
@@ -55,9 +55,10 @@ public class FetchResponseHandler implements ProtocolResponseHandler {
}
private void handleRowEofPacket(byte[] data) {
service.backendSpecialCleanUp();
if (service.getResponseHandler() != null) {
service.getResponseHandler().rowEofResponse(data, false, service);
ResponseHandler respHand = service.getResponseHandler();
if (respHand != null) {
service.backendSpecialCleanUp();
respHand.rowEofResponse(data, false, service);
} else {
closeNoHandler();
}
@@ -91,10 +91,11 @@ public class LoadDataResponseHandler extends DefaultResponseHandler {
if (service.getSession() != null && !service.isTesting() && service.getLogResponse().compareAndSet(false, true)) {
service.getSession().setBackendResponseEndTime(this.service);
}
ResponseHandler respHand = service.getResponseHandler();
service.getLogResponse().set(false);
service.backendSpecialCleanUp();
if (service.getResponseHandler() != null) {
service.getResponseHandler().rowEofResponse(data, false, service);
if (respHand != null) {
service.backendSpecialCleanUp();
respHand.rowEofResponse(data, false, service);
} else {
closeNoHandler();
}