#AI多租户管控

zhou zhou

19 小时以前 eb49fb9a98d6dd4e4361daf4eac4f9313236b8e8

 rsf-server/src/main/java/com/vincent/rsf/server/ai/service/impl/AiPromptServiceImpl.java

@@ -14,10 +14,13 @@
public class AiPromptServiceImpl extends ServiceImpl<AiPromptMapper, AiPrompt> implements AiPromptService {

    @Override
    public AiPrompt getActivePrompt(String code) {
    public AiPrompt getActivePrompt(String code, Long tenantId) {
        ensureTenantId(tenantId);
        AiPrompt aiPrompt = this.getOne(new LambdaQueryWrapper<AiPrompt>()
                .eq(AiPrompt::getTenantId, tenantId)
                .eq(AiPrompt::getCode, code)
                .eq(AiPrompt::getStatus, StatusType.ENABLE.val)
                .eq(AiPrompt::getDeleted, 0)
                .last("limit 1"));
        if (aiPrompt == null) {
            throw new CoolException("未找到启用中的 Prompt：" + code);
@@ -26,18 +29,23 @@
    }

    @Override
    public void validateBeforeSave(AiPrompt aiPrompt) {
    public void validateBeforeSave(AiPrompt aiPrompt, Long tenantId) {
        ensureTenantId(tenantId);
        aiPrompt.setTenantId(tenantId);
        ensureRequiredFields(aiPrompt);
        ensureUniqueCode(aiPrompt.getCode(), null);
        ensureUniqueCode(aiPrompt.getCode(), tenantId, null);
    }

    @Override
    public void validateBeforeUpdate(AiPrompt aiPrompt) {
    public void validateBeforeUpdate(AiPrompt aiPrompt, Long tenantId) {
        ensureTenantId(tenantId);
        if (aiPrompt.getId() == null) {
            throw new CoolException("Prompt ID 不能为空");
        }
        AiPrompt current = requireOwnedRecord(aiPrompt.getId(), tenantId);
        aiPrompt.setTenantId(current.getTenantId());
        ensureRequiredFields(aiPrompt);
        ensureUniqueCode(aiPrompt.getCode(), aiPrompt.getId());
        ensureUniqueCode(aiPrompt.getCode(), tenantId, aiPrompt.getId());
    }

    private void ensureRequiredFields(AiPrompt aiPrompt) {
@@ -55,9 +63,11 @@
        }
    }

    private void ensureUniqueCode(String code, Long selfId) {
    private void ensureUniqueCode(String code, Long tenantId, Long selfId) {
        LambdaQueryWrapper<AiPrompt> wrapper = new LambdaQueryWrapper<AiPrompt>()
                .eq(AiPrompt::getCode, code);
                .eq(AiPrompt::getTenantId, tenantId)
                .eq(AiPrompt::getCode, code)
                .eq(AiPrompt::getDeleted, 0);
        if (selfId != null) {
            wrapper.ne(AiPrompt::getId, selfId);
        }
@@ -65,4 +75,22 @@
            throw new CoolException("Prompt 编码已存在");
        }
    }

    private AiPrompt requireOwnedRecord(Long id, Long tenantId) {
        AiPrompt aiPrompt = this.getOne(new LambdaQueryWrapper<AiPrompt>()
                .eq(AiPrompt::getId, id)
                .eq(AiPrompt::getTenantId, tenantId)
                .eq(AiPrompt::getDeleted, 0)
                .last("limit 1"));
        if (aiPrompt == null) {
            throw new CoolException("Prompt 不存在或无权访问");
        }
        return aiPrompt;
    }

    private void ensureTenantId(Long tenantId) {
        if (tenantId == null) {
            throw new CoolException("当前租户不存在");
        }
    }
}