From 9ec26e7ac5d05c5864384f26e574a4dfe5346d49 Mon Sep 17 00:00:00 2001 From: stream Date: Mon, 13 Oct 2025 17:52:39 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E8=AF=AD=E6=96=99=E6=9F=A5=E8=AF=A2?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../com/cmvr/vi/service/impl/IViCorpusServiceImpl.java | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/cmvr-iot-vi/src/main/java/com/cmvr/vi/service/impl/IViCorpusServiceImpl.java b/cmvr-iot-vi/src/main/java/com/cmvr/vi/service/impl/IViCorpusServiceImpl.java index 73a0c8e..46bcbee 100644 --- a/cmvr-iot-vi/src/main/java/com/cmvr/vi/service/impl/IViCorpusServiceImpl.java +++ b/cmvr-iot-vi/src/main/java/com/cmvr/vi/service/impl/IViCorpusServiceImpl.java @@ -5,6 +5,7 @@ import cn.hutool.core.util.ObjUtil; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.cmvr.common.exception.GlobalException; import com.cmvr.common.utils.uuid.IdUtils; import com.cmvr.vi.enums.ViCorpusEnum; import com.cmvr.vi.mapper.ViCorpusMapper; @@ -38,15 +39,19 @@ public class IViCorpusServiceImpl extends ServiceImpl @Override public List selectViCorpusList(ViCorpusVO viCorpusVO) { + String corpusType = viCorpusVO.getType(); + if (ObjUtil.isEmpty(corpusType)) { + throw new GlobalException("语料类型不能为空!"); + } LambdaQueryWrapper wrapper = Wrappers.lambdaQuery(); // 如果是唤醒语料 - if (ViCorpusEnum.WAKEUP.name().equalsIgnoreCase(viCorpusVO.getType())) { + if (ViCorpusEnum.WAKEUP.name().equalsIgnoreCase(corpusType)) { wrapper.eq(ViCorpus::getType, ViCorpusEnum.WAKEUP); } // 如果是单次对话 - if (ViCorpusEnum.TEST.name().equalsIgnoreCase(viCorpusVO.getType())) { + if (ViCorpusEnum.TEST.name().equalsIgnoreCase(corpusType)) { wrapper.eq(ViCorpus::getType, ViCorpusEnum.TEST) .eq(ViCorpus::getContinuous, 1); // 只查单次 }