From 8828964e53d2e74260ed4a512015b94a764e6c41 Mon Sep 17 00:00:00 2001
From: chixiaopao <101961274+chixiaopao@users.noreply.github.com>
Date: Mon, 7 Apr 2025 20:55:41 +0800
Subject: [PATCH] (fix)(headless) fix dfs NullPointerException when model is
 null. (#2212)

---
 .../headless/server/utils/ModelClusterBuilder.java         | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelClusterBuilder.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelClusterBuilder.java
index 61b10867d..872a167ef 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelClusterBuilder.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/ModelClusterBuilder.java
@@ -38,7 +38,7 @@ public class ModelClusterBuilder {
     }
 
     private static ModelCluster getModelCluster(Map<Long, ModelSchemaResp> modelIdToModelSchema,
-            Set<Long> modelIds) {
+                                                Set<Long> modelIds) {
         boolean containsPartitionDimensions = modelIds.stream().map(modelIdToModelSchema::get)
                 .filter(Objects::nonNull).anyMatch(modelSchemaResp -> CollectionUtils
                         .isNotEmpty(modelSchemaResp.getTimeDimension()));
@@ -47,7 +47,10 @@ public class ModelClusterBuilder {
     }
 
     private static void dfs(ModelSchemaResp model, Map<Long, ModelSchemaResp> modelMap,
-            Set<Long> visited, Set<Long> modelCluster) {
+                            Set<Long> visited, Set<Long> modelCluster) {
+        if (Objects.isNull(model)) {
+            return;
+        }
         visited.add(model.getId());
         modelCluster.add(model.getId());
         for (Long neighborId : model.getModelClusterSet()) {