diff --git a/src/main/java/com/ceos/bankids/service/ExpoNotificationServiceImpl.java b/src/main/java/com/ceos/bankids/service/ExpoNotificationServiceImpl.java index 0bbb5bd7..f4a6897a 100644 --- a/src/main/java/com/ceos/bankids/service/ExpoNotificationServiceImpl.java +++ b/src/main/java/com/ceos/bankids/service/ExpoNotificationServiceImpl.java @@ -49,25 +49,27 @@ public NotificationListDTO readNotificationList(User user, Long lastId) { List notificationDTOS = byUserIdOrderByIdDesc.stream() .map(NotificationDTO::new) .collect(Collectors.toList()); - NotificationDTO lastNotification = notificationDTOS.get(notificationDTOS.size() - 1); - Long lastNotificationId = lastNotification.getId(); if (notificationDTOS.size() == 11L) { + NotificationDTO lastNotification = notificationDTOS.get( + notificationDTOS.size() - 1); + Long lastNotificationId = lastNotification.getId(); notificationDTOS.remove(10); return new NotificationListDTO(lastNotificationId, false, notificationDTOS); } else if (notificationDTOS.size() < 11L) { - return new NotificationListDTO(lastNotificationId, true, notificationDTOS); + return new NotificationListDTO(null, true, notificationDTOS); } } List notificationDTOList = notificationRepository.findByIdLessThanEqualAndUserIdOrderByIdDesc( lastId, user.getId(), pageRequest).stream() .map(NotificationDTO::new).collect(Collectors.toList()); - NotificationDTO lastNotification = notificationDTOList.get(notificationDTOList.size() - 1); - Long last = lastNotification.getId(); if (notificationDTOList.size() == 11L) { + NotificationDTO lastNotification = notificationDTOList.get( + notificationDTOList.size() - 1); + Long last = lastNotification.getId(); notificationDTOList.remove(10); return new NotificationListDTO(last, false, notificationDTOList); } else { - return new NotificationListDTO(last, true, notificationDTOList); + return new NotificationListDTO(null, true, notificationDTOList); } }