DATA_MIDDLE_PLATFORM/Data_Middle_Platform_Server
3
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

数据回流校验修改

Browse Source
This commit is contained in:
sunxiqing 2025-01-07 10:45:54 +08:00
parent c98455c8bd
commit e9a85646c4
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxconfig/DataRefluxConfigServiceImpl.java
View File

@ -1,12 +1,18 @@
package cn.iocoder.yudao.module.llm.service.datarefluxconfig;
import cn.hutool.core.util.ObjectUtil;
import cn.iocoder.yudao.framework.common.enums.UserTypeEnum;
import cn.iocoder.yudao.framework.common.exception.ErrorCode;
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
import cn.iocoder.yudao.framework.security.core.LoginUser;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.llm.dal.dataobject.basemodel.BaseModelDO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;
import cn.iocoder.yudao.module.llm.dal.mysql.basemodel.BaseModelMapper;
import cn.iocoder.yudao.module.llm.dal.mysql.modelservice.ModelServiceMapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
@ -49,9 +55,10 @@ public class DataRefluxConfigServiceImpl implements DataRefluxConfigService {
LambdaQueryWrapper<DataRefluxConfigDO> eq = new LambdaQueryWrapper<DataRefluxConfigDO>()
.eq(DataRefluxConfigDO::getModelServiceId, dataRefluxConfig.getModelServiceId())
.eq(DataRefluxConfigDO::getModelType, dataRefluxConfig.getModelType())
.eq(DataRefluxConfigDO::getCreator, dataRefluxConfig.getCreator());
.eq(DataRefluxConfigDO::getCreator, SecurityFrameworkUtils.getLoginUser().getId());
// LoginUser loginUser = SecurityFrameworkUtils.getLoginUser();
if (dataRefluxConfigMapper.selectCount(eq) > 0) {
throw new RuntimeException("该模型服务已存在");
throw exception(new ErrorCode(11000,"该模型服务已存在"));
}
dataRefluxConfigMapper.insert(dataRefluxConfig);
// 返回