Commit 4cc87539d8af08377876520129b82d03dd71c267
1 parent
410893bc
chat角色增加 add 方法
Showing
3 changed files
with
25 additions
and
3 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/convert/ChatRoleConvert.java
| 1 | package cn.iocoder.yudao.module.ai.convert; | 1 | package cn.iocoder.yudao.module.ai.convert; |
| 2 | 2 | ||
| 3 | import cn.iocoder.yudao.module.ai.dal.dataobject.AiChatRoleDO; | 3 | import cn.iocoder.yudao.module.ai.dal.dataobject.AiChatRoleDO; |
| 4 | +import cn.iocoder.yudao.module.ai.vo.ChatRoleAddReq; | ||
| 4 | import cn.iocoder.yudao.module.ai.vo.ChatRoleListRes; | 5 | import cn.iocoder.yudao.module.ai.vo.ChatRoleListRes; |
| 5 | import org.mapstruct.Mapper; | 6 | import org.mapstruct.Mapper; |
| 6 | import org.mapstruct.factory.Mappers; | 7 | import org.mapstruct.factory.Mappers; |
| @@ -26,4 +27,12 @@ public interface ChatRoleConvert { | @@ -26,4 +27,12 @@ public interface ChatRoleConvert { | ||
| 26 | * @return | 27 | * @return |
| 27 | */ | 28 | */ |
| 28 | List<ChatRoleListRes> convertChatRoleListRes(List<AiChatRoleDO> roleList); | 29 | List<ChatRoleListRes> convertChatRoleListRes(List<AiChatRoleDO> roleList); |
| 30 | + | ||
| 31 | + /** | ||
| 32 | + * 转换 - AiChatRoleDO | ||
| 33 | + * | ||
| 34 | + * @param req | ||
| 35 | + * @return | ||
| 36 | + */ | ||
| 37 | + AiChatRoleDO convertAiChatRoleDO(ChatRoleAddReq req); | ||
| 29 | } | 38 | } |
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/ChatRoleService.java
| @@ -28,7 +28,7 @@ public interface ChatRoleService { | @@ -28,7 +28,7 @@ public interface ChatRoleService { | ||
| 28 | * @param req | 28 | * @param req |
| 29 | * @return | 29 | * @return |
| 30 | */ | 30 | */ |
| 31 | - ChatRoleListRes add(ChatRoleAddReq req); | 31 | + void add(ChatRoleAddReq req); |
| 32 | 32 | ||
| 33 | /** | 33 | /** |
| 34 | * chat角色 - 修改 | 34 | * chat角色 - 修改 |
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/ChatRoleServiceImpl.java
| @@ -3,8 +3,12 @@ package cn.iocoder.yudao.module.ai.service.impl; | @@ -3,8 +3,12 @@ package cn.iocoder.yudao.module.ai.service.impl; | ||
| 3 | import cn.hutool.core.util.StrUtil; | 3 | import cn.hutool.core.util.StrUtil; |
| 4 | import cn.iocoder.yudao.framework.common.pojo.PageResult; | 4 | import cn.iocoder.yudao.framework.common.pojo.PageResult; |
| 5 | import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX; | 5 | import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX; |
| 6 | +import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils; | ||
| 6 | import cn.iocoder.yudao.module.ai.convert.ChatRoleConvert; | 7 | import cn.iocoder.yudao.module.ai.convert.ChatRoleConvert; |
| 7 | import cn.iocoder.yudao.module.ai.dal.dataobject.AiChatRoleDO; | 8 | import cn.iocoder.yudao.module.ai.dal.dataobject.AiChatRoleDO; |
| 9 | +import cn.iocoder.yudao.module.ai.enums.ChatRoleClassifyEnum; | ||
| 10 | +import cn.iocoder.yudao.module.ai.enums.ChatRoleSourceEnum; | ||
| 11 | +import cn.iocoder.yudao.module.ai.enums.ChatRoleVisibilityEnum; | ||
| 8 | import cn.iocoder.yudao.module.ai.mapper.AiChatRoleMapper; | 12 | import cn.iocoder.yudao.module.ai.mapper.AiChatRoleMapper; |
| 9 | import cn.iocoder.yudao.module.ai.service.ChatRoleService; | 13 | import cn.iocoder.yudao.module.ai.service.ChatRoleService; |
| 10 | import cn.iocoder.yudao.module.ai.vo.*; | 14 | import cn.iocoder.yudao.module.ai.vo.*; |
| @@ -47,8 +51,17 @@ public class ChatRoleServiceImpl implements ChatRoleService { | @@ -47,8 +51,17 @@ public class ChatRoleServiceImpl implements ChatRoleService { | ||
| 47 | } | 51 | } |
| 48 | 52 | ||
| 49 | @Override | 53 | @Override |
| 50 | - public ChatRoleListRes add(ChatRoleAddReq req) { | ||
| 51 | - return null; | 54 | + public void add(ChatRoleAddReq req) { |
| 55 | + // 转换enum，并校验enum | ||
| 56 | + ChatRoleClassifyEnum.valueOfClassify(req.getClassify()); | ||
| 57 | + ChatRoleVisibilityEnum.valueOfType(req.getVisibility()); | ||
| 58 | + ChatRoleSourceEnum.valueOfType(req.getRoleSource()); | ||
| 59 | + // 转换do | ||
| 60 | + AiChatRoleDO insertAiChatRoleDO = ChatRoleConvert.INSTANCE.convertAiChatRoleDO(req); | ||
| 61 | + insertAiChatRoleDO.setUserId(SecurityFrameworkUtils.getLoginUserId()); | ||
| 62 | + insertAiChatRoleDO.setUseCount(0); | ||
| 63 | + // 保存 | ||
| 64 | + aiChatRoleMapper.insert(insertAiChatRoleDO); | ||
| 52 | } | 65 | } |
| 53 | 66 | ||
| 54 | @Override | 67 | @Override |