提交笔录拆分分类的相关代码

src/main/java/com/supervision/police/domain/TripleInfo.java

@@ -50,6 +50,16 @@ public class TripleInfo implements Serializable {
      */
     private String addNeo4j;
 
+    /**
+     * neo4j开始节点的ID
+     */
+    private Long startNodeGraphId;
+
+    /**
+     * neo4j结束节点的ID
+     */
+    private Long endNodeGraphId;
+
     /**
      * 开始节点类型(neo4j节点类型)
      */

src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java

@@ -149,6 +149,13 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
 
             // 如果存在,则校验时间是否已经超过1天,如果超过了1天还没有执行完毕,就重新提交这个任务
             CaseTaskRecord caseTaskRecord = caseTaskRecordOpt.get();
+            // 如果未执行,则提交执行
+            if (caseTaskRecordOpt.get().getStatus() == 0) {
+                caseTaskRecord.setStatus(1);
+                caseTaskRecord.setSubmitTime(LocalDateTime.now());
+                caseTaskRecordService.updateById(caseTaskRecord);
+                return false;
+            }
             if (caseTaskRecordOpt.get().getStatus() == 1 && LocalDateTime.now().isAfter(caseTaskRecord.getSubmitTime().plusDays(1))) {
                 // 如果已经超过1天,则重新提交任务
                 caseTaskRecord.setStatus(1);
@@ -156,6 +163,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                 caseTaskRecordService.updateById(caseTaskRecord);
                 return false;
             } else if (caseTaskRecordOpt.get().getStatus() == 2) {
+                // 如果执行成功,就返回true,取反之后就可以返回三元组信息了
                 return true;
             } else if (caseTaskRecordOpt.get().getStatus() == 3) {
                 caseTaskRecord.setStatus(1);
@@ -310,6 +318,8 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                     neo4jService.saveRelation(rel);
                 }
                 tripleInfo.setAddNeo4j("1");
+                tripleInfo.setStartNodeGraphId(startNode.getId());
+                tripleInfo.setEndNodeGraphId(endNode.getId());
                 boolean updateResult = tripleInfoService.updateById(tripleInfo);
                 if (updateResult) {
                     i++;

src/main/java/com/supervision/thread/TripleExtractThread.java

@@ -103,7 +103,16 @@ public class TripleExtractThread implements Callable<TripleInfo> {
                     return null;
                 }
                 // 构建三元组信息
-                return new TripleInfo(headEntity.getName(), tailEntity.getName(), relation, caseId, recordId, recordSplitId, LocalDateTime.now(), prompt.getStartEntityType(), prompt.getEndEntityType());
+                TripleInfo tripleInfo = new TripleInfo();
+                tripleInfo.setStartNode(headEntity.getName());
+                tripleInfo.setEndNode(tailEntity.getName());
+                tripleInfo.setRelation(relation);
+                tripleInfo.setCaseId(caseId);
+                tripleInfo.setRecordId(recordId);
+                tripleInfo.setRecordSplitId(recordSplitId);
+                tripleInfo.setStartNodeType(prompt.getStartEntityType());
+                tripleInfo.setEndNodeType(prompt.getEndEntityType());
+                return tripleInfo;
             }
         } catch (Exception e) {
             log.error("提取三元组出现错误", e);