From 8245681dbaf99fb8c6153f871de0eec7fd4fde10 Mon Sep 17 00:00:00 2001
From: liu <liujiatong112@163.com>
Date: Wed, 7 Aug 2024 15:18:07 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/ModelAtomicResultService.java     |  7 +++++
 .../impl/ModelAtomicResultServiceImpl.java    | 13 +++++++++
 .../impl/ModelRecordTypeServiceImpl.java      |  4 +--
 .../impl/NoteRecordSplitServiceImpl.java      | 27 +++++++++++++------
 4 files changed, 41 insertions(+), 10 deletions(-)
 create mode 100644 src/main/java/com/supervision/police/service/ModelAtomicResultService.java
 create mode 100644 src/main/java/com/supervision/police/service/impl/ModelAtomicResultServiceImpl.java

diff --git a/src/main/java/com/supervision/police/service/ModelAtomicResultService.java b/src/main/java/com/supervision/police/service/ModelAtomicResultService.java
new file mode 100644
index 0000000..9a76078
--- /dev/null
+++ b/src/main/java/com/supervision/police/service/ModelAtomicResultService.java
@@ -0,0 +1,7 @@
+package com.supervision.police.service;
+
+import com.baomidou.mybatisplus.extension.service.IService;
+import com.supervision.police.domain.ModelAtomicResult;
+
+public interface ModelAtomicResultService extends IService<ModelAtomicResult> {
+}
diff --git a/src/main/java/com/supervision/police/service/impl/ModelAtomicResultServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelAtomicResultServiceImpl.java
new file mode 100644
index 0000000..b905e0c
--- /dev/null
+++ b/src/main/java/com/supervision/police/service/impl/ModelAtomicResultServiceImpl.java
@@ -0,0 +1,13 @@
+package com.supervision.police.service.impl;
+
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.supervision.police.domain.ModelAtomicResult;
+import com.supervision.police.mapper.ModelAtomicResultMapper;
+import com.supervision.police.service.ModelAtomicResultService;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Service;
+
+@Slf4j
+@Service
+public class ModelAtomicResultServiceImpl extends ServiceImpl<ModelAtomicResultMapper, ModelAtomicResult> implements ModelAtomicResultService {
+}
diff --git a/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java b/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
index 6f3a513..bf33ffa 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
@@ -465,13 +465,13 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
                 } else {
                     break;
                 }
-                if (ObjectUtil.isNotEmpty(tripleInfo.getStartNodeGraphId())) {
+                if (ObjectUtil.isNotEmpty(tripleInfo.getEndNodeGraphId())) {
                     // 再移除尾节点
                     neo4jService.delNode(tripleInfo.getEndNodeGraphId());
                 } else {
                     break;
                 }
-                if (ObjectUtil.isNotEmpty(tripleInfo.getEndNodeGraphId())) {
+                if (ObjectUtil.isNotEmpty(tripleInfo.getStartNodeGraphId())) {
                     // 尝试删除头节点(只有头节点不存在任何关系的时候才进行删除)
                     neo4jService.deleteNoRelationNode(tripleInfo.getStartNodeGraphId());
                 } else {
diff --git a/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java b/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
index ec8ad38..44aae43 100644
--- a/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/NoteRecordSplitServiceImpl.java
@@ -23,6 +23,7 @@ import com.supervision.neo4j.service.Neo4jService;
 import com.supervision.police.domain.*;
 import com.supervision.police.dto.NoteRecordDTO;
 import com.supervision.police.dto.NoteRecordDetailDTO;
+import com.supervision.police.mapper.ModelAtomicResultMapper;
 import com.supervision.police.mapper.NoteRecordMapper;
 import com.supervision.police.mapper.NoteRecordSplitMapper;
 import com.supervision.police.service.*;
@@ -74,6 +75,8 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl<NoteRecordSplitMappe
 
     private final ModelCaseService modelCaseService;
 
+    private final ModelAtomicResultService modelAtomicResultService;
+
 
     @Override
     @Transactional(transactionManager = "dataSourceTransactionManager", rollbackFor = Exception.class)
@@ -286,17 +289,14 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl<NoteRecordSplitMappe
     @Override
     @Transactional(transactionManager = "dataSourceTransactionManager", rollbackFor = Exception.class, noRollbackFor = BusinessException.class)
     public void delRecords(String id) {
-
         NoteRecord noteRecord = noteRecordMapper.selectById(id);
         // 去掉逻辑删除,直接删除
         // noteRecord.setDataStatus(StringUtils.getUUID());
         // noteRecordMapper.updateById(noteRecord);
         String fileIds = noteRecord.getFileIds();
         if (StringUtils.isNotEmpty(fileIds)) {
-
             //删除文件
             for (String fileId : fileIds.split(",")) {
-
                 MinioFile minioFile = minioFileMapper.selectById(fileId);
                 // 删除文件
                 LangChainChatRes langChainChatRes = langChainChatService.deleteFile(DeleteFileDTO.create(noteRecord.getCaseId(), minioFile.getFilename()));
@@ -306,7 +306,6 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl<NoteRecordSplitMappe
                 minioFile.setDataStatus(StringUtils.getUUID());
                 minioFileMapper.updateById(minioFile);
                 minioService.delFile(fileId);
-
             }
         }
         // 去掉逻辑删除,直接删除
@@ -324,15 +323,27 @@ public class NoteRecordSplitServiceImpl extends ServiceImpl<NoteRecordSplitMappe
             if (StrUtil.equals("1", item.getAddNeo4j())) {
                 // 删除尾节点(不删除头节点),如果删除尾节点,关系会被自动删除
                 try {
-                    neo4jService.delNode(item.getEndNodeGraphId());
+                    if (ObjectUtil.isNotEmpty(item.getRelGraphId())) {
+                        // 先移除关系
+                        neo4jService.deleteRel(item.getRelGraphId());
+                    }
+                    if (ObjectUtil.isNotEmpty(item.getEndNodeGraphId())) {
+                        // 再移除尾节点
+                        neo4jService.delNode(item.getEndNodeGraphId());
+                    }
+                    if (ObjectUtil.isNotEmpty(item.getStartNodeGraphId())) {
+                        // 尝试删除头节点(只有头节点不存在任何关系的时候才进行删除)
+                        neo4jService.deleteNoRelationNode(item.getStartNodeGraphId());
+                    }
                 } catch (Exception e) {
                     log.error("删除关系失败:{}", item.getRelation(), e);
                 }
-
             }
-
         });
-        // TODO 是否需要把model_atomic_result的结果也删除了?
+        // 删除三元组数据
+        tripleInfoService.lambdaUpdate().eq(TripleInfo::getRecordId, id).remove();
+        // 把model_atomic_result的结果也删除了
+        modelAtomicResultService.lambdaUpdate().eq(ModelAtomicResult::getRecordId, id).remove();
     }
 
 }