✨ 关注消息通知功能

src/main/java/com/rymcu/vertical/service/impl/FollowServiceImpl.java

@@ -1,10 +1,12 @@
 package com.rymcu.vertical.service.impl;
 
+import com.rymcu.vertical.core.constant.NotificationConstant;
 import com.rymcu.vertical.core.service.AbstractService;
 import com.rymcu.vertical.entity.Follow;
 import com.rymcu.vertical.entity.User;
 import com.rymcu.vertical.mapper.FollowMapper;
 import com.rymcu.vertical.service.FollowService;
+import com.rymcu.vertical.util.NotificationUtils;
 import com.rymcu.vertical.util.UserUtils;
 import com.rymcu.vertical.web.api.exception.BaseApiException;
 import org.springframework.stereotype.Service;
@@ -32,6 +34,9 @@ public class FollowServiceImpl extends AbstractService<Follow> implements Follow
         User tokenUser = UserUtils.getCurrentUserByToken();
         follow.setFollowerId(tokenUser.getIdUser());
         int result = followMapper.insertSelective(follow);
+        if (result > 0) {
+            NotificationUtils.saveNotification(follow.getFollowingId(), follow.getIdFollow(), NotificationConstant.Follow, tokenUser.getNickname() + " 关注了你!");
+        }
         return result > 0;
     }
 

src/main/java/com/rymcu/vertical/service/impl/NotificationServiceImpl.java

@@ -5,14 +5,13 @@ import com.rymcu.vertical.dto.ArticleDTO;
 import com.rymcu.vertical.dto.Author;
 import com.rymcu.vertical.dto.NotificationDTO;
 import com.rymcu.vertical.entity.Comment;
+import com.rymcu.vertical.entity.Follow;
 import com.rymcu.vertical.entity.Notification;
 import com.rymcu.vertical.entity.User;
 import com.rymcu.vertical.mapper.NotificationMapper;
-import com.rymcu.vertical.service.ArticleService;
+import com.rymcu.vertical.service.*;
-import com.rymcu.vertical.service.CommentService;
-import com.rymcu.vertical.service.NotificationService;
-import com.rymcu.vertical.service.UserService;
 import com.rymcu.vertical.util.BeanCopierUtil;
+import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -35,6 +34,10 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
     private CommentService commentService;
     @Resource
     private UserService userService;
+    @Resource
+    private FollowService followService;
+    @Value("${resource.domain}")
+    private String domain;
 
     @Override
     public List<Notification> findUnreadNotifications(Integer idUser) {
@@ -59,6 +62,7 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
         ArticleDTO article;
         Comment comment;
         User user;
+        Follow follow;
         switch (notification.getDataType()) {
             case "0":
                 // 系统公告/帖子
@@ -70,6 +74,11 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
                 break;
             case "1":
                 // 关注
+                follow = followService.findById(notification.getDataId().toString());
+                notificationDTO.setDataTitle("关注提醒");
+                user = userService.findById(follow.getFollowerId().toString());
+                notificationDTO.setDataUrl(getFollowLink(follow.getFollowingType(), user.getNickname()));
+                notificationDTO.setAuthor(genAuthor(user));
                 break;
             case "2":
                 // 回帖
@@ -80,10 +89,25 @@ public class NotificationServiceImpl extends AbstractService<Notification> imple
                 user = userService.findById(comment.getCommentAuthorId().toString());
                 notificationDTO.setAuthor(genAuthor(user));
                 break;
+            default:
+                break;
         }
         return notificationDTO;
     }
 
+    private String getFollowLink(String followingType, String id) {
+        StringBuilder url = new StringBuilder();
+        url.append(domain);
+        switch (followingType) {
+            case "0":
+                url = url.append("/user/").append(id);
+                break;
+            default:
+                url.append("/notification");
+        }
+        return url.toString();
+    }
+
     private Author genAuthor(User user) {
         Author author = new Author();
         author.setUserNickname(user.getNickname());