Skip to content

Commit

Permalink
4
Browse files Browse the repository at this point in the history
  • Loading branch information
xinyiZzz committed Dec 17, 2024
1 parent 4b6a18d commit 47414e2
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -272,10 +272,10 @@ public TReportExecStatusResult reportExecStatus(TReportExecStatusParams params,
}
ConnectContext ctx = info.getConnectContext();
if (ctx.getConnectType() == ConnectType.ARROW_FLIGHT_SQL && params.isDone()) {
// `params.isDone` may be true in multiple reportExecStatus.
if (ctx.getIsFlightSqlConnectProcessorClose() && !ctx.getFinalizeArrowFlightSqlRequestFinished()) {
ctx.finalizeArrowFlightSqlRequest();
}
// // `params.isDone` may be true in multiple reportExecStatus.
// if (ctx.getIsFlightSqlConnectProcessorClose() && !ctx.getFinalizeArrowFlightSqlRequestFinished()) {
// ctx.finalizeArrowFlightSqlRequest();
// }
ctx.setIsExecStatusDone();
}
result.setStatus(new TStatus(TStatusCode.OK));
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -193,9 +193,9 @@ public void fetchArrowFlightSchema(int timeoutMs) {

@Override
public void close() throws Exception {
if (ctx.getIsExecStatusDone()) {
ctx.finalizeArrowFlightSqlRequest();
}
// if (ctx.getIsExecStatusDone()) {
ctx.finalizeArrowFlightSqlRequest();
// }
ctx.setIsFlightSqlConnectProcessorClose();
ConnectContext.remove();
}
Expand Down

0 comments on commit 47414e2

Please sign in to comment.