From ef9f43e77e7e4a53a68585e124ef504625122861 Mon Sep 17 00:00:00 2001 From: YunaiV Date: Sun, 12 Jul 2026 12:26:44 +0800 Subject: [PATCH] feat(ai): 增加更多 get 获取详情接口 --- yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatConversationController.java | 14 ++++++++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java | 16 +++++++++++++++- yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java | 11 ++++++++++- yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java | 9 +++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/music/AiMusicController.java | 9 +++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/write/AiWriteController.java | 9 +++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageService.java | 16 ++++++++++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java | 10 ++++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java | 8 ++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java | 5 +++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteService.java | 8 ++++++++ yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteServiceImpl.java | 5 +++++ 12 files changed, 118 insertions(+), 2 deletions(-) diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatConversationController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatConversationController.java index 1450fe9..a1ef930 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatConversationController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatConversationController.java @@ -106,6 +106,20 @@ public class AiChatConversationController { conversation -> conversation.setMessageCount(messageCountMap.getOrDefault(conversation.getId(), 0)))); } + @GetMapping("/get") + @Operation(summary = "获得对话", description = "用于【对话管理】菜单") + @Parameter(name = "id", required = true, description = "对话编号", example = "1024") + @PreAuthorize("@ss.hasPermission('ai:chat-conversation:query')") + @TransMethodResult + public CommonResult getChatConversation(@RequestParam("id") Long id) { + AiChatConversationDO conversation = chatConversationService.getChatConversation(id); + AiChatConversationRespVO respVO = BeanUtils.toBean(conversation, AiChatConversationRespVO.class); + if (respVO != null) { + respVO.setMessageCount(chatMessageService.getChatMessageCount(id)); + } + return success(respVO); + } + @Operation(summary = "管理员删除对话") @DeleteMapping("/delete-by-admin") @Parameter(name = "id", required = true, description = "对话编号", example = "1024") diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java index bfd1e41..8e087d4 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/chat/AiChatMessageController.java @@ -131,7 +131,7 @@ public class AiChatMessageController { @GetMapping("/page") @Operation(summary = "获得消息分页", description = "用于【对话管理】菜单") - @PreAuthorize("@ss.hasPermission('ai:chat-conversation:query')") + @PreAuthorize("@ss.hasPermission('ai:chat-message:query')") public CommonResult> getChatMessagePage(AiChatMessagePageReqVO pageReqVO) { PageResult pageResult = chatMessageService.getChatMessagePage(pageReqVO); if (CollUtil.isEmpty(pageResult.getList())) { @@ -145,6 +145,20 @@ public class AiChatMessageController { role -> respVO.setRoleName(role.getName())))); } + @GetMapping("/get") + @Operation(summary = "获得消息", description = "用于【对话管理】菜单") + @Parameter(name = "id", required = true, description = "消息编号", example = "1024") + @PreAuthorize("@ss.hasPermission('ai:chat-message:query')") + public CommonResult getChatMessage(@RequestParam("id") Long id) { + AiChatMessageDO message = chatMessageService.getChatMessage(id); + AiChatMessageRespVO respVO = BeanUtils.toBean(message, AiChatMessageRespVO.class); + if (respVO != null && respVO.getRoleId() != null) { + AiChatRoleDO role = chatRoleService.getChatRole(respVO.getRoleId()); + respVO.setRoleName(role != null ? role.getName() : null); + } + return success(respVO); + } + @Operation(summary = "管理员删除消息") @DeleteMapping("/delete-by-admin") @Parameter(name = "id", required = true, description = "消息编号", example = "1024") diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java index 1d9503f..4065697 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/image/AiImageController.java @@ -119,6 +119,15 @@ public class AiImageController { return success(BeanUtils.toBean(pageResult, AiImageRespVO.class)); } + @GetMapping("/get") + @Operation(summary = "获得绘画") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('ai:image:query')") + public CommonResult getImage(@RequestParam("id") Long id) { + AiImageDO image = imageService.getImage(id); + return success(BeanUtils.toBean(image, AiImageRespVO.class)); + } + @PutMapping("/update") @Operation(summary = "更新绘画") @PreAuthorize("@ss.hasPermission('ai:image:update')") @@ -136,4 +145,4 @@ public class AiImageController { return success(true); } -} \ No newline at end of file +} diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java index db015e5..fc0f39e 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/mindmap/AiMindMapController.java @@ -46,6 +46,15 @@ public class AiMindMapController { return success(true); } + @GetMapping("/get") + @Operation(summary = "获得思维导图") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('ai:mind-map:query')") + public CommonResult getMindMap(@RequestParam("id") Long id) { + AiMindMapDO mindMap = mindMapService.getMindMap(id); + return success(BeanUtils.toBean(mindMap, AiMindMapRespVO.class)); + } + @GetMapping("/page") @Operation(summary = "获得思维导图分页") @PreAuthorize("@ss.hasPermission('ai:mind-map:query')") diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/music/AiMusicController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/music/AiMusicController.java index 6c09e4b..df76026 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/music/AiMusicController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/music/AiMusicController.java @@ -78,6 +78,15 @@ public class AiMusicController { return success(BeanUtils.toBean(pageResult, AiMusicRespVO.class)); } + @GetMapping("/get") + @Operation(summary = "获得音乐") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('ai:music:query')") + public CommonResult getMusic(@RequestParam("id") Long id) { + AiMusicDO music = musicService.getMusic(id); + return success(BeanUtils.toBean(music, AiMusicRespVO.class)); + } + @DeleteMapping("/delete") @Operation(summary = "删除音乐") @Parameter(name = "id", description = "编号", required = true) diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/write/AiWriteController.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/write/AiWriteController.java index 7ef208b..d50f5f9 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/write/AiWriteController.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/write/AiWriteController.java @@ -46,6 +46,15 @@ public class AiWriteController { return success(true); } + @GetMapping("/get") + @Operation(summary = "获得写作") + @Parameter(name = "id", description = "编号", required = true, example = "1024") + @PreAuthorize("@ss.hasPermission('ai:write:query')") + public CommonResult getWrite(@RequestParam("id") Long id) { + AiWriteDO write = writeService.getWrite(id); + return success(BeanUtils.toBean(write, AiWriteRespVO.class)); + } + @GetMapping("/page") @Operation(summary = "获得写作分页") @PreAuthorize("@ss.hasPermission('ai:write:query')") diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageService.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageService.java index f572bdd..e8c5bc1 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageService.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageService.java @@ -46,6 +46,14 @@ public interface AiChatMessageService { List getChatMessageListByConversationId(Long conversationId); /** + * 获得聊天消息 + * + * @param id 消息编号 + * @return 聊天消息 + */ + AiChatMessageDO getChatMessage(Long id); + + /** * 删除消息 * * @param id 消息编号 @@ -69,6 +77,14 @@ public interface AiChatMessageService { void deleteChatMessageByAdmin(Long id); /** + * 获得聊天对话的消息数量 + * + * @param conversationId 对话编号 + * @return 消息数量 + */ + Integer getChatMessageCount(Long conversationId); + + /** * 获得聊天对话的消息数量 Map * * @param conversationIds 对话编号数组 diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java index bdd2ba2..6a7f54a 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/chat/AiChatMessageServiceImpl.java @@ -525,6 +525,11 @@ public class AiChatMessageServiceImpl implements AiChatMessageService { } @Override + public AiChatMessageDO getChatMessage(Long id) { + return chatMessageMapper.selectById(id); + } + + @Override public void deleteChatMessage(Long id, Long userId) { // 1. 校验消息存在 AiChatMessageDO message = chatMessageMapper.selectById(id); @@ -558,6 +563,11 @@ public class AiChatMessageServiceImpl implements AiChatMessageService { } @Override + public Integer getChatMessageCount(Long conversationId) { + return chatMessageMapper.selectCount(AiChatMessageDO::getConversationId, conversationId).intValue(); + } + + @Override public Map getChatMessageCountMap(Collection conversationIds) { return chatMessageMapper.selectCountMapByConversationId(conversationIds); } diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java index 65a5aaf..15fa9f2 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapService.java @@ -31,6 +31,14 @@ public interface AiMindMapService { void deleteMindMap(Long id); /** + * 获得思维导图 + * + * @param id 编号 + * @return 思维导图 + */ + AiMindMapDO getMindMap(Long id); + + /** * 获得思维导图分页 * * @param pageReqVO 分页查询 diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java index f307dd7..a7e9f21 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/mindmap/AiMindMapServiceImpl.java @@ -154,6 +154,11 @@ public class AiMindMapServiceImpl implements AiMindMapService { } @Override + public AiMindMapDO getMindMap(Long id) { + return mindMapMapper.selectById(id); + } + + @Override public PageResult getMindMapPage(AiMindMapPageReqVO pageReqVO) { return mindMapMapper.selectPage(pageReqVO); } diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteService.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteService.java index f2dd489..f6a3f0a 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteService.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteService.java @@ -31,6 +31,14 @@ public interface AiWriteService { void deleteWrite(Long id); /** + * 获得写作 + * + * @param id 编号 + * @return AI 写作 + */ + AiWriteDO getWrite(Long id); + + /** * 获得写作分页 * * @param pageReqVO 分页查询 diff --git a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteServiceImpl.java b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteServiceImpl.java index 179188d..48b60b1 100644 --- a/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteServiceImpl.java +++ b/yudao-module-ai/src/main/java/cn/iocoder/yudao/module/ai/service/write/AiWriteServiceImpl.java @@ -171,6 +171,11 @@ public class AiWriteServiceImpl implements AiWriteService { } @Override + public AiWriteDO getWrite(Long id) { + return writeMapper.selectById(id); + } + + @Override public PageResult getWritePage(AiWritePageReqVO pageReqVO) { return writeMapper.selectPage(pageReqVO); } -- libgit2 0.21.4