1. 优化代码

jinan_dev
xueqingkun 5 months ago
parent 5de8ab3787
commit 36331c994a

@ -50,7 +50,7 @@ public interface ModelCaseService extends IService<ModelCase> {
IPage<IndexDetail> getIndexDetail(IndexResultQuery query, Integer page, Integer size);
void initCaseKnowledgeBase(String caseId);
boolean initCaseKnowledgeBase(String caseId);
/**

@ -381,7 +381,7 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
}
@Override
public void initCaseKnowledgeBase(String caseId) {
public boolean initCaseKnowledgeBase(String caseId) {
ModelCase modelCase = this.getById(caseId);
if (StrUtil.isEmpty(modelCase.getKnowledgeBaseId())){
log.info("案件:{} 尚未创建知识库,即将创建知识库库...",modelCase.getCaseName());
@ -389,7 +389,7 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
StrUtil.join("_", modelCase.getCaseName(), modelCase.getCaseNo()), modelCase.getCaseDetail());
if (StrUtil.isEmpty(knowledgeBaseId)){
log.error("案件:{} 创建知识库失败...",modelCase.getCaseName());
return;
return false;
}
modelCase.setKnowledgeBaseId(knowledgeBaseId);
modelCaseMapper.updateById(modelCase);
@ -400,12 +400,13 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
if (null == modelCase.getCaseAnalysisSuccessTime()){
log.info("案件:{} 尚未完成案件分析,不进行笔录上传...",modelCase.getCaseName());
return;
return true;
}
log.info("案件:{} 开始上传笔录到知识库...",modelCase.getCaseName());
difyApiUtil.syncCaseFileToDifyKnowledgeBase(modelCase,this.listCaseFileIds(caseId));
log.info("案件:{} 上传笔录到知识库成功...",modelCase.getCaseName());
return true;
}
@Override
@ -421,7 +422,12 @@ public class ModelCaseServiceImpl extends ServiceImpl<ModelCaseMapper, ModelCase
try {
log.info("开始迁移案件:{} 笔录到知识库,当前总体进度:{}...",modelCase.getCaseName(),NumberUtil.formatPercent((success + fail)/(allModelCase.size()*1.0), 2));
timer.start(modelCase.getId());
initCaseKnowledgeBase(modelCase.getId());
boolean ok = initCaseKnowledgeBase(modelCase.getId());
if (ok){
success++;
}else {
fail++;
}
log.info("案件:{} 迁移笔录到知识库成功,耗时:{}秒...",modelCase.getCaseName(),timer.intervalSecond(modelCase.getId()));
success++;
} catch (Exception e) {

Loading…
Cancel
Save