From 2fc22e7580c6e070f45762ceb568facfb5b72ae2 Mon Sep 17 00:00:00 2001 From: Tim <135014430+nagisa77@users.noreply.github.com> Date: Fri, 1 Aug 2025 11:47:43 +0800 Subject: [PATCH] Deduplicate post view notifications --- .../repository/NotificationRepository.java | 2 ++ .../openisle/service/NotificationService.java | 3 +++ .../service/NotificationServiceTest.java | 25 +++++++++++++++++++ 3 files changed, 30 insertions(+) diff --git a/src/main/java/com/openisle/repository/NotificationRepository.java b/src/main/java/com/openisle/repository/NotificationRepository.java index f0d538c49..1e4dfd1c0 100644 --- a/src/main/java/com/openisle/repository/NotificationRepository.java +++ b/src/main/java/com/openisle/repository/NotificationRepository.java @@ -18,4 +18,6 @@ public interface NotificationRepository extends JpaRepository findByComment(Comment comment); void deleteByTypeAndFromUser(NotificationType type, User fromUser); + + void deleteByUserAndTypeAndPostAndFromUser(User user, NotificationType type, Post post, User fromUser); } diff --git a/src/main/java/com/openisle/service/NotificationService.java b/src/main/java/com/openisle/service/NotificationService.java index 561b59b28..c14352368 100644 --- a/src/main/java/com/openisle/service/NotificationService.java +++ b/src/main/java/com/openisle/service/NotificationService.java @@ -58,6 +58,9 @@ public class NotificationService { public Notification createNotification(User user, NotificationType type, Post post, Comment comment, Boolean approved, User fromUser, ReactionType reactionType, String content) { + if (type == NotificationType.POST_VIEWED && post != null) { + notificationRepository.deleteByUserAndTypeAndPostAndFromUser(user, type, post, fromUser); + } Notification n = new Notification(); n.setUser(user); n.setType(type); diff --git a/src/test/java/com/openisle/service/NotificationServiceTest.java b/src/test/java/com/openisle/service/NotificationServiceTest.java index 59d43d5e4..9906efbfc 100644 --- a/src/test/java/com/openisle/service/NotificationServiceTest.java +++ b/src/test/java/com/openisle/service/NotificationServiceTest.java @@ -168,4 +168,29 @@ class NotificationServiceTest { verify(email).sendEmail("a@a.com", "有人回复了你", "https://ex.com/posts/1#comment-2"); verify(push).sendNotification(eq(user), contains("/posts/1#comment-2")); } + + @Test + void postViewedDeletesOldNotification() { + NotificationRepository nRepo = mock(NotificationRepository.class); + UserRepository uRepo = mock(UserRepository.class); + ReactionRepository rRepo = mock(ReactionRepository.class); + EmailSender email = mock(EmailSender.class); + PushNotificationService push = mock(PushNotificationService.class); + Executor executor = Runnable::run; + NotificationService service = new NotificationService(nRepo, uRepo, email, push, rRepo, executor); + + User author = new User(); + author.setId(1L); + User viewer = new User(); + viewer.setId(2L); + Post post = new Post(); + post.setId(3L); + + when(nRepo.save(any(Notification.class))).thenAnswer(i -> i.getArgument(0)); + + service.createNotification(author, NotificationType.POST_VIEWED, post, null, null, viewer, null, null); + + verify(nRepo).deleteByUserAndTypeAndPostAndFromUser(author, NotificationType.POST_VIEWED, post, viewer); + verify(nRepo).save(any(Notification.class)); + } }