[CELEBORN-1661] Make sure that the sortedFilesDb is initialized successfully when worker enable graceful shutdown

### What changes were proposed in this pull request?

### Why are the changes needed?
Similar to CELEBORN-1457, `sortedFilesDb` may also fail to initialize.

### Does this PR introduce _any_ user-facing change?
No

### How was this patch tested?
GA

Closes #2831 from cxzl25/CELEBORN-1661.

Authored-by: sychen <sychen@ctrip.com>
Signed-off-by: Shuang <lvshuang.xjs@alibaba-inc.com>
This commit is contained in:
sychen 2024-10-22 10:19:31 +08:00 committed by Shuang
parent dc5f3fb96b
commit 15d1463be8

View File

@ -125,8 +125,8 @@ public class PartitionFilesSorter extends ShuffleRecoverHelper {
this.sortedFilesDb = DBProvider.initDB(dbBackend, recoverFile, CURRENT_VERSION);
reloadAndCleanSortedShuffleFiles(this.sortedFilesDb);
} catch (Exception e) {
logger.error("Failed to reload DB for sorted shuffle files from: " + recoverFile, e);
this.sortedFilesDb = null;
throw new IllegalStateException(
"Failed to reload DB for sorted shuffle files from: " + recoverFile, e);
}
} else {
this.sortedFilesDb = null;