SmartBench/CMVR-IOT
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 语料查询

Browse Source
This commit is contained in:
stream 2025-10-13 17:52:39 +08:00
parent 39a8b597d1
commit 9ec26e7ac5
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmvr-iot-vi/src/main/java/com/cmvr/vi/service/impl/IViCorpusServiceImpl.java
View File

@ -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<ViCorpusMapper, ViCorpus>
@Override
public List<ViCorpus> selectViCorpusList(ViCorpusVO viCorpusVO) {
String corpusType = viCorpusVO.getType();
if (ObjUtil.isEmpty(corpusType)) {
throw new GlobalException("语料类型不能为空!");
}
LambdaQueryWrapper<ViCorpus> 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); // 只查单次
}