✅ 버킷리스트 작성자, 쪽지 수신인 일치시키지

Moyeothon-NewTeams · Oct 31, 2024 · fea92cf · fea92cf
1 parent 934306f
commit fea92cf
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/main/java/com/example/moyeothon/Service/MessageService.java b/src/main/java/com/example/moyeothon/Service/MessageService.java
@@ -35,6 +35,9 @@ public MessageDTO createMessage(String uid, Long bucketListId, MessageDTO messag
         UserEntity sender = userRepository.findByUid(uid);
         UserEntity receiver = userRepository.findById(messageDTO.getReceiverId()).orElseThrow();
         BucketlistEntity bucketList = bucketRepository.findById(bucketListId).orElseThrow();
+        if (!bucketList.getUser().getUid().equals(receiver.getUid())) {
+            throw new RuntimeException("해당 버킷리스트의 생성자에게만 쪽지를 보낼 수 있습니다.");
+        }
         MessageEntity messageEntity = messageDTO.dtoToEntity(sender, receiver, bucketList);
         messageEntity.setCreateTime(LocalDateTime.now());
         messageEntity.setStatus(MessageStatus.안읽음);
@@ -51,6 +54,9 @@ public MessageDTO replyMessage(Long messageId, String uid, Long bucketListId, Me
         UserEntity sender = userRepository.findByUid(uid);
         UserEntity receiver = originalMessage.getSender();
         BucketlistEntity bucketList = bucketRepository.findById(bucketListId).orElseThrow();
+        if (!bucketList.getUser().getUid().equals(receiver.getUid())) {
+            throw new RuntimeException("해당 버킷리스트의 생성자에게만 쪽지를 답장할 수 있습니다.");
+        }
         MessageEntity messageEntity = messageDTO.dtoToEntity(sender, receiver, bucketList);
         messageEntity.setCreateTime(LocalDateTime.now());
         messageEntity.setStatus(MessageStatus.안읽음);