diff --git a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxconfig/DataRefluxConfigServiceImpl.java b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxconfig/DataRefluxConfigServiceImpl.java index d4e31d59e..038bbe522 100644 --- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxconfig/DataRefluxConfigServiceImpl.java +++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/datarefluxconfig/DataRefluxConfigServiceImpl.java @@ -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 eq = new LambdaQueryWrapper() .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); // 返回