(uniffle) branch master updated: [#2609] feat(spark): Expose `checkDataIfAnyFailure` method so that Gluten can invoke it to trigger reassign ASSP (#2610)

[zuston]

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

zuston pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/uniffle.git


The following commit(s) were added to refs/heads/master by this push:
     new 7015613ad [#2609] feat(spark): Expose `checkDataIfAnyFailure` method 
so that Gluten can invoke it to trigger reassign ASSP (#2610)
7015613ad is described below

commit 7015613adb6759861ff85478dd0590485f131b84
Author: Junfan Zhang <zus...@apache.org>
AuthorDate: Tue Sep 16 10:06:54 2025 +0800

    [#2609] feat(spark): Expose `checkDataIfAnyFailure` method so that Gluten 
can invoke it to trigger reassign ASSP (#2610)
    
    ### What changes were proposed in this pull request?
    
    This PR is exposing the `checkDataIfAnyFailure` method with protected 
modifier, so that the Gluten can invoke this to trigger reassign mechanism as 
soon as possible.
    
    ### Why are the changes needed?
    
    for #2609
    
    ### Does this PR introduce _any_ user-facing change?
    
    No.
    
    ### How was this patch tested?
    
    Neen't
---
 .../main/java/org/apache/spark/shuffle/writer/RssShuffleWriter.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git 
a/client-spark/spark3/src/main/java/org/apache/spark/shuffle/writer/RssShuffleWriter.java
 
b/client-spark/spark3/src/main/java/org/apache/spark/shuffle/writer/RssShuffleWriter.java
index b51ab00d5..122f9548c 100644
--- 
a/client-spark/spark3/src/main/java/org/apache/spark/shuffle/writer/RssShuffleWriter.java
+++ 
b/client-spark/spark3/src/main/java/org/apache/spark/shuffle/writer/RssShuffleWriter.java
@@ -562,7 +562,8 @@ public class RssShuffleWriter<K, V, C> extends 
ShuffleWriter<K, V> {
     }
   }
 
-  private void checkDataIfAnyFailure() {
+  // This method should remain protected so that Gluten can invoke it
+  protected void checkDataIfAnyFailure() {
     if (blockFailSentRetryEnabled) {
       collectFailedBlocksToResend();
     } else {