From 4954d3978cf1967729a5a2d5b90f6baef18974da Mon Sep 17 00:00:00 2001
From: zhou zhou <3272660260@qq.com>
Date: 星期一, 23 三月 2026 09:35:10 +0800
Subject: [PATCH] #ai redis+页面优化
---
rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiParamServiceImpl.java | 242 ++++++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 232 insertions(+), 10 deletions(-)
diff --git a/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiParamServiceImpl.java b/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiParamServiceImpl.java
index 41a014b..4e1e83e 100644
--- a/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiParamServiceImpl.java
+++ b/rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiParamServiceImpl.java
@@ -4,20 +4,35 @@
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.vincent.rsf.framework.exception.CoolException;
import com.vincent.rsf.server.ai.config.AiDefaults;
+import com.vincent.rsf.server.ai.dto.AiChatModelOptionDto;
+import com.vincent.rsf.server.ai.dto.AiParamValidateResultDto;
import com.vincent.rsf.server.ai.entity.AiParam;
import com.vincent.rsf.server.ai.mapper.AiParamMapper;
import com.vincent.rsf.server.ai.service.AiParamService;
import com.vincent.rsf.server.system.enums.StatusType;
+import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.List;
+
@Service("aiParamService")
+@RequiredArgsConstructor
public class AiParamServiceImpl extends ServiceImpl<AiParamMapper, AiParam> implements AiParamService {
+ private final AiParamValidationSupport aiParamValidationSupport;
+ private final AiRedisSupport aiRedisSupport;
+
@Override
- public AiParam getActiveParam() {
+ public AiParam getActiveParam(Long tenantId) {
+ ensureTenantId(tenantId);
AiParam aiParam = this.getOne(new LambdaQueryWrapper<AiParam>()
+ .eq(AiParam::getTenantId, tenantId)
.eq(AiParam::getStatus, StatusType.ENABLE.val)
+ .eq(AiParam::getDeleted, 0)
.last("limit 1"));
if (aiParam == null) {
throw new CoolException("鏈壘鍒板惎鐢ㄤ腑鐨� AI 鍙傛暟閰嶇疆");
@@ -26,20 +41,134 @@
}
@Override
- public void validateBeforeSave(AiParam aiParam) {
- fillDefaults(aiParam);
- ensureBaseFields(aiParam);
- ensureSingleActive(aiParam, null);
+ public AiParam getChatParam(Long tenantId, Long aiParamId) {
+ ensureTenantId(tenantId);
+ if (aiParamId == null) {
+ return getActiveParam(tenantId);
+ }
+ AiParam aiParam = requireOwnedRecord(aiParamId, tenantId);
+ if (!AiDefaults.PARAM_VALIDATE_VALID.equals(aiParam.getValidateStatus())) {
+ throw new CoolException("鎵�閫� AI 妯″瀷鏈�氳繃鏍￠獙锛屾殏涓嶅彲鐢ㄤ簬瀵硅瘽");
+ }
+ return aiParam;
}
@Override
- public void validateBeforeUpdate(AiParam aiParam) {
+ public List<AiChatModelOptionDto> listChatModelOptions(Long tenantId) {
+ ensureTenantId(tenantId);
+ List<AiParam> params = this.list(new LambdaQueryWrapper<AiParam>()
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getDeleted, 0)
+ .eq(AiParam::getValidateStatus, AiDefaults.PARAM_VALIDATE_VALID)
+ .orderByDesc(AiParam::getStatus)
+ .orderByDesc(AiParam::getUpdateTime)
+ .orderByDesc(AiParam::getCreateTime)
+ .orderByDesc(AiParam::getId));
+ if (params.isEmpty()) {
+ return List.of(toChatModelOption(getActiveParam(tenantId)));
+ }
+ return params.stream()
+ .map(this::toChatModelOption)
+ .toList();
+ }
+
+ @Override
+ @Transactional(rollbackFor = Exception.class)
+ public AiParam setDefaultParam(Long id, Long tenantId, Long userId) {
+ ensureTenantId(tenantId);
+ if (id == null) {
+ throw new CoolException("AI 鍙傛暟 ID 涓嶈兘涓虹┖");
+ }
+ AiParam target = requireOwnedRecord(id, tenantId);
+ if (!AiDefaults.PARAM_VALIDATE_VALID.equals(target.getValidateStatus())) {
+ throw new CoolException("浠呭厑璁稿皢鏍￠獙閫氳繃鐨� AI 鍙傛暟璁剧疆涓洪粯璁�");
+ }
+ Date now = new Date();
+ this.lambdaUpdate()
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getDeleted, 0)
+ .set(AiParam::getStatus, StatusType.DISABLE.val)
+ .set(AiParam::getUpdateBy, userId)
+ .set(AiParam::getUpdateTime, now)
+ .update();
+ target.setStatus(StatusType.ENABLE.val);
+ target.setUpdateBy(userId);
+ target.setUpdateTime(now);
+ if (!super.updateById(target)) {
+ throw new CoolException("璁剧疆榛樿 AI 鍙傛暟澶辫触");
+ }
+ aiRedisSupport.evictTenantConfigCaches(tenantId);
+ return target;
+ }
+
+ @Override
+ public void validateBeforeSave(AiParam aiParam, Long tenantId) {
+ ensureTenantId(tenantId);
+ aiParam.setTenantId(tenantId);
+ fillDefaults(aiParam);
+ ensureBaseFields(aiParam);
+ ensureSingleActive(tenantId, null, aiParam.getStatus());
+ applyValidation(aiParam);
+ }
+
+ @Override
+ public void validateBeforeUpdate(AiParam aiParam, Long tenantId) {
+ ensureTenantId(tenantId);
fillDefaults(aiParam);
if (aiParam.getId() == null) {
throw new CoolException("AI 鍙傛暟 ID 涓嶈兘涓虹┖");
}
+ AiParam current = requireOwnedRecord(aiParam.getId(), tenantId);
+ aiParam.setTenantId(current.getTenantId());
ensureBaseFields(aiParam);
- ensureSingleActive(aiParam, aiParam.getId());
+ ensureDefaultStillExists(tenantId, current, aiParam.getStatus());
+ ensureSingleActive(tenantId, aiParam.getId(), aiParam.getStatus());
+ applyValidation(aiParam);
+ }
+
+ @Override
+ public AiParamValidateResultDto validateDraft(AiParam aiParam, Long tenantId) {
+ ensureTenantId(tenantId);
+ fillDefaults(aiParam);
+ ensureBaseFields(aiParam);
+ return aiParamValidationSupport.validate(aiParam);
+ }
+
+ @Override
+ public boolean save(AiParam entity) {
+ boolean saved = super.save(entity);
+ if (saved && entity != null && entity.getTenantId() != null) {
+ aiRedisSupport.evictTenantConfigCaches(entity.getTenantId());
+ }
+ return saved;
+ }
+
+ @Override
+ public boolean updateById(AiParam entity) {
+ boolean updated = super.updateById(entity);
+ if (updated && entity != null && entity.getTenantId() != null) {
+ aiRedisSupport.evictTenantConfigCaches(entity.getTenantId());
+ }
+ return updated;
+ }
+
+ @Override
+ public boolean removeByIds(java.util.Collection<?> list) {
+ java.util.List<java.io.Serializable> ids = list == null ? java.util.List.of() : list.stream()
+ .filter(java.util.Objects::nonNull)
+ .map(item -> (java.io.Serializable) item)
+ .toList();
+ java.util.List<AiParam> records = this.listByIds(ids);
+ ensureRemovingDefaultIsSafe(records);
+ boolean removed = super.removeByIds(list);
+ if (removed) {
+ records.stream()
+ .map(AiParam::getTenantId)
+ .filter(java.util.Objects::nonNull)
+ .distinct()
+ .forEach(aiRedisSupport::evictTenantConfigCaches);
+ }
+ return removed;
}
private void ensureBaseFields(AiParam aiParam) {
@@ -60,17 +189,75 @@
}
}
- private void ensureSingleActive(AiParam aiParam, Long selfId) {
- if (aiParam.getStatus() == null || aiParam.getStatus() != StatusType.ENABLE.val) {
+ private void ensureSingleActive(Long tenantId, Long selfId, Integer status) {
+ if (status == null || status != StatusType.ENABLE.val) {
return;
}
LambdaQueryWrapper<AiParam> wrapper = new LambdaQueryWrapper<AiParam>()
- .eq(AiParam::getStatus, StatusType.ENABLE.val);
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getStatus, StatusType.ENABLE.val)
+ .eq(AiParam::getDeleted, 0);
if (selfId != null) {
wrapper.ne(AiParam::getId, selfId);
}
if (this.count(wrapper) > 0) {
throw new CoolException("鍚屼竴绉熸埛浠呭厑璁镐竴鏉″惎鐢ㄤ腑鐨� AI 鍙傛暟閰嶇疆");
+ }
+ }
+
+ private void ensureDefaultStillExists(Long tenantId, AiParam current, Integer nextStatus) {
+ if (current == null || current.getStatus() == null || current.getStatus() != StatusType.ENABLE.val) {
+ return;
+ }
+ if (nextStatus != null && nextStatus == StatusType.ENABLE.val) {
+ return;
+ }
+ long otherDefaultCount = this.count(new LambdaQueryWrapper<AiParam>()
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getDeleted, 0)
+ .eq(AiParam::getStatus, StatusType.ENABLE.val)
+ .ne(AiParam::getId, current.getId()));
+ if (otherDefaultCount == 0) {
+ throw new CoolException("璇峰厛灏嗗叾浠� AI 鍙傛暟璁剧疆涓洪粯璁わ紝鍐嶅彇娑堝綋鍓嶉粯璁�");
+ }
+ }
+
+ private void ensureRemovingDefaultIsSafe(List<AiParam> records) {
+ if (records == null || records.isEmpty()) {
+ return;
+ }
+ records.stream()
+ .filter(item -> item.getTenantId() != null && item.getStatus() != null && item.getStatus() == StatusType.ENABLE.val)
+ .map(AiParam::getTenantId)
+ .distinct()
+ .forEach(this::ensureTenantHasRemainingDefaultAfterRemove);
+ }
+
+ private void ensureTenantHasRemainingDefaultAfterRemove(Long tenantId) {
+ long defaultCount = this.count(new LambdaQueryWrapper<AiParam>()
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getDeleted, 0)
+ .eq(AiParam::getStatus, StatusType.ENABLE.val));
+ if (defaultCount <= 1) {
+ throw new CoolException("榛樿 AI 鍙傛暟涓嶈兘鐩存帴鍒犻櫎锛岃鍏堝皢鍏朵粬閰嶇疆璁句负榛樿");
+ }
+ }
+
+ private AiParam requireOwnedRecord(Long id, Long tenantId) {
+ AiParam aiParam = this.getOne(new LambdaQueryWrapper<AiParam>()
+ .eq(AiParam::getId, id)
+ .eq(AiParam::getTenantId, tenantId)
+ .eq(AiParam::getDeleted, 0)
+ .last("limit 1"));
+ if (aiParam == null) {
+ throw new CoolException("AI 鍙傛暟涓嶅瓨鍦ㄦ垨鏃犳潈璁块棶");
+ }
+ return aiParam;
+ }
+
+ private void ensureTenantId(Long tenantId) {
+ if (tenantId == null) {
+ throw new CoolException("褰撳墠绉熸埛涓嶅瓨鍦�");
}
}
@@ -90,8 +277,43 @@
if (aiParam.getStreamingEnabled() == null) {
aiParam.setStreamingEnabled(Boolean.TRUE);
}
+ if (!StringUtils.hasText(aiParam.getValidateStatus())) {
+ aiParam.setValidateStatus(AiDefaults.PARAM_VALIDATE_NOT_TESTED);
+ }
if (aiParam.getStatus() == null) {
aiParam.setStatus(StatusType.ENABLE.val);
}
}
+
+ private void applyValidation(AiParam aiParam) {
+ AiParamValidateResultDto validateResult = aiParamValidationSupport.validate(aiParam);
+ aiParam.setValidateStatus(validateResult.getStatus());
+ aiParam.setLastValidateMessage(validateResult.getMessage());
+ aiParam.setLastValidateElapsedMs(validateResult.getElapsedMs());
+ aiParam.setLastValidateTime(parseDate(validateResult.getValidatedAt()));
+ if (!AiDefaults.PARAM_VALIDATE_VALID.equals(validateResult.getStatus())) {
+ throw new CoolException(validateResult.getMessage());
+ }
+ }
+
+ private Date parseDate(String dateTime) {
+ if (!StringUtils.hasText(dateTime)) {
+ return null;
+ }
+ try {
+ return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").parse(dateTime);
+ } catch (Exception e) {
+ throw new CoolException("瑙ｆ瀽鏍￠獙鏃堕棿澶辫触: " + e.getMessage());
+ }
+ }
+
+ private AiChatModelOptionDto toChatModelOption(AiParam aiParam) {
+ return AiChatModelOptionDto.builder()
+ .aiParamId(aiParam.getId())
+ .name(aiParam.getName())
+ .model(aiParam.getModel())
+ .providerType(aiParam.getProviderType())
+ .active(aiParam.getStatus() != null && aiParam.getStatus() == StatusType.ENABLE.val)
+ .build();
+ }
}
--
Gitblit v1.9.1