'commit'

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<Record> 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);
+    }
 }

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
     }
 }

src/main/java/Tools/MaxKb/UploadQA.java

@@ -16,5 +16,7 @@ public class UploadQA {
         MaxKbDb.Init();
         //上传文件
         MaxKbService.uploadQA(uploadFile);
+
+        System.out.println("恭喜，所有操作成功完成！");
     }
 }