diff --git a/src/main/java/Tools/MaxKb/Model/MaxKbModel.java b/src/main/java/Tools/MaxKb/Model/MaxKbModel.java index 09262a8c..caef67ed 100644 --- a/src/main/java/Tools/MaxKb/Model/MaxKbModel.java +++ b/src/main/java/Tools/MaxKb/Model/MaxKbModel.java @@ -17,6 +17,7 @@ public class MaxKbModel { /** * 根据数据集id获取文档列表 + * * @param dataset_id 数据集id * @return */ @@ -24,4 +25,22 @@ public class MaxKbModel { String sql = "select * from document where dataset_id=?::uuid order by create_time desc"; return Db.find(sql, dataset_id); } + + /** + * 删除文档 + * @param documentId 文档id + */ + public void delDocumentProblem(String documentId) { + //根据文档id,查找它相关有哪些问题id,并且删除问题 + String sql = "select * from problem_paragraph_mapping where document_id=?"; + List recordList = Db.find(sql, documentId); + for (Record record : recordList) { + String problem_id = record.getStr("problem_id"); + sql = "delete from problem where id=?"; + Db.update(sql, problem_id); + } + //相关的都删除掉了,需要删除问题与段落的关联关系了 + sql = "delete from problem_paragraph_mapping where document_id=?"; + Db.update(sql, documentId); + } } diff --git a/src/main/java/Tools/MaxKb/Service/MaxKbService.java b/src/main/java/Tools/MaxKb/Service/MaxKbService.java index 0d73f07f..6a8bc142 100644 --- a/src/main/java/Tools/MaxKb/Service/MaxKbService.java +++ b/src/main/java/Tools/MaxKb/Service/MaxKbService.java @@ -48,7 +48,10 @@ public class MaxKbService { } } //上传新的文档 - MaxKbImpl.uploadQA(authCode, zskId, uploadFile); - System.out.println("恭喜,所有操作成功完成!"); + String documentId = MaxKbImpl.uploadQA(authCode, zskId, uploadFile); + //1、需要找出根据此文档有哪些已经生成的问题,并且需要删除掉所有已经生成的问题 + mm.delDocumentProblem(documentId); + //2、需要重新生成问题 + //TODO } } diff --git a/src/main/java/Tools/MaxKb/UploadQA.java b/src/main/java/Tools/MaxKb/UploadQA.java index 11caec8b..d82d7e1c 100644 --- a/src/main/java/Tools/MaxKb/UploadQA.java +++ b/src/main/java/Tools/MaxKb/UploadQA.java @@ -16,5 +16,7 @@ public class UploadQA { MaxKbDb.Init(); //上传文件 MaxKbService.uploadQA(uploadFile); + + System.out.println("恭喜,所有操作成功完成!"); } } \ No newline at end of file