Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
4
Browse files Browse the repository at this point in the history
xinyiZzz committed Dec 17, 2024
1 parent 4b6a18d commit 47414e2
Showing 2 changed files with 7 additions and 7 deletions.
Original file line number Diff line number Diff line change
@@ -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));
Original file line number Diff line number Diff line change
@@ -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();
}

0 comments on commit 47414e2

Please sign in to comment.