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

Loading…
Cancel
Save