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); // 只查单次 }