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 modelIdToModelSchema, - Set modelIds) { + Set 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 modelMap, - Set visited, Set modelCluster) { + Set visited, Set modelCluster) { + if (Objects.isNull(model)) { + return; + } visited.add(model.getId()); modelCluster.add(model.getId()); for (Long neighborId : model.getModelClusterSet()) {