[CELEBORN-1726][FOLLOWUP] Avoid NPE when transition worker state

### What changes were proposed in this pull request? Avoid NPE when transition worker state ### Why are the changes needed? Fix test. ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? No need. Closes #2938 from reswqa/fix-npe. Authored-by: Weijie Guo <[email protected]> Signed-off-by: SteNicholas <[email protected]>
apache · Nov 22, 2024 · d722b76 · d722b76
1 parent a2d3972
commit d722b76
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/worker/src/main/scala/org/apache/celeborn/service/deploy/worker/WorkerStatusManager.scala b/worker/src/main/scala/org/apache/celeborn/service/deploy/worker/WorkerStatusManager.scala
@@ -129,7 +129,9 @@ private[celeborn] class WorkerStatusManager(conf: CelebornConf) extends Logging
     if (allowStates != null && allowStates.contains(state)) {
       logInfo(s"Worker transition status from ${currentWorkerStatus.getState} to $state.")
       currentWorkerStatus = new WorkerStatus(state.getNumber, System.currentTimeMillis())
-      worker.workerInfo.setWorkerStatus(currentWorkerStatus)
+      if (worker != null && worker.workerInfo != null) {
+        worker.workerInfo.setWorkerStatus(currentWorkerStatus)
+      }
     } else {
       logWarning(
         s"Worker transition status from ${currentWorkerStatus.getState} to $state is not allowed.")