feat: show unread message count

src/main/java/com/openisle/controller/NotificationController.java

@@ -29,6 +29,14 @@ public class NotificationController {
                 .collect(Collectors.toList());
     }
 
+    @GetMapping("/unread-count")
+    public UnreadCount unreadCount(Authentication auth) {
+        long count = notificationService.countUnread(auth.getName());
+        UnreadCount uc = new UnreadCount();
+        uc.setCount(count);
+        return uc;
+    }
+
     @PostMapping("/read")
     public void markRead(@RequestBody MarkReadRequest req, Authentication auth) {
         notificationService.markRead(auth.getName(), req.getIds());
@@ -115,4 +123,9 @@ public class NotificationController {
         private String username;
         private String avatar;
     }
+
+    @Data
+    private static class UnreadCount {
+        private long count;
+    }
 }

src/main/java/com/openisle/repository/NotificationRepository.java

@@ -10,4 +10,5 @@ import java.util.List;
 public interface NotificationRepository extends JpaRepository<Notification, Long> {
     List<Notification> findByUserOrderByCreatedAtDesc(User user);
     List<Notification> findByUserAndReadOrderByCreatedAtDesc(User user, boolean read);
+    long countByUserAndRead(User user, boolean read);
 }

src/main/java/com/openisle/service/NotificationService.java

@@ -45,4 +45,10 @@ public class NotificationService {
         }
         notificationRepository.saveAll(notifs);
     }
+
+    public long countUnread(String username) {
+        User user = userRepository.findByUsername(username)
+                .orElseThrow(() -> new IllegalArgumentException("User not found"));
+        return notificationRepository.countByUserAndRead(user, false);
+    }
 }

src/test/java/com/openisle/controller/NotificationControllerTest.java

@@ -60,4 +60,14 @@ class NotificationControllerTest {
 
         verify(notificationService).markRead("alice", List.of(1L,2L));
     }
+
+    @Test
+    void unreadCountEndpoint() throws Exception {
+        Mockito.when(notificationService.countUnread("alice")).thenReturn(3L);
+
+        mockMvc.perform(get("/api/notifications/unread-count")
+                        .principal(new UsernamePasswordAuthenticationToken("alice","p")))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.count").value(3));
+    }
 }

src/test/java/com/openisle/service/NotificationServiceTest.java

@@ -60,4 +60,22 @@ class NotificationServiceTest {
         assertEquals(1, list.size());
         verify(nRepo).findByUserOrderByCreatedAtDesc(user);
     }
+
+    @Test
+    void countUnreadReturnsRepositoryValue() {
+        NotificationRepository nRepo = mock(NotificationRepository.class);
+        UserRepository uRepo = mock(UserRepository.class);
+        NotificationService service = new NotificationService(nRepo, uRepo);
+
+        User user = new User();
+        user.setId(3L);
+        user.setUsername("carl");
+        when(uRepo.findByUsername("carl")).thenReturn(Optional.of(user));
+        when(nRepo.countByUserAndRead(user, false)).thenReturn(5L);
+
+        long count = service.countUnread("carl");
+
+        assertEquals(5L, count);
+        verify(nRepo).countByUserAndRead(user, false);
+    }
 }