Deduplicate post view notifications

This commit is contained in:
Tim
2025-08-01 11:55:17 +08:00
parent e8e77b6467
commit db31b5d6c1
3 changed files with 28 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 deleteByTypeAndFromUserAndPost(NotificationType type, User fromUser, Post post);
}

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 && fromUser != null) {
notificationRepository.deleteByTypeAndFromUserAndPost(type, fromUser, post);
}
Notification n = new Notification();
n.setUser(user);
n.setType(type);

View File

@@ -168,4 +168,27 @@ 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 postViewedNotificationDeletesOldOnes() {
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);
org.springframework.test.util.ReflectionTestUtils.setField(service, "websiteUrl", "https://ex.com");
User owner = new User();
User viewer = new User();
Post post = new Post();
when(nRepo.save(any(Notification.class))).thenAnswer(i -> i.getArgument(0));
service.createNotification(owner, NotificationType.POST_VIEWED, post, null, null, viewer, null, null);
verify(nRepo).deleteByTypeAndFromUserAndPost(NotificationType.POST_VIEWED, viewer, post);
verify(nRepo).save(any(Notification.class));
}
}