(celeborn) branch branch-0.6 updated: [CELEBORN-2006] LifecycleManager should avoid parsing shufflePartitionType every time

[feiwang]

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

feiwang pushed a commit to branch branch-0.6
in repository https://gitbox.apache.org/repos/asf/celeborn.git


The following commit(s) were added to refs/heads/branch-0.6 by this push:
     new e319705ba [CELEBORN-2006] LifecycleManager should avoid parsing 
shufflePartitionType every time
e319705ba is described below

commit e319705ba4c632a50342bbbb0e526e4443397242
Author: sychen <syc...@ctrip.com>
AuthorDate: Thu May 22 17:23:07 2025 -0700

    [CELEBORN-2006] LifecycleManager should avoid parsing shufflePartitionType 
every time
    
    ### What changes were proposed in this pull request?
    
    ### Why are the changes needed?
    `org.apache.celeborn.client.LifecycleManager.getPartitionType` may be 
called frequently, but in Spark scenario it requires each parsing 
configuration, which is not necessary.
    
    ### Does this PR introduce _any_ user-facing change?
    No
    
    ### How was this patch tested?
    GA
    
    Closes #3273 from cxzl25/CELEBORN-2006.
    
    Authored-by: sychen <syc...@ctrip.com>
    Signed-off-by: Wang, Fei <fwan...@ebay.com>
    (cherry picked from commit a554261f32cfd9c7556261f0bbe03ff69bafa66c)
    Signed-off-by: Wang, Fei <fwan...@ebay.com>
---
 .../src/main/scala/org/apache/celeborn/client/LifecycleManager.scala  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git 
a/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala 
b/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala
index 6bc12b8e8..1b44c14bb 100644
--- a/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala
+++ b/client/src/main/scala/org/apache/celeborn/client/LifecycleManager.scala
@@ -329,8 +329,10 @@ class LifecycleManager(val appUniqueId: String, val conf: 
CelebornConf) extends
     rpcEnv.address.port
   }
 
+  private val partitionType = conf.shufflePartitionType
+
   def getPartitionType(shuffleId: Int): PartitionType = {
-    shufflePartitionType.getOrDefault(shuffleId, conf.shufflePartitionType)
+    shufflePartitionType.getOrDefault(shuffleId, partitionType)
   }
 
   override def receive: PartialFunction[Any, Unit] = {