Deduplicate post view notifications

This commit is contained in:
Tim
2025-08-01 11:47:43 +08:00
parent ba2299e882
commit 2fc22e7580
3 changed files with 30 additions and 0 deletions

View File

@@ -18,4 +18,6 @@ public interface NotificationRepository extends JpaRepository<Notification, Long
List<Notification> findByComment(Comment comment);
void deleteByTypeAndFromUser(NotificationType type, User fromUser);
void deleteByUserAndTypeAndPostAndFromUser(User user, NotificationType type, Post post, User fromUser);
}

View File

@@ -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);

View File

@@ -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));
}
}