From 211f589d84d36ef068b81fea5aa656661adb13a0 Mon Sep 17 00:00:00 2001
From: shahaibo <1023316923@qq.com>
Date: Tue, 14 Nov 2023 15:59:11 +0800
Subject: [PATCH] =?UTF-8?q?TASK:=20library=E5=85=B3=E8=81=94=E5=88=86?=
 =?UTF-8?q?=E7=B1=BB;?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ai/da/controller/ClassificationController.java |  7 +++++++
 .../com/ai/da/service/ClassificationService.java   |  2 ++
 .../da/service/impl/ClassificationServiceImpl.java | 14 ++++++++++++++
 .../com/ai/da/service/impl/LibraryServiceImpl.java |  2 ++
 4 files changed, 25 insertions(+)

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<List<Long>> 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<Long> getLibraryIdListByClassificationId(Long classificationId);
+
+    List<Long> 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<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);
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<LibraryMapper, Library> impl
             }
             if (CollectionUtil.isNotEmpty(libraryIdList)) {
                 queryWrapper.lambda().in(Library::getId, libraryIdList);
+            }else {
+                return PageBaseResponse.success(new Page<>());
             }
         }
         queryWrapper.orderByDesc("id");