Feature: 교육 결승진출자, 우승자 계산, 재전송 API 구현

src/main/java/org/cotato/csquiz/api/education/controller/EducationController.java

@@ -13,6 +13,7 @@
 import org.cotato.csquiz.api.education.dto.WinnerInfoResponse;
 import org.cotato.csquiz.api.quiz.dto.KingMemberInfo;
 import org.cotato.csquiz.domain.education.service.EducationService;
+import org.cotato.csquiz.domain.education.service.KingMemberService;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.GetMapping;
@@ -30,6 +31,7 @@
 public class EducationController {
 
     private final EducationService educationService;
+    private final KingMemberService kingMemberService;
 
     @GetMapping
     public ResponseEntity<List<AllEducationResponse>> findEducationListByGeneration(
@@ -66,9 +68,37 @@ public ResponseEntity<List<KingMemberInfo>> findFinalKingMembers(@RequestParam("
         return ResponseEntity.ok().body(educationService.findKingMemberInfo(educationId));
     }
 
+    @PostMapping("/result/kings")
+    public ResponseEntity<Void> calculateKingMembers(@RequestParam("educationId") Long educationId) {
+        log.info("[{} 교육 결승진출자 계산하기]", educationId);
+        kingMemberService.saveKingMember(educationId);
+        return ResponseEntity.noContent().build();
+    }
+
+    @GetMapping("/send/kings")
+    public ResponseEntity<Void> sendKingCommand(@RequestParam("educationId") Long educationId) {
+        log.info("[{} 교육 결승진출자 재전송하기]", educationId);
+        kingMemberService.sendKingCommand(educationId);
+        return ResponseEntity.noContent().build();
+    }
+
     @GetMapping("/result/winner")
     public ResponseEntity<WinnerInfoResponse> findWinner(@RequestParam("educationId") Long educationId) {
         log.info("[{} 교육 우승자 조회 컨트롤러]", educationId);
         return ResponseEntity.ok().body(educationService.findWinner(educationId));
     }
+
+    @PostMapping("/result/winner")
+    public ResponseEntity<Void> calculateWinner(@RequestParam("educationId") Long educationId) {
+        log.info("[{} 교육 우승자 계산]", educationId);
+        kingMemberService.calculateWinner(educationId);
+        return ResponseEntity.noContent().build();
+    }
+
+    @GetMapping("/send/winner")
+    public ResponseEntity<Void> sendWinnerCommand(@RequestParam("educationId") Long educationId) {
+        log.info("[{} 교육 결승진출자 재전송하기]", educationId);
+        kingMemberService.sendWinnerCommand(educationId);
+        return ResponseEntity.noContent().build();
+    }
 }

src/main/java/org/cotato/csquiz/api/socket/dto/EducationResultResponse.java

@@ -0,0 +1,13 @@
+package org.cotato.csquiz.api.socket.dto;
+
+import org.cotato.csquiz.domain.education.entity.Education;
+
+public record EducationResultResponse(
+        String command,
+        Long educationId
+) {
+    public static EducationResultResponse of(String command, Long educationId) {
+        return new EducationResultResponse(command, educationId);
+    }
+
+}

src/main/java/org/cotato/csquiz/api/socket/dto/QuizStopResponse.java

@@ -4,9 +4,9 @@ public record QuizStopResponse(
         String command,
         Long quizId
 ) {
-    public static QuizStopResponse from(String command, Long quizId) {
+    public static QuizStopResponse from(Long quizId) {
         return new QuizStopResponse(
-                command,
+                "stop",
                 quizId
         );
     }

src/main/java/org/cotato/csquiz/common/error/ErrorCode.java

@@ -56,8 +56,12 @@ public enum ErrorCode {
 
     SUBJECT_INVALID(HttpStatus.BAD_REQUEST, "E-000", "교육 주제는 NULL이거나 비어있을 수 없습니다."),
 
+    KING_MEMBER_EXIST(HttpStatus.CONFLICT, "K-301", "이미 킹킹 멤버가 존재합니다"),
+
     ALREADY_REPLY_CORRECT(HttpStatus.BAD_REQUEST, "R-301", "해당 사용자는 이미 정답 처리되었습니다."),
 
+    LAST_QUIZ_SCORER_NOT_EXIST(HttpStatus.CONFLICT, "W-201", "아직 마지막 문제 득점자가 없습니다"),
+    WINNER_EXIST(HttpStatus.CONFLICT, "W-301", "이미 우승자가 존재합니다"),
     // 500 오류 -> 서버측에서 처리가 실패한 부분들
     WEBSOCKET_SEND_EXCEPTION(HttpStatus.INTERNAL_SERVER_ERROR, "S-001", "소캣 메세지 전송 실패"),
     IMAGE_PROCESSING_FAIL(HttpStatus.INTERNAL_SERVER_ERROR, "S-002", "이미지 처리에 실패했습니다."),

src/main/java/org/cotato/csquiz/common/websocket/WebSocketHandler.java

@@ -2,9 +2,11 @@
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import org.cotato.csquiz.api.socket.dto.CsQuizStopResponse;
+import org.cotato.csquiz.api.socket.dto.EducationResultResponse;
 import org.cotato.csquiz.api.socket.dto.QuizStartResponse;
 import org.cotato.csquiz.api.socket.dto.QuizStatusResponse;
 import org.cotato.csquiz.api.socket.dto.QuizStopResponse;
+import org.cotato.csquiz.domain.education.entity.Education;
 import org.cotato.csquiz.domain.education.entity.Quiz;
 import org.cotato.csquiz.domain.auth.enums.MemberRole;
 import org.cotato.csquiz.domain.auth.enums.MemberRoleGroup;
@@ -17,6 +19,7 @@
 import java.util.concurrent.ConcurrentHashMap;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.cotato.csquiz.domain.education.service.EducationService;
 import org.springframework.stereotype.Component;
 import org.springframework.web.socket.CloseStatus;
 import org.springframework.web.socket.TextMessage;
@@ -39,6 +42,7 @@ public class WebSocketHandler extends TextWebSocketHandler {
     private static final String ROLE_KEY = "role";
     private final ObjectMapper objectMapper = new ObjectMapper();
     private final QuizRepository quizRepository;
+    private final EducationService educationService;
 
     @Override
     public void afterConnectionEstablished(WebSocketSession session) {
@@ -118,14 +122,23 @@ public void startQuiz(Long quizId) {
     }
 
     public void stopQuiz(Quiz quiz) {
-        String command = "";
-        if (quiz.getNumber() == 9) {
-            command = KING_COMMAND;
+        QuizStopResponse response = QuizStopResponse.from(quiz.getId());
+        for (WebSocketSession clientSession : CLIENTS.values()) {
+            sendMessage(clientSession, response);
         }
-        if (quiz.getNumber() == 10) {
-            command = WINNER_COMMAND;
+    }
+
+    public void sendKingMemberCommand(Long educationId) {
+        EducationResultResponse response = EducationResultResponse.of(KING_COMMAND, educationId);
+
+        for (WebSocketSession clientSession : CLIENTS.values()) {
+            sendMessage(clientSession, response);
         }
-        QuizStopResponse response = QuizStopResponse.from(command, quiz.getId());
+    }
+
+    public void sendWinnerCommand(Long educationId) {
+        EducationResultResponse response = EducationResultResponse.of(WINNER_COMMAND, educationId);
+
         for (WebSocketSession clientSession : CLIENTS.values()) {
             sendMessage(clientSession, response);
         }

src/main/java/org/cotato/csquiz/domain/education/repository/KingMemberRepository.java

@@ -7,4 +7,6 @@
 
 public interface KingMemberRepository extends JpaRepository<KingMember, Long> {
     List<KingMember> findAllByEducation(Education education);
+
+    boolean existsByEducation(Education education);
 }

src/main/java/org/cotato/csquiz/domain/education/repository/QuizRepository.java

@@ -1,5 +1,6 @@
 package org.cotato.csquiz.domain.education.repository;
 
+import org.cotato.csquiz.domain.education.entity.Education;
 import org.cotato.csquiz.domain.education.entity.Quiz;
 import org.cotato.csquiz.domain.education.enums.QuizStatus;
 import java.util.List;
@@ -29,4 +30,8 @@ public interface QuizRepository extends JpaRepository<Quiz, Long> {
     @Modifying
     @Query("select q from Quiz q where q.education.id in :educationIds")
     List<Quiz> findAllByEducationIdsInQuery(@Param("educationIds") List<Long> educationIds);
+
+    Optional<Quiz> findByEducationIdAndNumber(Long educationId, Integer i);
+
+    Optional<Quiz> findFirstByEducationOrderByNumberDesc(Education education);
 }

src/main/java/org/cotato/csquiz/domain/education/repository/WinnerRepository.java

@@ -7,4 +7,6 @@
 
 public interface WinnerRepository extends JpaRepository<Winner, Long> {
     Optional<Winner> findByEducation(Education education);
+
+    boolean existsByEducation(Education education);
 }

src/main/java/org/cotato/csquiz/domain/education/service/KingMemberService.java

@@ -8,11 +8,15 @@
 import java.util.stream.Collectors;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.cotato.csquiz.common.error.ErrorCode;
+import org.cotato.csquiz.common.error.exception.AppException;
+import org.cotato.csquiz.common.websocket.WebSocketHandler;
 import org.cotato.csquiz.domain.education.entity.Education;
 import org.cotato.csquiz.domain.education.entity.KingMember;
 import org.cotato.csquiz.domain.education.entity.Quiz;
 import org.cotato.csquiz.domain.education.entity.Scorer;
 import org.cotato.csquiz.domain.education.entity.Winner;
+import org.cotato.csquiz.domain.education.repository.EducationRepository;
 import org.cotato.csquiz.domain.education.repository.QuizRepository;
 import org.cotato.csquiz.domain.education.repository.ScorerRepository;
 import org.cotato.csquiz.domain.auth.entity.Member;
@@ -30,18 +34,26 @@
 public class KingMemberService {
 
     private final MemberService memberService;
+    private final WebSocketHandler webSocketHandler;
     private final MemberRepository memberRepository;
     private final QuizRepository quizRepository;
     private final KingMemberRepository kingMemberRepository;
     private final ScorerRepository scorerRepository;
     private final WinnerRepository winnerRepository;
+    private final EducationRepository educationRepository;
 
-    public List<KingMember> calculateKingMember(Education education) {
-        List<Member> members = findKingMembersFromEducation(education);
+    @Transactional
+    public void saveKingMember(Long educationId) {
+        Education education = findEducationById(educationId);
+
+        checkKingMemberExist(education);
 
-        return members.stream()
+        List<KingMember> kingMembers = findKingMembersFromEducation(education).stream()
                 .map(member -> KingMember.of(member, education))
                 .toList();
+
+        saveKingMembers(kingMembers);
+        saveWinnerIfKingMemberIsOne(education);
     }
 
     private List<Member> findKingMembersFromEducation(Education education) {
@@ -64,13 +76,11 @@ private List<Member> findKingMembers(List<Scorer> scorers) {
                 .toList();
     }
 
-    @Transactional
-    public void saveKingMembers(List<KingMember> kingMembers) {
+    private void saveKingMembers(List<KingMember> kingMembers) {
         kingMemberRepository.saveAll(kingMembers);
     }
 
-    @Transactional
-    public void saveWinnerIfKingMemberIsOne(Education education) {
+    private void saveWinnerIfKingMemberIsOne(Education education) {
         List<KingMember> kingMembers = kingMemberRepository.findAllByEducation(education);
         if (kingMembers.size() == 1) {
             Member findMember = memberRepository.findById(kingMembers.get(0).getMemberId())
@@ -79,14 +89,27 @@ public void saveWinnerIfKingMemberIsOne(Education education) {
         }
     }
 
+    public void sendKingCommand(Long educationId) {
+        webSocketHandler.sendKingMemberCommand(educationId);
+    }
+
     @Transactional
-    public void saveWinnerIfNoWinnerExist(Quiz quiz) {
-        Education education = quiz.getEducation();
+    public void calculateWinner(Long educationId) {
+        Education education = findEducationById(educationId);
         if (isWinnerExist(education)) {
-            Scorer findScorer = scorerRepository.findByQuizId(quiz.getId())
-                    .orElseThrow(() -> new EntityNotFoundException("해당 퀴즈엔 득점자가 존재하지 않습니다."));
-            saveWinner(findScorer.getMemberId(), education);
+            throw new AppException(ErrorCode.WINNER_EXIST);
         }
+
+        Quiz quiz = quizRepository.findFirstByEducationOrderByNumberDesc(education)
+                .orElseThrow(() -> new EntityNotFoundException("마지막 문제가 없습니다"));
+        Scorer lastQuizScorer = scorerRepository.findByQuizId(quiz.getId())
+                .orElseThrow(() -> new EntityNotFoundException("마지막 퀴즈 득점자가 존재하지 않습니다."));
+
+        saveWinner(lastQuizScorer.getMemberId(), education);
+    }
+
+    public void sendWinnerCommand(Long educationId) {
+        webSocketHandler.sendWinnerCommand(educationId);
     }
 
     @Transactional
@@ -96,6 +119,17 @@ public void saveWinner(final Long memberId, final Education education) {
     }
 
     public boolean isWinnerExist(Education education) {
-        return winnerRepository.findByEducation(education).isPresent();
+        return winnerRepository.existsByEducation(education);
+    }
+
+    private Education findEducationById(Long educationId) {
+        return educationRepository.findById(educationId)
+                .orElseThrow(() -> new AppException(ErrorCode.ENTITY_NOT_FOUND));
+    }
+
+    private void checkKingMemberExist(Education education) {
+        if (kingMemberRepository.existsByEducation(education)) {
+            throw new AppException(ErrorCode.KING_MEMBER_EXIST);
+        }
     }
 }

src/main/java/org/cotato/csquiz/domain/education/service/SocketService.java

@@ -106,16 +106,8 @@ public void stopQuizSolve(QuizSocketRequest request) {
         checkEducationOpen(quiz.getEducation());
 
         quiz.updateStart(QuizStatus.QUIZ_OFF);
-        if (quiz.getNumber() == 9) {
-            List<KingMember> kingMembers = kingMemberService.calculateKingMember(quiz.getEducation());
-            kingMemberService.saveKingMembers(kingMembers);
-            kingMemberService.saveWinnerIfKingMemberIsOne(quiz.getEducation());
-            webSocketHandler.stopQuiz(quiz);
-        }
-        if (quiz.getNumber() == 10) {
-            kingMemberService.saveWinnerIfNoWinnerExist(quiz);
-            webSocketHandler.stopQuiz(quiz);
-        }
+
+        webSocketHandler.stopQuiz(quiz);
     }
 
     @Transactional