From 930dca3b59f5f3dfbe6b8d9884af42e41292cc3b Mon Sep 17 00:00:00 2001
From: ronger <ronger-x@outlook.com>
Date: Wed, 25 May 2022 21:36:53 +0800
Subject: [PATCH] =?UTF-8?q?:sparkles:=20=E6=B6=88=E6=81=AF=E4=B8=AD?=
 =?UTF-8?q?=E5=BF=83-=E4=B8=80=E9=94=AE=E6=A0=87=E8=AE=B0=E5=B7=B2?=
 =?UTF-8?q?=E8=AF=BB=E5=8A=9F=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../java/com/rymcu/forest/mapper/NotificationMapper.java | 7 +++++++
 .../com/rymcu/forest/service/NotificationService.java    | 9 +++++++++
 .../forest/service/impl/NotificationServiceImpl.java     | 7 +++++++
 .../web/api/notification/NotificationController.java     | 9 +++++++++
 src/main/java/mapper/NotificationMapper.xml              | 3 +++
 5 files changed, 35 insertions(+)

diff --git a/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java b/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
index 2ae27ad..fee64a9 100644
--- a/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
+++ b/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
@@ -51,4 +51,11 @@ public interface NotificationMapper extends Mapper<Notification> {
      * @return
      */
     Integer readNotification(@Param("id") Integer id);
+
+    /**
+     * 标记所有消息已读
+     * @param idUser
+     * @return
+     */
+    Integer readAllNotification(@Param("idUser") Integer idUser);
 }
diff --git a/src/main/java/com/rymcu/forest/service/NotificationService.java b/src/main/java/com/rymcu/forest/service/NotificationService.java
index 4cf634b..da7109b 100644
--- a/src/main/java/com/rymcu/forest/service/NotificationService.java
+++ b/src/main/java/com/rymcu/forest/service/NotificationService.java
@@ -3,6 +3,7 @@ package com.rymcu.forest.service;
 import com.rymcu.forest.core.service.Service;
 import com.rymcu.forest.dto.NotificationDTO;
 import com.rymcu.forest.entity.Notification;
+import com.rymcu.forest.web.api.exception.BaseApiException;
 
 import java.util.List;
 
@@ -47,6 +48,14 @@ public interface NotificationService extends Service<Notification> {
     /**
      * 标记消息已读
      * @param id
+     * @return
      */
     Integer readNotification(Integer id);
+
+    /**
+     * 标记所有消息已读
+     * @return
+     * @throws BaseApiException
+     */
+    Integer readAllNotification() throws BaseApiException;
 }
diff --git a/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java b/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java
index b571e6f..c409d8e 100644
--- a/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java
+++ b/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java
@@ -7,6 +7,8 @@ import com.rymcu.forest.mapper.NotificationMapper;
 import com.rymcu.forest.service.NotificationService;
 import com.rymcu.forest.util.BeanCopierUtil;
 import com.rymcu.forest.util.NotificationUtils;
+import com.rymcu.forest.util.UserUtils;
+import com.rymcu.forest.web.api.exception.BaseApiException;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -70,4 +72,9 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
     public Integer readNotification(Integer id) {
         return notificationMapper.readNotification(id);
     }
+
+    @Override
+    public Integer readAllNotification() throws BaseApiException {
+        return notificationMapper.readAllNotification(Objects.requireNonNull(UserUtils.getCurrentUserByToken()).getIdUser());
+    }
 }
diff --git a/src/main/java/com/rymcu/forest/web/api/notification/NotificationController.java b/src/main/java/com/rymcu/forest/web/api/notification/NotificationController.java
index 42c5072..ac8e403 100644
--- a/src/main/java/com/rymcu/forest/web/api/notification/NotificationController.java
+++ b/src/main/java/com/rymcu/forest/web/api/notification/NotificationController.java
@@ -65,4 +65,13 @@ public class NotificationController {
         return GlobalResultGenerator.genSuccessResult("标记已读成功");
     }
 
+    @PutMapping("/read-all")
+    public GlobalResult readAll() throws BaseApiException {
+        Integer result = notificationService.readAllNotification();
+        if (result == 0) {
+            return GlobalResultGenerator.genErrorResult("标记已读失败");
+        }
+        return GlobalResultGenerator.genSuccessResult("标记已读成功");
+    }
+
 }
diff --git a/src/main/java/mapper/NotificationMapper.xml b/src/main/java/mapper/NotificationMapper.xml
index 144d4ca..3e68965 100644
--- a/src/main/java/mapper/NotificationMapper.xml
+++ b/src/main/java/mapper/NotificationMapper.xml
@@ -25,6 +25,9 @@
     <update id="readNotification">
         update forest_notification set has_read = '1' where id = #{id}
     </update>
+    <update id="readAllNotification">
+        update forest_notification set has_read = '1' where id_user = #{idUser} and has_read = '0'
+    </update>
     <select id="selectUnreadNotifications" resultMap="BaseResultMapper">
         select * from forest_notification where has_read = '0' and id_user = #{idUser} order by created_time desc
     </select>