jffwang
/
hetu-core
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Pre Merge pull request !1531 from Kishore Kumar M V/0rows_after_recovery

This commit is contained in:
Kishore Kumar M V 2022-06-23 05:55:30 +00:00 committed by Gitee
parent 8289812e65 2876fc31f0
commit f12fa68b38
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
presto-main/src/main/java/io/prestosql/execution/scheduler/SqlQueryScheduler.java
View File

@ -97,6 +97,7 @@ import static io.prestosql.SystemSessionProperties.getConcurrentLifespansPerNode
import static io.prestosql.SystemSessionProperties.getWriterMinSize;
import static io.prestosql.SystemSessionProperties.isReuseTableScanEnabled;
import static io.prestosql.execution.BasicStageStats.aggregateBasicStageStats;
import static io.prestosql.execution.QueryState.RECOVERING;
import static io.prestosql.execution.SqlStageExecution.createSqlStageExecution;
import static io.prestosql.execution.StageState.ABORTED;
import static io.prestosql.execution.StageState.CANCELED;
@ -106,7 +107,6 @@ import static io.prestosql.execution.StageState.RUNNING;
import static io.prestosql.execution.StageState.SCHEDULED;
import static io.prestosql.execution.scheduler.SourcePartitionedScheduler.newSourcePartitionedSchedulerAsStageScheduler;
import static io.prestosql.snapshot.RecoveryConfig.calculateTaskCount;
import static io.prestosql.snapshot.RecoveryState.STOPPING_FOR_RESCHEDULE;
import static io.prestosql.spi.StandardErrorCode.GENERIC_INTERNAL_ERROR;
import static io.prestosql.spi.StandardErrorCode.NO_NODES_AVAILABLE;
import static io.prestosql.spi.connector.CatalogName.isInternalSystemConnector;
@ -859,7 +859,7 @@ public class SqlQueryScheduler
RuntimeException closeError = new RuntimeException();
for (StageScheduler scheduler : stageSchedulers.values()) {
try {
if (queryRecoveryManager.getState() != STOPPING_FOR_RESCHEDULE) {
if (queryStateMachine.getQueryState() != RECOVERING && !queryRecoveryManager.isRecovering()) {
scheduler.close();
}
}