(celeborn) branch main updated: [CELEBORN-1662] Handle PUSH_DATA_FAIL_PARTITION_NOT_FOUND in getPushDataFailCause

[nicholasjiang]

This is an automated email from the ASF dual-hosted git repository.

nicholasjiang pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/celeborn.git


The following commit(s) were added to refs/heads/main by this push:
     new dc5f3fb96 [CELEBORN-1662] Handle PUSH_DATA_FAIL_PARTITION_NOT_FOUND in 
getPushDataFailCause
dc5f3fb96 is described below

commit dc5f3fb96b608eb6eff4aec30f69f2c16ae5b1b2
Author: jiang13021 <jiangyanze....@antgroup.com>
AuthorDate: Mon Oct 21 21:06:06 2024 +0800

    [CELEBORN-1662] Handle PUSH_DATA_FAIL_PARTITION_NOT_FOUND in 
getPushDataFailCause
    
    ### What changes were proposed in this pull request?
    Add a condition at the start of the failure cause logic to check for 
PUSH_DATA_FAIL_PARTITION_NOT_FOUND.
    
    ### Why are the changes needed?
    Currently, the getPushDataFailCause method does not identify and handle the 
PUSH_DATA_FAIL_PARTITION_NOT_FOUND error type. All other failure causes are 
explicitly checked and managed, but this specific error type is overlooked.
    
    ### Does this PR introduce _any_ user-facing change?
    No
    
    ### How was this patch tested?
    Manual test
    
    Closes #2833 from jiang13021/celeborn-1662.
    
    Authored-by: jiang13021 <jiangyanze....@antgroup.com>
    Signed-off-by: SteNicholas <programg...@163.com>
---
 client/src/main/java/org/apache/celeborn/client/ShuffleClientImpl.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git 
a/client/src/main/java/org/apache/celeborn/client/ShuffleClientImpl.java 
b/client/src/main/java/org/apache/celeborn/client/ShuffleClientImpl.java
index 31677f7b8..b43bd9598 100644
--- a/client/src/main/java/org/apache/celeborn/client/ShuffleClientImpl.java
+++ b/client/src/main/java/org/apache/celeborn/client/ShuffleClientImpl.java
@@ -1888,6 +1888,8 @@ public class ShuffleClientImpl extends ShuffleClient {
       cause = StatusCode.PUSH_DATA_PRIMARY_WORKER_EXCLUDED;
     } else if 
(message.startsWith(StatusCode.PUSH_DATA_REPLICA_WORKER_EXCLUDED.name())) {
       cause = StatusCode.PUSH_DATA_REPLICA_WORKER_EXCLUDED;
+    } else if 
(message.startsWith(StatusCode.PUSH_DATA_FAIL_PARTITION_NOT_FOUND.name())) {
+      cause = StatusCode.PUSH_DATA_FAIL_PARTITION_NOT_FOUND;
     } else if (ExceptionUtils.connectFail(message)) {
       // Throw when push to primary worker connection causeException.
       cause = StatusCode.PUSH_DATA_CONNECTION_EXCEPTION_PRIMARY;