BJ_ZTEITS-Developers / parking-cloud-yudao

Commit 4cc87539d8af08377876520129b82d03dd71c267

Authored by cherishsince
1 parent 410893bc

chat角色增加 add 方法

Inline Side-by-side
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
  View file @4cc8753
1 1 package cn.iocoder.yudao.module.ai.convert;
2 2  
3 3 import cn.iocoder.yudao.module.ai.dal.dataobject.AiChatRoleDO;
  4 +import cn.iocoder.yudao.module.ai.vo.ChatRoleAddReq;
4 5 import cn.iocoder.yudao.module.ai.vo.ChatRoleListRes;
5 6 import org.mapstruct.Mapper;
6 7 import org.mapstruct.factory.Mappers;
... ... @@ -26,4 +27,12 @@ public interface ChatRoleConvert {
26 27 * @return
27 28 */
28 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
  View file @4cc8753
... ... @@ -28,7 +28,7 @@ public interface ChatRoleService {
28 28 * @param req
29 29 * @return
30 30 */
31   - ChatRoleListRes add(ChatRoleAddReq req);
  31 + void add(ChatRoleAddReq req);
32 32  
33 33 /**
34 34 * chat角色 - 修改
... ...
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/ChatRoleServiceImpl.java
  View file @4cc8753
... ... @@ -3,8 +3,12 @@ package cn.iocoder.yudao.module.ai.service.impl;
3 3 import cn.hutool.core.util.StrUtil;
4 4 import cn.iocoder.yudao.framework.common.pojo.PageResult;
5 5 import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
  6 +import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
6 7 import cn.iocoder.yudao.module.ai.convert.ChatRoleConvert;
7 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 12 import cn.iocoder.yudao.module.ai.mapper.AiChatRoleMapper;
9 13 import cn.iocoder.yudao.module.ai.service.ChatRoleService;
10 14 import cn.iocoder.yudao.module.ai.vo.*;
... ... @@ -47,8 +51,17 @@ public class ChatRoleServiceImpl implements ChatRoleService {
47 51 }
48 52  
49 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 67 @Override
... ...