TASK: library关联分类;

src/main/java/com/ai/da/service/ClassificationService.java

@@ -31,4 +31,6 @@ public interface ClassificationService {
     Boolean relationLibrary(ClassificationDTO classificationDTO);
 
     List<Long> getLibraryIdListByClassificationId(Long classificationId);
+
+    List<Long> getRelClassificationIdList(ClassificationDTO classificationDTO);
 }

src/main/java/com/ai/da/service/impl/ClassificationServiceImpl.java

@@ -177,6 +177,20 @@ public class ClassificationServiceImpl implements ClassificationService {
         return libraryIdList;
     }
 
+    @Override
+    public List<Long> getRelClassificationIdList(ClassificationDTO classificationDTO) {
+        Long libraryId = classificationDTO.getLibraryIdList().get(0);
+        QueryWrapper<ClassificationRelLibrary> qw = new QueryWrapper<>();
+        qw.lambda().eq(ClassificationRelLibrary::getLibraryId, libraryId);
+        List<ClassificationRelLibrary> classificationRelLibraryList = classificationRelLibraryMapper.selectList(qw);
+        if (CollectionUtil.isNotEmpty(classificationRelLibraryList)) {
+            return classificationRelLibraryList.stream()
+                    .map(ClassificationRelLibrary::getLibraryId)
+                    .collect(Collectors.toList());
+        }
+        return new ArrayList<>();
+    }
+
     private List<Long> getTreeClassificationIdListByClassificationId(Long classificationId) {
         List<Long> classificationIdList = new ArrayList<>();
         classificationIdList.add(classificationId);

src/main/java/com/ai/da/service/impl/LibraryServiceImpl.java

@@ -141,6 +141,8 @@ public class LibraryServiceImpl extends ServiceImpl<LibraryMapper, Library> impl
             }
             if (CollectionUtil.isNotEmpty(libraryIdList)) {
                 queryWrapper.lambda().in(Library::getId, libraryIdList);
+            }else {
+                return PageBaseResponse.success(new Page<>());
             }
         }
         queryWrapper.orderByDesc("id");