Skip to content

Conversation

@LZD-PratyushBhatt
Copy link
Collaborator

@LZD-PratyushBhatt LZD-PratyushBhatt commented Dec 18, 2025

Issues

  • My PR addresses the following Helix issues and references them in the PR description:
    Fixes a NullPointerException in WagedInstanceCapacity that occurs when an instance config is deleted while partitions are still assigned to it. The exception prevents the controller pipeline from completing, blocking rebalancing and leaving stale assignments in IdealState.

(apache#200 - Link your issue number here: You can write "Fixes #XXX". Please use the proper keyword so that the issue gets closed automatically. See https://docs.github.com/en/github/managing-your-work-on-github/linking-a-pull-request-to-an-issue
Any of the following keywords can be used: close, closes, closed, fix, fixes, fixed, resolve, resolves, resolved)

Description

When an instance config is deleted:

  1. The instance is removed from _instanceCapacityMap during initialization
  2. CurrentState may still have partitions assigned to that instance
  3. checkAndReduceInstanceCapacity() access instanceCapacity.keySet() on a null value
  4. This causes a NullPointerException that crashes the DEFAULT pipeline
  5. The rebalancer never runs, so the deleted instance remains in IdealState assignments

Added null checks in methods to handle deleted instance configs:

  • checkAndReduceInstanceCapacity(): Returns false and logs when instance config is missing
    This allows the pipeline to continue, and the rebalancer removes the deleted instance from assignments in the next cycle.

  • Here are some details about my PR, including screenshots of any UI changes:

(Write a concise description including what, why, how)

Tests

  • Reproduced the issue locally by deleting an instance config while it had partition assignments

  • Verified the fix prevents the NPE and allows rebalancing to proceed

  • Confirmed the deleted instance is removed from IdealState after rebalancing

  • The following tests are written for this issue:

(List the names of added unit/integration tests)

  • The following is the result of the "mvn test" command on the appropriate module:

(If CI test fails due to known issue, please specify the issue and test PR locally. Then copy & paste the result of "mvn test" to here.)

Changes that Break Backward Compatibility (Optional)

  • My PR contains changes that break backward compatibility or previous assumptions for certain methods or API. They include:

(Consider including all behavior changes for public methods or API. Also include these changes in merge description so that other developers are aware of these changes. This allows them to make relevant code changes in feature branches accounting for the new method/API behavior.)

Documentation (Optional)

  • In case of new functionality, my PR adds documentation in the following wiki page:

(Link the GitHub wiki you added)

Commits

  • My commits all reference appropriate Apache Helix GitHub issues in their subject lines. In addition, my commits follow the guidelines from "How to write a good git commit message":
    1. Subject is separated from body by a blank line
    2. Subject is limited to 50 characters (not including Jira issue reference)
    3. Subject does not end with a period
    4. Subject uses the imperative mood ("add", not "adding")
    5. Body wraps at 72 characters
    6. Body explains "what" and "why", not "how"

Code Quality

  • My diff has been formatted using helix-style.xml
    (helix-style-intellij.xml if IntelliJ IDE is used)

@LZD-PratyushBhatt LZD-PratyushBhatt force-pushed the lzd/WagedInstanceCapacityNPEFix branch from dc4e302 to 8eeb726 Compare December 18, 2025 17:45
Copy link
Collaborator

@kabragaurav kabragaurav left a comment

Choose a reason for hiding this comment

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

LGTM

@LZD-PratyushBhatt LZD-PratyushBhatt merged commit 772daee into dev Jan 6, 2026
2 of 3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants