From f0c2b5dfd8c2ca7bd12f7d2a2a517a8bb97092db Mon Sep 17 00:00:00 2001
From: "DESKTOP-DDTUS3E\\yaxin" <daixiaoyi0622@gmail.com>
Date: Thu, 24 Oct 2024 14:40:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=B0=E5=AF=8C=E8=AF=A6=E6=83=85=E5=85=B3?=
 =?UTF-8?q?=E8=81=94=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../police/controller/PromptController.java          |  1 +
 .../com/supervision/police/domain/NotePrompt.java    |  5 +++--
 .../service/impl/ModelRecordTypeServiceImpl.java     |  2 +-
 .../police/service/impl/NotePromptServiceImpl.java   | 12 ++++++++++++
 4 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/supervision/police/controller/PromptController.java b/src/main/java/com/supervision/police/controller/PromptController.java
index fe4307f..2c0fdc9 100644
--- a/src/main/java/com/supervision/police/controller/PromptController.java
+++ b/src/main/java/com/supervision/police/controller/PromptController.java
@@ -48,4 +48,5 @@ public class PromptController {
     public R<NotePromptDTO> getById(@RequestBody NotePrompt notePrompt) {
         return R.ok(promptService.getById(notePrompt.getId()));
     }
+
 }
diff --git a/src/main/java/com/supervision/police/domain/NotePrompt.java b/src/main/java/com/supervision/police/domain/NotePrompt.java
index 994b245..2a9cc5d 100644
--- a/src/main/java/com/supervision/police/domain/NotePrompt.java
+++ b/src/main/java/com/supervision/police/domain/NotePrompt.java
@@ -13,6 +13,7 @@ import lombok.Data;
 import java.io.Serial;
 import java.io.Serializable;
 import java.time.LocalDateTime;
+import java.util.ArrayList;
 import java.util.List;
 
 @TableName(value = "note_prompt", autoResultMap = true)
@@ -33,7 +34,7 @@ public class NotePrompt implements Serializable {
      * prompt关联的类型
      */
     @TableField(exist = false)
-    private List<String> typeList;
+    private List<String> typeList = new ArrayList<>();
 
     /**
      * 提示词
@@ -72,7 +73,7 @@ public class NotePrompt implements Serializable {
 
 
     @TableField(exist = false)
-    private List<TripleInfoDTO> tripleList;
+    private List<TripleInfoDTO> tripleList = new ArrayList<>();
 
     /**
      * 创建人ID
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 ed06f29..9127d6a 100644
--- a/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/ModelRecordTypeServiceImpl.java
@@ -104,7 +104,7 @@ public class ModelRecordTypeServiceImpl extends ServiceImpl<ModelRecordTypeMappe
         return list;
     }
 
-    private List<TripleInfoDTO> buildTripleInfo(NotePrompt notePrompt) {
+    public static List<TripleInfoDTO> buildTripleInfo(NotePrompt notePrompt) {
         List<TripleInfoDTO> list = new ArrayList<>();
         TripleInfoDTO dto = new TripleInfoDTO();
         dto.setType("头节点");
diff --git a/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java b/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
index 1e46dda..b28989e 100644
--- a/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
+++ b/src/main/java/com/supervision/police/service/impl/NotePromptServiceImpl.java
@@ -3,6 +3,7 @@ package com.supervision.police.service.impl;
 import cn.hutool.core.collection.CollUtil;
 import cn.hutool.core.util.StrUtil;
 import cn.hutool.json.JSONUtil;
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
@@ -37,6 +38,8 @@ import java.nio.charset.StandardCharsets;
 import java.util.*;
 import java.util.stream.Collectors;
 
+import static com.supervision.police.service.impl.ModelRecordTypeServiceImpl.buildTripleInfo;
+
 @Slf4j
 @Service
 @RequiredArgsConstructor
@@ -185,6 +188,15 @@ public class NotePromptServiceImpl extends ServiceImpl<NotePromptMapper, NotePro
         NotePrompt notePrompt = super.getById(id);
         NotePromptDTO notePromptDTO = new NotePromptDTO();
         BeanUtils.copyProperties(notePrompt, notePromptDTO);
+
+        notePromptDTO.setTripleList(buildTripleInfo(notePrompt));
+        //根据notePrompt的ID调用notePromptTypeRelService查询prompt_id相等的list
+
+        List<NotePromptTypeRel> notePromptTypeRels = notePromptTypeRelService.list(new LambdaQueryWrapper<NotePromptTypeRel>().eq(NotePromptTypeRel::getPromptId, notePrompt.getId()));
+        if (notePromptTypeRels != null && !notePromptTypeRels.isEmpty()) {
+            notePromptDTO.setTypeList(notePromptTypeRels.stream().map(NotePromptTypeRel::getTypeId).collect(Collectors.toList()));
+        }
+
         String evidenceCategoryId = notePromptDTO.getEvidenceCategoryId();
         if (StringUtils.isNotEmpty(evidenceCategoryId)) {
             EvidenceCategory category = evidenceCategoryService.getById(evidenceCategoryId);