From eacb753e55ea769f05bcf83a8cd7f18d3b8b2313 Mon Sep 17 00:00:00 2001
From: ronger <ronger-x@outlook.com>
Date: Wed, 24 Feb 2021 09:08:56 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E6=B6=88=E6=81=AF=E9=80=9A=E7=9F=A5?=
 =?UTF-8?q?=E5=88=86=E9=A1=B5=E9=97=AE=E9=A2=98=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/rymcu/forest/mapper/NotificationMapper.java   |  3 ++-
 .../forest/service/impl/NotificationServiceImpl.java  |  9 ++++-----
 src/main/java/mapper/NotificationMapper.xml           | 11 ++++++++++-
 3 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java b/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
index 8a3dd6f..2ae27ad 100644
--- a/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
+++ b/src/main/java/com/rymcu/forest/mapper/NotificationMapper.java
@@ -1,6 +1,7 @@
 package com.rymcu.forest.mapper;
 
 import com.rymcu.forest.core.mapper.Mapper;
+import com.rymcu.forest.dto.NotificationDTO;
 import com.rymcu.forest.entity.Notification;
 import org.apache.ibatis.annotations.Param;
 
@@ -23,7 +24,7 @@ public interface NotificationMapper extends Mapper<Notification> {
      * @param idUser
      * @return
      */
-    List<Notification> selectNotifications(@Param("idUser") Integer idUser);
+    List<NotificationDTO> selectNotifications(@Param("idUser") Integer idUser);
 
     /**
      * 获取消息数据
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 7aef442..4883182 100644
--- a/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java
+++ b/src/main/java/com/rymcu/forest/service/impl/NotificationServiceImpl.java
@@ -49,13 +49,12 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
 
     @Override
     public List<NotificationDTO> findNotifications(Integer idUser) {
-        List<Notification> list = notificationMapper.selectNotifications(idUser);
-        List<NotificationDTO> notifications = new ArrayList<>();
+        List<NotificationDTO> list = notificationMapper.selectNotifications(idUser);
         list.forEach(notification -> {
             NotificationDTO notificationDTO = genNotification(notification);
             // 判断关联数据是否已删除
             if (Objects.nonNull(notificationDTO.getAuthor())) {
-                notifications.add(notificationDTO);
+                BeanCopierUtil.copy(notificationDTO, notification);
             } else {
                 // 关联数据已删除,且未读
                 if (unRead.equals(notification.getHasRead())) {
@@ -66,10 +65,10 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
                 dto.setDataType("-1");
                 dto.setHasRead("1");
                 dto.setCreatedTime(notification.getCreatedTime());
-                notifications.add(dto);
+                BeanCopierUtil.copy(dto, notification);
             }
         });
-        return notifications;
+        return list;
     }
 
     private NotificationDTO genNotification(Notification notification) {
diff --git a/src/main/java/mapper/NotificationMapper.xml b/src/main/java/mapper/NotificationMapper.xml
index 81fa9b0..80ef662 100644
--- a/src/main/java/mapper/NotificationMapper.xml
+++ b/src/main/java/mapper/NotificationMapper.xml
@@ -10,6 +10,15 @@
         <result column="has_read" property="hasRead"></result>
         <result column="created_time" property="createdTime"></result>
     </resultMap>
+    <resultMap id="DTOResultMapper" type="com.rymcu.forest.dto.NotificationDTO">
+        <id column="id" property="idNotification"></id>
+        <result column="id_user" property="idUser"></result>
+        <result column="data_type" property="dataType"></result>
+        <result column="data_id" property="dataId"></result>
+        <result column="data_summary" property="dataSummary"></result>
+        <result column="has_read" property="hasRead"></result>
+        <result column="created_time" property="createdTime"></result>
+    </resultMap>
     <insert id="insertNotification">
         insert into forest_notification (id_user, data_type, data_id, data_summary, created_time) values (#{idUser}, #{dataType}, #{dataId}, #{dataSummary}, sysdate())
     </insert>
@@ -19,7 +28,7 @@
     <select id="selectUnreadNotifications" resultMap="BaseResultMapper">
         select * from forest_notification where has_read = '0' and id_user = #{idUser} order by created_time desc
     </select>
-    <select id="selectNotifications" resultMap="BaseResultMapper">
+    <select id="selectNotifications" resultMap="DTOResultMapper">
         select * from forest_notification where id_user = #{idUser} order by created_time desc
     </select>
     <select id="selectNotification" resultMap="BaseResultMapper">