From e9a85646c477aa5830d9d735ab41141af8451fd1 Mon Sep 17 00:00:00 2001 From: sunxiqing <2240398334@qq.com> Date: Tue, 7 Jan 2025 10:45:54 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=9B=9E=E6=B5=81=E6=A0=A1?= =?UTF-8?q?=E9=AA=8C=E4=BF=AE=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../datarefluxconfig/DataRefluxConfigServiceImpl.java | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) 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); // 返回