Skip to content

[AURON #2052] [BUG] NPE risk in AuronUniffleShuffleReader.#2053

Merged
zuston merged 1 commit intoapache:masterfrom
slfan1989:auron-2052
Mar 4, 2026
Merged

[AURON #2052] [BUG] NPE risk in AuronUniffleShuffleReader.#2053
zuston merged 1 commit intoapache:masterfrom
slfan1989:auron-2052

Conversation

@slfan1989
Copy link
Contributor

@slfan1989 slfan1989 commented Mar 1, 2026

Which issue does this PR close?

Closes #2052

Rationale for this change

In AuronUniffleShuffleReader, there are potential NullPointerException risks when handling partitions:

  • partitionToExpectBlocks.get(partition) may return null, but the code directly calls .isEmpty() without null check
  • partitionToShuffleServers.get(partition) may return null, but the code uses it directly without validationThis can cause the shuffle reader to crash when reading partitions that have missing or incomplete metadata.

This can cause the shuffle reader to crash when reading partitions that have missing or incomplete metadata.

What changes are included in this PR?

Are there any user-facing changes?

No.

How was this patch tested?

Exists Junit Test.

Signed-off-by: slfan1989 <slfan1989@apache.org>
@slfan1989
Copy link
Contributor Author

@zuston @lihao712 @cxzl25 Can you please help review this PR? Thank you very much!

Copy link
Member

@zuston zuston left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm. thanks @slfan1989

@zuston zuston merged commit f4ef044 into apache:master Mar 4, 2026
209 of 210 checks passed
@slfan1989
Copy link
Contributor Author

lgtm. thanks @slfan1989

@zuston Thank you very much for helping to review the code!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[BUG] NPE risk in AuronUniffleShuffleReader

2 participants