Merge pull request #188 from dd-jiyun/main

메세지함 수정
SKHU-TITTO · Mar 24, 2024 · ff08af3 · ff08af3
2 parents 125cf96 + 260300e
commit ff08af3
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 3 deletions.
diff --git a/...Backend/src/main/java/com/example/titto_backend/message/controller/MessageController.java b/...Backend/src/main/java/com/example/titto_backend/message/controller/MessageController.java
@@ -48,7 +48,7 @@ public ResponseEntity<String> writeMessage(@RequestBody MessageDTO.Request reque
     // 쪽지함 목록 조회
     @GetMapping("/all")
     public ResponseEntity<List<MessageDTO.Preview>> getAllMessages(@AuthenticationPrincipal UserDetails userDetails) {
-        List<Message> userMessages = messageService.getUserMessages(userDetails.getUsername());
+        List<Message> userMessages = messageService.getUserConversationsMessages(userDetails.getUsername());
         List<MessageDTO.Preview> previews = userMessages.stream()
                 .map(MessageDTO.Preview::new)
                 .collect(Collectors.toList());

diff --git a/...Backend/src/main/java/com/example/titto_backend/message/repository/MessageRepository.java b/...Backend/src/main/java/com/example/titto_backend/message/repository/MessageRepository.java
@@ -15,4 +15,7 @@ List<Message> findBySenderAndReceiverAndDeletedBySenderFalseOrReceiverAndSenderA
 
     List<Message> findBySenderAndDeletedBySenderFalseOrReceiverAndDeletedByReceiverFalseOrderBySentAtDesc(User sender, User receiver);
 
+    List<Message> findBySenderAndDeletedBySenderFalseOrderBySentAtDesc(User user);
+
+    List<Message> findByReceiverAndDeletedByReceiverFalseOrderBySentAtDesc(User user);
 }
diff --git a/Titto_Backend/src/main/java/com/example/titto_backend/message/service/MessageService.java b/Titto_Backend/src/main/java/com/example/titto_backend/message/service/MessageService.java
@@ -7,6 +7,8 @@
 import com.example.titto_backend.message.domain.Message;
 import com.example.titto_backend.message.dto.MessageDTO;
 import com.example.titto_backend.message.repository.MessageRepository;
+import java.util.ArrayList;
+import java.util.Comparator;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -54,14 +56,32 @@ public List<MessageDTO.Response> getBothMessages(String email, Long selectedUser
     }
 
     @Transactional
-    public List<Message> getUserMessages(String email) {
+    public List<Message> getUserConversationsMessages(String email) {
         User user = userRepository.findByEmail(email)
                 .orElseThrow(() -> new CustomException(ErrorCode.USER_NOT_FOUND));
 
-        List<Message> userMessages = messageRepository.findBySenderAndDeletedBySenderFalseOrReceiverAndDeletedByReceiverFalseOrderBySentAtDesc(user, user);
+        List<Message> userSentMessages = messageRepository.findBySenderAndDeletedBySenderFalseOrderBySentAtDesc(user);
+        List<Message> userReceivedMessages = messageRepository.findByReceiverAndDeletedByReceiverFalseOrderBySentAtDesc(user);
+
+        List<Message> userMessages = new ArrayList<>();
+        userMessages.addAll(userSentMessages);
+        userMessages.addAll(userReceivedMessages);
+
+        // 메시지를 시간 순서대로 정렬
+        userMessages.sort(Comparator.comparing(Message::getSentAt).reversed());
+
         return userMessages;
     }
 
+//    @Transactional
+//    public List<Message> getUserMessages(String email) {
+//        User user = userRepository.findByEmail(email)
+//                .orElseThrow(() -> new CustomException(ErrorCode.USER_NOT_FOUND));
+//
+//        List<Message> userMessages = messageRepository.findBySenderAndDeletedBySenderFalseOrReceiverAndDeletedByReceiverFalseOrderBySentAtDesc(user, user);
+//        return userMessages;
+//    }
+
     // 메세지함에서 사용자 별 가장 최근 메세지만 보내줌.
 //    @Transactional
 //    public Map<User, Message> getUserConversations(String email) {