Skip to content

[FIX] 비동기로 인해, ChatMessage가 두 번 저장하려하면서 락 장애 발생하는 오류 수정 #263

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
yooooonshine merged 1 commit into from
Nov 16, 2025
Merged

[FIX] 비동기로 인해, ChatMessage가 두 번 저장하려하면서 락 장애 발생하는 오류 수정 #263

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
29 changes: 10 additions & 19 deletions src/main/java/hanium/modic/backend/domain/ai/aiChat/service/AiChatMessageService.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,6 @@
import static hanium.modic.backend.common.error.ErrorCode.*;
import static hanium.modic.backend.domain.ai.aiServer.enums.AiImageStatus.*;

import java.util.List;

import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
Expand All @@ -19,7 +17,6 @@
import hanium.modic.backend.domain.ai.aiChat.entity.AiChatRoomEntity;
import hanium.modic.backend.domain.ai.aiChat.repository.AiChatMessageRepository;
import hanium.modic.backend.domain.ai.aiChat.repository.AiChatRoomRepository;
import hanium.modic.backend.domain.ai.aiServer.entity.AiChatImageEntity;
import hanium.modic.backend.domain.ai.aiServer.enums.SenderType;
import hanium.modic.backend.domain.ai.aiServer.repository.AiChatImageRepository;
import hanium.modic.backend.domain.ai.aiServer.service.AiServerService;
Expand Down Expand Up @@ -58,6 +55,8 @@ public class AiChatMessageService {
public ChatMessageResponse sendUserMessage(Long userId, Long postId, ChatMessageRequest request) {
// 메시지랑 이미지 둘 다 비어있으면 에러
validateRequestMessageNotEmpty(request);
// 이미지가 다른 사람의 이미지면 에러
validateAiChatImageOwnership(userId, request.aiChatImageId());

// 채팅방 조회
AiChatRoomEntity aiChatRoom = aiChatRoomRepository.findByUserIdAndPostId(userId, postId)
Expand Down Expand Up @@ -85,18 +84,8 @@ public ChatMessageResponse sendUserMessage(Long userId, Long postId, ChatMessage
.build();
aiChatMessageRepository.save(message);

// 이미지 조회
List<AiChatImageEntity> aiChatImages = List.of();
if (request.aiChatImageId() != null) {
aiChatImages = aiChatImageRepository.findById(request.aiChatImageId())
.map(List::of)
.orElse(List.of());
// 이미지들이 자신의 것인지 확인
validateAiChatImagesOwnership(userId, aiChatImages);
}

// Ai 요청
aiServerService.processAiRequest(userId, message, aiChatImages);
aiServerService.processAiRequest(userId, message.getId(), request.aiChatImageId());

// 이미지 유무에 따른 응답 생성
return createChatMessageResponseByAiChatImageId(message);
Expand Down Expand Up @@ -135,11 +124,13 @@ private void validateRequestMessageNotEmpty(ChatMessageRequest request) {
}

// 이미지들이 자신의 것인지 확인
private void validateAiChatImagesOwnership(Long userId, List<AiChatImageEntity> aiChatImages) {
for (AiChatImageEntity image : aiChatImages) {
if (!image.getUserId().equals(userId)) {
throw new AppException(ErrorCode.IMAGE_CAN_NOT_BE_STOLEN_EXCEPTION);
}
private void validateAiChatImageOwnership(Long userId, Long aiChatImageId) {
if (aiChatImageId == null) {
return; // 이미지 ID가 null인 경우 소유권 검증 불필요
}
boolean hasUserImage = aiChatImageRepository.existsByIdAndUserId(aiChatImageId, userId);
if (!hasUserImage) {
throw new AppException(ErrorCode.IMAGE_CAN_NOT_BE_STOLEN_EXCEPTION);
}
}

Expand Down
16 changes: 14 additions & 2 deletions src/main/java/hanium/modic/backend/domain/ai/aiServer/service/AiServerService.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,9 +67,21 @@ public class AiServerService {
@Async("llmTaskExecutor")
public void processAiRequest(
final Long nowUserId,
AiChatMessageEntity chatMessage,
List<AiChatImageEntity> aiChatImages // 없으면 빈 리스트
final Long messageId,
final Long aiChatImageId // 없으면 null, null이 아니면 해당 이미지는 반드시 존재해야 함
) {
Comment on lines +70 to 72
Copy link

@coderabbitai coderabbitai bot Nov 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Potential issue | 🟠 Major

메서드 계약과 구현 간의 불일치 확인 필요

Line 71의 주석은 aiChatImageId가 null이 아니면 해당 이미지가 반드시 존재해야 한다고 명시하고 있습니다. 하지만 Lines 80-82의 구현은 .orElse(List.of())를 사용하여 이미지가 없을 경우 조용히 빈 리스트를 반환합니다.

이는 다음과 같은 문제를 야기할 수 있습니다:

  • AiChatMessageService에서 이미지 존재 여부를 검증(Line 65)한 후, 비동기 메서드가 실행되기 전에 이미지가 삭제되면 조용히 무시됨
  • 디버깅이 어려워지고 예상치 못한 동작 발생 가능

다음 중 하나를 선택하여 일관성을 확보하세요:

옵션 1 (권장): 이미지가 없으면 예외 발생

 		// 이미지가 있으면 해당 이미지 조회
 		List<AiChatImageEntity> aiChatImages = List.of();
 		if (aiChatImageId != null) {
-			aiChatImages = aiChatImageRepository.findById(aiChatImageId)
-				.map(List::of)
-				.orElse(List.of());
+			AiChatImageEntity aiChatImage = aiChatImageRepository.findById(aiChatImageId)
+				.orElseThrow(() -> new AppException(ErrorCode.IMAGE_NOT_FOUND_EXCEPTION));
+			aiChatImages = List.of(aiChatImage);
 		}

옵션 2: 주석을 구현에 맞게 수정

-		final Long aiChatImageId // 없으면 null, null이 아니면 해당 이미지는 반드시 존재해야 함
+		final Long aiChatImageId // 없으면 null, null이 아니거나 이미지가 삭제된 경우 빈 리스트로 처리
📝 Committable suggestion

‼️ IMPORTANT
Carefully review the code before committing. Ensure that it accurately replaces the highlighted code, contains no missing lines, and has no issues with indentation. Thoroughly test & benchmark the code to ensure it meets the requirements.

Suggested change
final Long messageId,
final Long aiChatImageId // 없으면 null, null이 아니면 해당 이미지는 반드시 존재해야 함
) {
final Long messageId,
final Long aiChatImageId // 없으면 null, null이 아니거나 이미지가 삭제된 경우 빈 리스트로 처리
) {

// 비동기 통신이라 chatMessage를 외부에서 받아오지 않고 여기서 다시 조회
AiChatMessageEntity chatMessage = aiChatMessageRepository.findById(messageId)
.orElseThrow(() -> new AppException(ErrorCode.AI_CHAT_MESSAGE_NOT_FOUND));

// 이미지가 있으면 해당 이미지 조회
List<AiChatImageEntity> aiChatImages = List.of();
if (aiChatImageId != null) {
aiChatImages = aiChatImageRepository.findById(aiChatImageId)
.map(List::of)
.orElse(List.of());
}

final Long postId = chatMessage.getPostId();

// 해당 Post에 대한 AI 이미지 생성 권한 검증
Expand Down