Merge pull request #257 from CSID-DGU/develop

7월 6일 작업분

Author: dongmin0204

src/main/java/DGU_AI_LAB/admin_be/domain/alarm/dto/SlackMessageDto.java

@@ -18,13 +18,20 @@ public enum MessageType {
         DM       // 사용자 개인 DM
     }
 
-    private MessageType type;   // 메시지 타입 구분
-    private String message;     // 보낼 메시지 내용
+    private MessageType type;
+    private String message;
 
     // Webhook용 필드
     private String webhookUrl;
 
     // DM용 필드
     private String username;
     private String email;
+
+    @Builder.Default
+    private int retryCount = 0;
+
+    public void incrementRetryCount() {
+        this.retryCount++;
+    }
 }

src/main/java/DGU_AI_LAB/admin_be/domain/requests/dto/request/SaveRequestRequestDTO.java

@@ -11,8 +11,10 @@
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.swagger.v3.oas.annotations.media.Schema;
+import jakarta.validation.constraints.Future;
 import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.NotNull;
+import jakarta.validation.constraints.Positive;
 import jakarta.validation.constraints.Size;
 import lombok.Builder;
 
@@ -41,15 +43,20 @@ public record SaveRequestRequestDTO(
         String ubuntuPassword,
 
         @Schema(description = "볼륨 사이즈 (GiB)", example = "20")
+        @NotNull(message = "Volume size cannot be null")
+        @Positive(message = "Volume size must be positive")
         Long volumeSizeGiB,
 
         @Schema(description = "사용 목적", example = "딥러닝 모델 학습")
+        @NotBlank(message = "Usage purpose cannot be blank")
         String usagePurpose,
 
         @Schema(description = "폼 응답", example = "{\"question\": \"answer\"}")
         Map<String, Object> formAnswers,
 
         @Schema(description = "서버 만료 일시", example = "2026-12-31T23:59:59")
+        @NotNull(message = "Expires date cannot be null")
+        @Future(message = "Expires date must be in the future")
         LocalDateTime expiresAt,
         @Schema(description = "Ubuntu GID 목록", example = "[1005, 1006]")
         Set<Long> ubuntuGids,

src/main/java/DGU_AI_LAB/admin_be/domain/requests/dto/response/AcceptInfoResponseDTO.java

@@ -1,5 +1,6 @@
 package DGU_AI_LAB.admin_be.domain.requests.dto.response;
 
+import DGU_AI_LAB.admin_be.domain.nodes.entity.Node;
 import DGU_AI_LAB.admin_be.domain.portRequests.entity.PortRequests;
 import DGU_AI_LAB.admin_be.domain.requests.entity.Request;
 import io.swagger.v3.oas.annotations.media.Schema;
@@ -15,15 +16,37 @@ public record AcceptInfoResponseDTO(
         String username,
         @Schema(description = "컨테이너 이미지 (이름:버전)", example = "cuda:11.8")
         String image,
-        @Schema(description = "Ubuntu GID 목록", example = "[1005, 1006]")
-        List<Long> gid,
+        @Schema(description = "그룹 목록 (GID + 그룹명)")
+        List<GroupDTO> groups,
         @Schema(description = "볼륨 크기 (GiB)", example = "20")
         Long volume_size,
-        @Schema(description = "GPU 필요 여부", example = "true")
-        Boolean gpu_required,
+        @Schema(description = "GPU 노드 목록")
+        List<GpuNodeDTO> gpu_nodes,
         @Schema(description = "추가 포트 목록")
         List<AdditionalPortDTO> additional_ports
 ) {
+    @Schema(description = "그룹 정보")
+    @Builder
+    public record GroupDTO(
+            @Schema(description = "Ubuntu GID", example = "10004")
+            Long gid,
+            @Schema(description = "그룹명", example = "ailab")
+            String name
+    ) {}
+
+    @Schema(description = "GPU 노드 정보")
+    @Builder
+    public record GpuNodeDTO(
+            @Schema(description = "노드명", example = "farm2")
+            String node_name,
+            @Schema(description = "GPU 수", example = "2")
+            int num_gpu,
+            @Schema(description = "CPU 제한 (k8s 포맷)", example = "4000m")
+            String cpu_limit,
+            @Schema(description = "메모리 제한 (k8s 포맷)", example = "8192Mi")
+            String memory_limit
+    ) {}
+
     @Schema(description = "추가 포트 정보")
     @Builder
     public record AdditionalPortDTO(
@@ -33,26 +56,38 @@ public record AdditionalPortDTO(
             String usage_purpose
     ) {}
 
-    public static AcceptInfoResponseDTO fromEntity(Request request, List<PortRequests> portRequests) {
+    public static AcceptInfoResponseDTO fromEntity(Request request, List<PortRequests> portRequests, List<Node> nodes) {
         var image = request.getContainerImage();
 
+        List<GroupDTO> groupDTOList = request.getRequestGroups().stream()
+                .map(rg -> GroupDTO.builder()
+                        .gid(rg.getGroup().getUbuntuGid())
+                        .name(rg.getGroup().getGroupName())
+                        .build())
+                .toList();
+
+        List<GpuNodeDTO> gpuNodeDTOList = nodes.stream()
+                .map(node -> GpuNodeDTO.builder()
+                        .node_name(node.getNodeId())
+                        .num_gpu(node.getNumberGpu())
+                        .cpu_limit(node.getCpuCoreCount() * 1000 + "m")
+                        .memory_limit(node.getMemorySizeGB() * 1024 + "Mi")
+                        .build())
+                .toList();
+
         List<AdditionalPortDTO> additionalPortDTOList = portRequests.stream()
                 .map(portRequest -> AdditionalPortDTO.builder()
                         .internal_port(portRequest.getInternalPort())
                         .usage_purpose(portRequest.getUsagePurpose())
-                        .build()
-                ).toList();
+                        .build())
+                .toList();
 
         return AcceptInfoResponseDTO.builder()
                 .username(request.getUbuntuUsername())
                 .image(image.getImageName() + ":" + image.getImageVersion())
-                .gid(
-                        request.getRequestGroups().stream()
-                                .map(rg -> rg.getGroup().getUbuntuGid())
-                                .toList()
-                )
+                .groups(groupDTOList)
                 .volume_size(request.getVolumeSizeGiB())
-                .gpu_required(true)
+                .gpu_nodes(gpuNodeDTOList)
                 .additional_ports(additionalPortDTOList)
                 .build();
     }

src/main/java/DGU_AI_LAB/admin_be/domain/requests/dto/response/ContainerInfoDTO.java

@@ -16,6 +16,10 @@ public record ContainerInfoDTO(
         String userName,
         @Schema(description = "Ubuntu 사용자명", example = "test2014")
         String ubuntuUsername,
+        @Schema(description = "Pod 이름", example = "pod-test2014-mock")
+        String podName,
+        @Schema(description = "배포된 노드명", example = "farm1")
+        String nodeName,
         @Schema(description = "Ubuntu GID 목록", example = "[1005, 1006]")
         List<Long> ubuntuGids,
         @Schema(description = "리소스 그룹 ID", example = "1")
@@ -32,6 +36,8 @@ public static ContainerInfoDTO fromEntity(Request request) {
                 .userId(request.getUser().getUserId())
                 .userName(request.getUser().getName())
                 .ubuntuUsername(request.getUbuntuUsername())
+                .podName(request.getPodName())
+                .nodeName(request.getNodeName())
                 .ubuntuGids(
                         request.getRequestGroups().stream()
                                 .map(rg -> rg.getGroup().getUbuntuGid())

src/main/java/DGU_AI_LAB/admin_be/domain/requests/service/ConfigRequestService.java

@@ -1,5 +1,7 @@
 package DGU_AI_LAB.admin_be.domain.requests.service;
 
+import DGU_AI_LAB.admin_be.domain.nodes.entity.Node;
+import DGU_AI_LAB.admin_be.domain.nodes.repository.NodeRepository;
 import DGU_AI_LAB.admin_be.domain.portRequests.entity.PortRequests;
 import DGU_AI_LAB.admin_be.domain.portRequests.repository.PortRequestRepository;
 import DGU_AI_LAB.admin_be.domain.requests.dto.response.AcceptInfoResponseDTO;
@@ -21,6 +23,7 @@ public class ConfigRequestService {
 
     private final RequestRepository requestRepository;
     private final PortRequestRepository portRequestRepository;
+    private final NodeRepository nodeRepository;
 
     /** ubuntu username 중복 검사 */
     @Transactional(readOnly = true)
@@ -29,10 +32,10 @@ public boolean isUbuntuUsernameAvailable(String username) {
     }
 
     /** config server용 acceptinfo */
+    @Transactional(readOnly = true)
     public AcceptInfoResponseDTO getAcceptInfo(String username) {
         log.info("사용자 승인 정보 조회를 시작합니다. username: {}", username);
 
-        // 사용자 요청 정보 조회
         Request request = requestRepository.findByUbuntuUsername(username)
                 .orElseThrow(() -> {
                     log.warn("사용자 '{}'에 대한 승인 정보가 존재하지 않습니다.", username);
@@ -41,12 +44,13 @@ public AcceptInfoResponseDTO getAcceptInfo(String username) {
 
         log.debug("사용자 '{}'의 요청 정보를 성공적으로 찾았습니다. 요청 ID: {}", username, request.getRequestId());
 
-        // 포트 요청 정보 조회
         List<PortRequests> portRequests = portRequestRepository.findByRequestRequestId(request.getRequestId());
         log.debug("요청 '{}'에 속한 포트 요청 {}개를 조회했습니다.", request.getRequestId(), portRequests.size());
 
-        // 응답 DTO 생성 및 반환
-        AcceptInfoResponseDTO response = AcceptInfoResponseDTO.fromEntity(request, portRequests);
+        List<Node> nodes = nodeRepository.findAllByResourceGroup(request.getResourceGroup());
+        log.debug("리소스 그룹 소속 노드 {}개를 조회했습니다.", nodes.size());
+
+        AcceptInfoResponseDTO response = AcceptInfoResponseDTO.fromEntity(request, portRequests, nodes);
         log.info("사용자 '{}'에 대한 AcceptInfoResponseDTO 생성을 완료했습니다.", username);
 
         return response;

src/main/java/DGU_AI_LAB/admin_be/domain/requests/service/RequestExpiryService.java

@@ -20,6 +20,7 @@ public class RequestExpiryService {
 
     private final RequestRepository requestRepository;
     private final UbuntuAccountService ubuntuAccountService;
+    private final PodService podService;
     private final ApplicationEventPublisher eventPublisher;
 
     @Transactional
@@ -33,6 +34,7 @@ public void deleteExpiredRequest(Long requestId) {
         String ubuntuUsername = request.getUbuntuUsername();
         User user = request.getUser();
 
+        podService.deletePod(request.getPodName());
         ubuntuAccountService.deleteUbuntuAccount(ubuntuUsername);
 
         request.delete();

src/main/java/DGU_AI_LAB/admin_be/domain/scheduler/RequestSchedulerService.java

@@ -14,6 +14,7 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.time.LocalDateTime;
+import java.time.ZoneId;
 import java.util.List;
 
 /**
@@ -33,7 +34,7 @@ public class RequestSchedulerService {
     @Scheduled(cron = "0 00 08 * * ?", zone = "Asia/Seoul")
     public void runScheduler() {
         log.info("🗓️ [스케줄러 시작] 만료 계정 관리 작업");
-        LocalDateTime now = LocalDateTime.now();
+        LocalDateTime now = LocalDateTime.now(ZoneId.of("Asia/Seoul"));
 
         sendPreExpiryNotification(now.plusDays(7), "7일");
         sendPreExpiryNotification(now.plusDays(3), "3일");

src/main/java/DGU_AI_LAB/admin_be/domain/scheduler/SlackNotificationWorker.java

@@ -13,6 +13,7 @@
 /**
  * Consumer / Worker
  * Redis 큐에 쌓인 알림 요청을 하나씩 꺼내서 실제로 처리하는 Consumer입니다.
+ * 전송 실패 시 최대 MAX_RETRY_COUNT회 재시도합니다.
  */
 @Component
 @RequiredArgsConstructor
@@ -24,15 +25,22 @@ public class SlackNotificationWorker {
     private final ObjectMapper objectMapper;
 
     private static final String SLACK_QUEUE_KEY = "slack:notification:queue";
+    private static final int MAX_RETRY_COUNT = 3;
 
     @Scheduled(fixedDelay = 1000)
     public void processSlackQueue() {
-        try {
-            Object messageObj = redisTemplate.opsForList().leftPop(SLACK_QUEUE_KEY);
-            if (messageObj == null) return;
+        Object messageObj = redisTemplate.opsForList().leftPop(SLACK_QUEUE_KEY);
+        if (messageObj == null) return;
 
-            SlackMessageDto dto = objectMapper.convertValue(messageObj, SlackMessageDto.class);
+        SlackMessageDto dto;
+        try {
+            dto = objectMapper.convertValue(messageObj, SlackMessageDto.class);
+        } catch (Exception e) {
+            log.error("Slack 큐 메시지 역직렬화 실패 (폐기): {}", e.getMessage());
+            return;
+        }
 
+        try {
             if (dto.getType() == SlackMessageDto.MessageType.WEBHOOK) {
                 slackApiService.sendWebhook(dto.getWebhookUrl(), dto.getMessage());
                 log.info("Slack Webhook 전송 성공 (Queue)");
@@ -43,10 +51,23 @@ public void processSlackQueue() {
             }
 
         } catch (BusinessException e) {
-            log.warn("Slack 알림 처리 실패 (Business): {}", e.getMessage());
+            // 비즈니스 예외(유저 없음 등)는 재시도해도 동일하게 실패하므로 바로 폐기
+            log.warn("Slack 알림 처리 실패 (Business, 폐기): {}", e.getMessage());
 
         } catch (Exception e) {
-            log.error("Slack 큐 처리 중 시스템 오류 (재시도 필요 시 큐 복귀 고려)", e);
+            // 일시적 장애(네트워크, Slack API 다운 등)는 재시도
+            requeue(dto, e);
+        }
+    }
+
+    private void requeue(SlackMessageDto dto, Exception cause) {
+        if (dto.getRetryCount() < MAX_RETRY_COUNT) {
+            dto.incrementRetryCount();
+            redisTemplate.opsForList().rightPush(SLACK_QUEUE_KEY, dto);
+            log.warn("Slack 메시지 재시도 예약 ({}/{}회): {}", dto.getRetryCount(), MAX_RETRY_COUNT, cause.getMessage());
+        } else {
+            log.error("Slack 메시지 최대 재시도({}) 초과, 폐기: {} | 원인: {}",
+                    MAX_RETRY_COUNT, dto.getMessage(), cause.getMessage());
         }
     }
-}
+}