hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-10-30 09:48:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

增加chat role 可见性修改

Browse Source
This commit is contained in:
cherishsince
2024-04-24 17:19:34 +08:00
parent c44f558e3d
commit 591002f059
2 changed files with 28 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/impl/ChatRoleServiceImpl.java
View File

@@ -73,22 +73,37 @@ public class ChatRoleServiceImpl implements ChatRoleService {
ChatRoleVisibilityEnum.valueOfType(req.getVisibility()); ChatRoleVisibilityEnum.valueOfType(req.getVisibility());
ChatRoleSourceEnum.valueOfType(req.getRoleSource()); ChatRoleSourceEnum.valueOfType(req.getRoleSource());
// 检查角色是否存在 // 检查角色是否存在
AiChatRoleDO aiChatRoleDO = aiChatRoleMapper.selectById(req.getId()); validateChatRoleExists(req.getId());
if (aiChatRoleDO == null) {
throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_CHAT_ROLE_NOT_EXIST);
}
// 转换do // 转换do
AiChatRoleDO updateChatRole = ChatRoleConvert.INSTANCE.convertAiChatRoleDO(req); AiChatRoleDO updateChatRole = ChatRoleConvert.INSTANCE.convertAiChatRoleDO(req);
aiChatRoleMapper.updateById(updateChatRole); aiChatRoleMapper.updateById(updateChatRole);
} }
@Override @Override
public void updateVisibility(ChatRoleUpdateVisibilityReq req) { public void updateVisibility(ChatRoleUpdateVisibilityReq req) {
// 转换enum，并校验enum
ChatRoleVisibilityEnum.valueOfType(req.getVisibility());
// 检查角色是否存在
validateChatRoleExists(req.getId());
// 更新
aiChatRoleMapper.updateById(new AiChatRoleDO()
.setId(req.getId())
.setVisibility(req.getVisibility())
);
} }
@Override @Override
public void delete(Long chatRoleId) { public void delete(Long chatRoleId) {
} }
private AiChatRoleDO validateChatRoleExists(Long id) {
AiChatRoleDO aiChatRoleDO = aiChatRoleMapper.selectById(id);
if (aiChatRoleDO == null) {
throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_CHAT_ROLE_NOT_EXIST);
}
return aiChatRoleDO;
}
} }

10
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/vo/ChatRoleUpdateVisibilityReq.java
View File

@@ -2,6 +2,7 @@ package cn.iocoder.yudao.module.ai.vo;
import cn.iocoder.yudao.framework.common.pojo.PageParam; import cn.iocoder.yudao.framework.common.pojo.PageParam;
import io.swagger.v3.oas.annotations.media.Schema; import io.swagger.v3.oas.annotations.media.Schema;
import jakarta.validation.constraints.NotNull;
import lombok.Data; import lombok.Data;
import lombok.experimental.Accessors; import lombok.experimental.Accessors;
@@ -15,6 +16,11 @@ import lombok.experimental.Accessors;
@Accessors(chain = true) @Accessors(chain = true)
public class ChatRoleUpdateVisibilityReq extends PageParam { public class ChatRoleUpdateVisibilityReq extends PageParam {
@Schema(description = "查询") @NotNull
private String search; @Schema(description = "编号")
private Long id;
@NotNull
@Schema(description = "发布状态，0表示仅自己可见，1表示公开，2表示禁用")
private String visibility;
} }