diff --git a/dk-modules/system/src/main/java/org/dromara/system/service/impl/AiLabelServiceImpl.java b/dk-modules/system/src/main/java/org/dromara/system/service/impl/AiLabelServiceImpl.java index b42756c..749d1d9 100644 --- a/dk-modules/system/src/main/java/org/dromara/system/service/impl/AiLabelServiceImpl.java +++ b/dk-modules/system/src/main/java/org/dromara/system/service/impl/AiLabelServiceImpl.java @@ -23,10 +23,7 @@ import org.dromara.system.domain.AiLabel; import org.dromara.system.mapper.AiLabelMapper; import org.dromara.system.service.IAiLabelService; -import java.util.Comparator; -import java.util.List; -import java.util.Map; -import java.util.Collection; +import java.util.*; import java.util.stream.Collectors; /** @@ -184,8 +181,17 @@ public class AiLabelServiceImpl implements IAiLabelService { @Override public List selectLabelList() { + List sysPostVos = sysPostService.selectPostAll(); + + List postIdList = Optional.ofNullable(sysPostVos) + .orElse(Collections.emptyList()) + .stream() + .map(SysPostVo::getPostId) + .distinct() + .toList(); + List aiLabelList = aiLabelPostService.getAiLabelList(); - return aiLabelList.stream() + return aiLabelList.stream().filter(p-> postIdList.contains(p.getPostId())) .map(aiLabel -> { AiLabelVo aiLabelVo = new AiLabelVo(); aiLabelVo.setLabelId(aiLabel.getLabelId());