From a65c068051b1c75d2fc93dbe92eda7b82be1f5fc Mon Sep 17 00:00:00 2001 From: jiang Date: Wed, 20 Nov 2024 14:07:37 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../DataSetBasicFileController.java | 8 ++--- .../ai/service/DataSetBasicFileService.java | 2 +- .../Impl/DataSetBasicFileServiceImpl.java | 33 ++----------------- .../mapper/DataSetBasicFileMapper.xml | 10 +++--- scripts/ai_bootstrap.yml | 14 ++++---- scripts/auth_bootstrap.yml | 4 +-- scripts/deploy.sh | 24 ++++---------- scripts/gateway_bootstrap.yml | 4 +-- scripts/gen_bootstrap.yml | 8 ++--- sql/bonus_ai.sql | 2 +- 10 files changed, 37 insertions(+), 72 deletions(-) diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetBasicFileController.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetBasicFileController.java index 56f38f1..0c16fdf 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetBasicFileController.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/controller/DataSetBasicFileController.java @@ -109,7 +109,7 @@ public class DataSetBasicFileController extends BaseController * 获取文件基础详细信息 */ @RequiresPermissions("dataCenter:dataSetBasicFile:query") - @GetMapping(value = "/{fileId}") + @GetMapping(value = "getFile/{fileId}") public AjaxResult getInfo(@PathVariable("fileId") Long fileId) { return dataSetBasicFileService.selectDataSetBasicFileByFileId(fileId); @@ -215,9 +215,9 @@ public class DataSetBasicFileController extends BaseController * 获取文件树型结构 * @return */ - @PostMapping("/getFileTerr") - public AjaxResult getFileTerr(){ - return dataSetBasicFileService.getFileTerr(); + @PostMapping("/getFileTerr/{fileId}") + public AjaxResult getFileTerr(@PathVariable("fileId") Long fileId){ + return dataSetBasicFileService.getFileTerr(fileId); } } diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/DataSetBasicFileService.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/DataSetBasicFileService.java index 66d9e9a..c7b7951 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/DataSetBasicFileService.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/DataSetBasicFileService.java @@ -111,5 +111,5 @@ public interface DataSetBasicFileService * 获取树形结构 * @return */ - AjaxResult getFileTerr(); + AjaxResult getFileTerr(Long fileId); } diff --git a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetBasicFileServiceImpl.java b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetBasicFileServiceImpl.java index 75c9317..b6b9cb8 100644 --- a/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetBasicFileServiceImpl.java +++ b/bonus-modules/bonus-ai/src/main/java/com/bonus/ai/service/Impl/DataSetBasicFileServiceImpl.java @@ -55,7 +55,7 @@ public class DataSetBasicFileServiceImpl implements DataSetBasicFileService { if (ObjectUtils.isEmpty(aiBasicFile)) { return AjaxResult.error(); } else { - return AjaxResult.success(); + return AjaxResult.success(aiBasicFile); } } catch (Exception e) { return AjaxResult.error(); @@ -291,42 +291,15 @@ public class DataSetBasicFileServiceImpl implements DataSetBasicFileService { * @return */ @Override - public AjaxResult getFileTerr() { + public AjaxResult getFileTerr(Long fileId) { DataSetBasicFileEntity entity = new DataSetBasicFileEntity(); entity.setIsDirectory("1"); entity.setCreateBy(SecurityUtils.getUserId().toString()); List allFiles = dataSetBasicFileMapper.selectDataSetBasicFileList(entity); + allFiles.removeIf(file -> file.getFileId().equals(fileId)); return AjaxResult.success(allFiles); } - /** - * 递归树形结构 - * @param allFiles - * @param parentId - * @return - */ - private List buildTree(List allFiles, Long parentId) { - List result = new ArrayList<>(); - for (DataSetBasicFileEntity entity : allFiles) { - // 判断当前节点的 parentId 是否与给定的 parentId 相等 - if (entity.getParentId().equals(parentId) && "1".equals(entity.getIsDirectory())) { - // 将当前实体转换成 TreeNode - TreeNode node = new TreeNode(); - node.setId(entity.getFileId()); - node.setLabel(entity.getFileName()); - node.setParentId(entity.getParentId()); - // 递归查找当前节点的子节点 - node.setChildren(buildTree(allFiles, entity.getFileId())); - // 将该节点添加到结果列表中 - result.add(node); - } - } - return result; - } - - - - /** * 获取同级目录是否是否存在共享数据 * @return diff --git a/bonus-modules/bonus-ai/src/main/resources/mapper/DataSetBasicFileMapper.xml b/bonus-modules/bonus-ai/src/main/resources/mapper/DataSetBasicFileMapper.xml index 89761a1..abeb803 100644 --- a/bonus-modules/bonus-ai/src/main/resources/mapper/DataSetBasicFileMapper.xml +++ b/bonus-modules/bonus-ai/src/main/resources/mapper/DataSetBasicFileMapper.xml @@ -23,25 +23,27 @@ select file_id, parent_id, ancestors, file_name, file_url, file_size, file_last_modifytime, upload_time, is_directory, is_public, del_flag, create_by, create_time, update_by, update_time from ai_basic_file - - + where del_flag ='0' and create_by = #{createBy} and is_directory = #{isDirectory} - and parent_id = #{parentId} + and parent_id = #{parentId} and file_name like concat('%', #{fileName}, '%') and is_public =#{isPublic} - + ORDER BY is_directory DESC ,create_time DESC