[incubator-uniffle] branch master updated: [#1261] fix(spark): Throw out InterruptedException for sleep in requestExecutorMemory #1262

[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/incubator-uniffle.git


The following commit(s) were added to refs/heads/master by this push:
     new 9dd4c4df [#1261] fix(spark): Throw out InterruptedException for sleep 
in requestExecutorMemory #1262
9dd4c4df is described below

commit 9dd4c4df3617e72288ae77b58ac8907229e9e9ef
Author: Zhen Wang <643348...@qq.com>
AuthorDate: Tue Oct 24 13:37:07 2023 +0800

    [#1261] fix(spark): Throw out InterruptedException for sleep in 
requestExecutorMemory #1262
    
    ### What changes were proposed in this pull request?
    
    Pass InterruptedException in requestExecutorMemory.
    
    ### Why are the changes needed?
    
    Fix:  #1261
    
    ### Does this PR introduce _any_ user-facing change?
    
    No.
    
    ### How was this patch tested?
    
    Existing tests.
---
 .../main/java/org/apache/spark/shuffle/writer/WriteBufferManager.java   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git 
a/client-spark/common/src/main/java/org/apache/spark/shuffle/writer/WriteBufferManager.java
 
b/client-spark/common/src/main/java/org/apache/spark/shuffle/writer/WriteBufferManager.java
index 6e468cd2..a76c04af 100644
--- 
a/client-spark/common/src/main/java/org/apache/spark/shuffle/writer/WriteBufferManager.java
+++ 
b/client-spark/common/src/main/java/org/apache/spark/shuffle/writer/WriteBufferManager.java
@@ -372,7 +372,7 @@ public class WriteBufferManager extends MemoryConsumer {
       try {
         Thread.sleep(requireMemoryInterval);
       } catch (InterruptedException ie) {
-        LOG.warn("Exception happened when waiting for memory.", ie);
+        throw new RssException("Interrupted when waiting for memory.", ie);
       }
       gotMem = acquireMemory(askExecutorMemory);
       allocatedBytes.addAndGet(gotMem);