[CELEBORN-2276] Fix race condition in MemoryManager.releaseSortMemory

[cxzl25]

What changes were proposed in this pull request?

Use updateAndGet

Why are the changes needed?

reserveSortMemory calls sortMemoryCounter.addAndGet without any synchronized, so the lock in releaseSortMemory doesn't actually protect against races with reserveSortMemory anyway.

Does this PR resolve a correctness bug?

No

Does this PR introduce any user-facing change?

No

How was this patch tested?

GHA

[eolivelli]

The change looks good

can you please add dedicated unit tests around this piece of code ?
it would be great to have a test that updates the counter using reserveSortMemory and releaseSortMemory from different threads to demostrate the bug and the fix

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 66.88%. Comparing base (2dd1b7a) to head (b26b906).
⚠️ Report is 35 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3620      +/-   ##
==========================================
- Coverage   67.13%   66.88%   -0.24%     
==========================================
  Files         357      357              
  Lines       21860    21967     +107     
  Branches     1943     1949       +6     
==========================================
+ Hits        14674    14691      +17     
- Misses       6166     6262      +96     
+ Partials     1020     1014       -6     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
• 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.