diff --git a/src/main/java/com/ai/da/controller/ClassificationController.java b/src/main/java/com/ai/da/controller/ClassificationController.java index 48f6f85e..8c8f75ae 100644 --- a/src/main/java/com/ai/da/controller/ClassificationController.java +++ b/src/main/java/com/ai/da/controller/ClassificationController.java @@ -74,4 +74,11 @@ public class ClassificationController { return Response.success(classificationService.relationLibrary(classificationDTO)); } + @PostMapping("/getRelClassificationIdList") + @ApiOperationSupport(order = 5) + @ApiModelProperty(value = "获取关联分类IDList", notes = "传入ClassificationDTO") + public Response> getRelClassificationIdList(@Valid @RequestBody ClassificationDTO classificationDTO) { + return Response.success(classificationService.getRelClassificationIdList(classificationDTO)); + } + } diff --git a/src/main/java/com/ai/da/service/ClassificationService.java b/src/main/java/com/ai/da/service/ClassificationService.java index 7dd31dbb..5ab946c0 100644 --- a/src/main/java/com/ai/da/service/ClassificationService.java +++ b/src/main/java/com/ai/da/service/ClassificationService.java @@ -31,4 +31,6 @@ public interface ClassificationService { Boolean relationLibrary(ClassificationDTO classificationDTO); List getLibraryIdListByClassificationId(Long classificationId); + + List getRelClassificationIdList(ClassificationDTO classificationDTO); } diff --git a/src/main/java/com/ai/da/service/impl/ClassificationServiceImpl.java b/src/main/java/com/ai/da/service/impl/ClassificationServiceImpl.java index 2540d805..d759d7eb 100644 --- a/src/main/java/com/ai/da/service/impl/ClassificationServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/ClassificationServiceImpl.java @@ -177,6 +177,20 @@ public class ClassificationServiceImpl implements ClassificationService { return libraryIdList; } + @Override + public List getRelClassificationIdList(ClassificationDTO classificationDTO) { + Long libraryId = classificationDTO.getLibraryIdList().get(0); + QueryWrapper qw = new QueryWrapper<>(); + qw.lambda().eq(ClassificationRelLibrary::getLibraryId, libraryId); + List classificationRelLibraryList = classificationRelLibraryMapper.selectList(qw); + if (CollectionUtil.isNotEmpty(classificationRelLibraryList)) { + return classificationRelLibraryList.stream() + .map(ClassificationRelLibrary::getLibraryId) + .collect(Collectors.toList()); + } + return new ArrayList<>(); + } + private List getTreeClassificationIdListByClassificationId(Long classificationId) { List classificationIdList = new ArrayList<>(); classificationIdList.add(classificationId); diff --git a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java index 1e14e126..73d17a2e 100644 --- a/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java @@ -141,6 +141,8 @@ public class LibraryServiceImpl extends ServiceImpl impl } if (CollectionUtil.isNotEmpty(libraryIdList)) { queryWrapper.lambda().in(Library::getId, libraryIdList); + }else { + return PageBaseResponse.success(new Page<>()); } } queryWrapper.orderByDesc("id");