diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/ModelServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/ModelServiceImpl.java index 35a55a340..4350c2361 100644 --- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/ModelServiceImpl.java +++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/web/service/impl/ModelServiceImpl.java @@ -289,18 +289,17 @@ public class ModelServiceImpl implements ModelService { public List getModelRespAuthInheritDomain(User user, Long domainId, AuthType authType) { List domainIds = domainService.getDomainAuthSet(user, authType) - .stream().map(DomainResp::getId) + .stream().filter(domainResp -> { + if (domainId == null) { + return true; + } else { + return domainId.equals(domainResp.getId()) || domainId.equals(domainResp.getParentId()); + } + }).map(DomainResp::getId) .collect(Collectors.toList()); if (CollectionUtils.isEmpty(domainIds)) { return Lists.newArrayList(); } - if (domainId != null) { - if (domainIds.contains(domainId)) { - domainIds = Lists.newArrayList(domainId); - } else { - return Lists.newArrayList(); - } - } ModelFilter modelFilter = new ModelFilter(); modelFilter.setIncludesDetail(false); modelFilter.setDomainIds(domainIds);