From d73a63855e06c633d5fd4d6d84a9428a165fca57 Mon Sep 17 00:00:00 2001
From: PushM <584406942@qq.com>
Date: Mon, 24 Jun 2024 16:59:54 +0800
Subject: [PATCH] =?UTF-8?q?1=E3=80=81=E5=A2=9E=E5=8A=A0=E9=80=80=E5=9B=9E?=
 =?UTF-8?q?=E4=BB=BB=E5=8A=A1=E9=80=9A=E7=9F=A5=E6=8E=A5=E5=8F=A3,?=
 =?UTF-8?q?=E5=B9=B6=E5=AE=8C=E5=96=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../realtime/protection/server/task/TaskMapper.java   |  2 ++
 .../realtime/protection/server/task/TaskService.java  | 11 ++++++++++-
 src/main/resources/mappers/TaskMapper.xml             |  8 ++++++++
 3 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/realtime/protection/server/task/TaskMapper.java b/src/main/java/com/realtime/protection/server/task/TaskMapper.java
index e4284ae..a0abe01 100644
--- a/src/main/java/com/realtime/protection/server/task/TaskMapper.java
+++ b/src/main/java/com/realtime/protection/server/task/TaskMapper.java
@@ -130,4 +130,6 @@ public interface TaskMapper {
     String queryEventTypeAct(String eventType);
 
     List<Task> queryAuditInfoNotification(String userId , Integer auditStatus);
+
+    void updateNotificationByTaskId(@Param("taskIds") List<Long> taskIdList);
 }
diff --git a/src/main/java/com/realtime/protection/server/task/TaskService.java b/src/main/java/com/realtime/protection/server/task/TaskService.java
index c14e085..09d1562 100644
--- a/src/main/java/com/realtime/protection/server/task/TaskService.java
+++ b/src/main/java/com/realtime/protection/server/task/TaskService.java
@@ -622,7 +622,16 @@ public class TaskService {
 
     }
 
+    @Transactional
     public List<Task> auditInfoNotification(String userId) {
-        return taskMapper.queryAuditInfoNotification(userId,AuditStatusEnum.RETURNED.getNum());
+        List<Task> tasksNotification = taskMapper.queryAuditInfoNotification(userId,AuditStatusEnum.RETURNED.getNum());
+        if (tasksNotification.isEmpty()){
+            return tasksNotification;
+        }
+        List<Long> taskIdList = tasksNotification.stream().map(Task::getTaskId).toList();
+        taskMapper.updateNotificationByTaskId(taskIdList);
+
+        return tasksNotification;
+
     }
 }
diff --git a/src/main/resources/mappers/TaskMapper.xml b/src/main/resources/mappers/TaskMapper.xml
index 50478ee..ac36640 100644
--- a/src/main/resources/mappers/TaskMapper.xml
+++ b/src/main/resources/mappers/TaskMapper.xml
@@ -304,6 +304,14 @@
         </foreach>
         AND expire_time IS NULL
     </update>
+    <update id="updateNotificationByTaskId">
+        UPDATE t_task
+        SET is_need_notification = 0
+        WHERE task_id IN
+        <foreach collection="taskIds" item="taskId" open="(" separator="," close=")">
+            #{taskId}
+        </foreach>
+    </update>
 
     <delete id="deleteTask">
         DELETE