Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
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
4 changes: 2 additions & 2 deletions .github/workflows/cd.yml
Original file line number Diff line number Diff line change
Expand Up @@ -76,10 +76,10 @@ jobs:
echo "DOCKER_USERNAME=${{ secrets.DOCKER_USERNAME }}" >> .env

# Docker Hub에서 방금 만든 최신 이미지 가져오기
sudo docker-compose pull
sudo docker compose pull

# 컨테이너 재실행
sudo docker-compose up -d
sudo docker compose up -d

# 불필요한 이미지 삭제
sudo docker image prune -f
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ public class AuthResDTO {
// SMS 발송 응답
@JsonInclude(JsonInclude.Include.NON_NULL)
public record SmsSendResponse(
LocalDateTime expiredAt, // 인증번호 만료 시간 (발송 시간 + 3분)
LocalDateTime expiredAt, // 인증번호 만료 시간 (발송 시간 + 5분)
String verificationCode // 개발/QA 환경에서만 포함 (프로덕션에서는 null)
) {}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,9 @@ public enum AuthErrorCode implements BaseErrorCode {
SMS_COOLDOWN(HttpStatus.TOO_MANY_REQUESTS,
"AUTH429_1",
"잠시 후 다시 시도해주세요. (1분 후 재요청 가능)"),
VERIFICATION_ATTEMPTS_EXCEEDED(HttpStatus.TOO_MANY_REQUESTS,
"AUTH429_2",
"인증 시도 횟수를 초과했습니다. 인증번호를 다시 요청해주세요."),

// 회원가입/로그인 관련
ALREADY_REGISTERED_PHONE(HttpStatus.CONFLICT,
Expand Down
55 changes: 33 additions & 22 deletions src/main/java/com/example/scoi/domain/auth/service/AuthService.java
Original file line number Diff line number Diff line change
Expand Up @@ -65,9 +65,11 @@ public class AuthService {
private static final String BLACKLIST_PREFIX = "blacklist:";
private static final String SMS_COOLDOWN_PREFIX = "sms:cooldown:";
private static final String SMS_REQUIRED_PREFIX = "sms_required:";
private static final String SMS_VERIFY_FAIL_PREFIX = "sms:verify:fail:";

// 상수
private static final int SMS_CODE_LENGTH = 6;
private static final int MAX_VERIFY_ATTEMPTS = 5;
private static final long SMS_EXPIRATION_MINUTES = 5;
private static final long VERIFICATION_EXPIRATION_MINUTES = 10;
private static final long REFRESH_TOKEN_SLIDING_DAYS = 14; // 비활성 기준 만료
Expand All @@ -85,9 +87,10 @@ public AuthResDTO.SmsSendResponse sendSms(AuthReqDTO.SmsSendRequest request) {
// 1. 인증번호 생성 (6자리)
String verificationCode = String.format("%06d", ThreadLocalRandom.current().nextInt(1000000));

// 2. Redis 저장
// 2. Redis 저장 (새 코드에는 새 시도 예산을 부여)
String redisKey = SMS_PREFIX + request.phoneNumber();
redisUtil.set(redisKey, verificationCode, SMS_EXPIRATION_MINUTES, TimeUnit.MINUTES);
redisUtil.delete(SMS_VERIFY_FAIL_PREFIX + request.phoneNumber());

// 3. CoolSMS 발송
if (smsEnabled) {
Expand All @@ -99,7 +102,7 @@ public AuthResDTO.SmsSendResponse sendSms(AuthReqDTO.SmsSendRequest request) {
);
CoolSmsDTO.SendRequest smsRequest = new CoolSmsDTO.SendRequest(message);
coolSmsClient.sendMessage(smsRequest);
log.info("SMS 발송 성공: phoneNumber={}, code={}", request.phoneNumber(), verificationCode);
log.info("SMS 발송 성공: phoneNumber={}", request.phoneNumber());
} catch (Exception e) {
log.error("SMS 발송 실패: {}", e.getMessage());
throw new AuthException(AuthErrorCode.SMS_SEND_FAILED);
Expand All @@ -120,19 +123,37 @@ public AuthResDTO.SmsSendResponse sendSms(AuthReqDTO.SmsSendRequest request) {
public AuthResDTO.SmsVerifyResponse verifySms(AuthReqDTO.SmsVerifyRequest request) {
// 1. Redis 조회
String redisKey = SMS_PREFIX + request.phoneNumber();
String failKey = SMS_VERIFY_FAIL_PREFIX + request.phoneNumber();
String storedCode = redisUtil.get(redisKey);

if (storedCode == null) {
throw new AuthException(AuthErrorCode.VERIFICATION_CODE_EXPIRED);
}

// 2. 검증
// 2. 검증 (불일치 시 실패 카운터 증가 → 한도 초과 시 코드 무효화)
if (!storedCode.equals(request.verificationCode())) {
throw new AuthException(AuthErrorCode.INVALID_VERIFICATION_CODE);
long failCount = redisUtil.increment(failKey, SMS_EXPIRATION_MINUTES, TimeUnit.MINUTES);

if (failCount >= MAX_VERIFY_ATTEMPTS) {
// 한도 초과: 코드와 카운터 삭제 → 새 코드 재요청 유도
redisUtil.delete(redisKey);
redisUtil.delete(failKey);
log.warn("SMS 인증 시도 횟수 초과: phoneNumber={}, failCount={}", request.phoneNumber(), failCount);
throw new AuthException(AuthErrorCode.VERIFICATION_ATTEMPTS_EXCEEDED);
}

long remainingAttempts = MAX_VERIFY_ATTEMPTS - failCount;
log.warn("SMS 인증번호 불일치: phoneNumber={}, failCount={}, remainingAttempts={}",
request.phoneNumber(), failCount, remainingAttempts);
throw new AuthException(
AuthErrorCode.INVALID_VERIFICATION_CODE,
Map.of("remainingAttempts", String.valueOf(remainingAttempts))
);
}

// 3. SMS 코드 삭제
// 3. SMS 코드 및 실패 카운터 삭제 (성공 시 초기화)
redisUtil.delete(redisKey);
redisUtil.delete(failKey);

// 4. Verification Token 발급
String verificationToken = jwtUtil.createVerificationToken(request.phoneNumber());
Expand All @@ -151,7 +172,8 @@ public AuthResDTO.SmsVerifyResponse verifySms(AuthReqDTO.SmsVerifyRequest reques
/**
* Verification Token 검증 (범용)
* SMS 인증 완료 후 발급된 토큰이 유효한지 검증합니다.
* 다른 도메인(회원가입, 비밀번호 찾기, 휴대폰 번호 변경 등)에서 재사용 가능합니다.
* 토큰은 소멸되지 않으며, TTL(10분) 동안 회원가입·로그인·비밀번호 재설정 등
* 여러 동작에서 반복 재사용 가능합니다. (매번 SMS 재인증을 요구하지 않기 위한 의도된 설계)
*
* @param verificationToken SMS 인증 완료 후 발급받은 토큰
* @param phoneNumber 검증할 휴대폰 번호
Expand All @@ -175,11 +197,11 @@ public String validateVerificationToken(String verificationToken, String phoneNu
}

// 간편 비밀번호 재설정
@jakarta.transaction.Transactional
@Transactional
public Void resetPassword(
AuthReqDTO.ResetPassword dto
) {
// Verification Token 검증 및 소멸 (SMS 인증 완료 확인)
// Verification Token 검증 (SMS 인증 완료 확인 / 토큰은 TTL 동안 재사용 가능)
String phoneNumber = validateVerificationToken(dto.verificationToken(), dto.phoneNumber());

// 사용자 가져오기
Expand Down Expand Up @@ -210,6 +232,9 @@ public Void resetPassword(

// 로그인 횟수 -> 0
member.resetLoginFailCount();

// RT 만료로 걸린 SMS 재인증 플래그 해제 (이미 SMS 인증을 통과했으므로 잠금 완전 해제)
redisUtil.delete(SMS_REQUIRED_PREFIX + phoneNumber);
return null;
}

Expand Down Expand Up @@ -460,18 +485,4 @@ public void logout(String phoneNumber, String accessToken) {

log.info("로그아웃 성공: phoneNumber={}", phoneNumber);
}

// 임시
public String generateSmsToken(
String phoneNumber
) {
// 4. Verification Token 발급
String verificationToken = jwtUtil.createVerificationToken(phoneNumber);

// 5. Redis 저장 (10분 TTL)
String tokenKey = VERIFICATION_PREFIX + verificationToken;
redisUtil.set(tokenKey, phoneNumber, VERIFICATION_EXPIRATION_MINUTES, TimeUnit.MINUTES);

return verificationToken;
}
}
Original file line number Diff line number Diff line change
@@ -1,27 +1,107 @@
package com.example.scoi.domain.websocket;

import com.example.scoi.domain.websocket.code.WebsocketErrorCode;
import com.example.scoi.domain.websocket.event.WebsocketConnectionEvent.ConnectedEvent;
import com.example.scoi.domain.websocket.event.WebsocketConnectionEvent.DisconnectedEvent;
import com.example.scoi.domain.websocket.exception.WebsocketException;
import com.example.scoi.domain.websocket.handler.UpbitTickerHandler;
import jakarta.annotation.PreDestroy;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import org.springframework.retry.annotation.Backoff;
import org.springframework.retry.annotation.Recover;
import org.springframework.retry.annotation.Retryable;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.web.socket.client.WebSocketClient;

import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledFuture;
import java.util.concurrent.TimeUnit;

@Service
@RequiredArgsConstructor
@Slf4j
public class WebsocketConnect {

private final WebSocketClient webSocketClient;
private final UpbitTickerHandler upbitTickerHandler;
private final ObjectProvider<WebsocketConnect> selfProvider; // AOP 우회 방지를 위한 프록시 의존성

private static final String PUBLIC_URL = "wss://api.upbit.com/websocket/v1";

// 실시간 가격 변동 체크
// 워치독(Half-Open 헬스체크) 전용 스케줄러
private final ScheduledExecutorService scheduler = Executors.newSingleThreadScheduledExecutor();
private ScheduledFuture<?> watchdogTask;
Comment on lines +37 to +39

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

🩺 Stability & Availability | 🟠 Major | ⚡ Quick win

watchdogTask 필드의 스레드 안전성 문제.

watchdogTaskvolatile이 아니며, onConnected()/onDisconnected()는 이벤트 발행 스레드(WebSocket 컨테이너 스레드 또는 @Async 스레드)에서 호출될 수 있어 서로 다른 스레드에서 동시 실행될 가능성이 있습니다. check-then-act 패턴(null 체크 → cancel → 재할당)이 원자적이지 않아, 경쟁 상태에서 워치독 취소가 누락되거나 중복 스케줄될 수 있습니다.

최소한 volatile 선언과 함께, 가능하면 synchronized 블록이나 AtomicReference로 감싸는 것을 권장합니다.

Also applies to: 76-100

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/main/java/com/example/scoi/domain/websocket/WebsocketConnect.java` around
lines 37 - 39, Make watchdogTask thread-safe across onConnected() and
onDisconnected(): declare it volatile and synchronize the
check/cancel/reschedule operations, or use an AtomicReference to perform atomic
updates. Ensure concurrent callbacks cannot leave an uncancelled task or
schedule duplicate watchdog tasks.


@EventListener(ApplicationReadyEvent.class)
public void connect(){
log.info("[ Websocket ]: 디페깅 알고리즘 구동 시작...");
webSocketClient.execute(upbitTickerHandler, PUBLIC_URL);
public void init() {
log.info("[ Websocket ]: 애플리케이션 준비 완료.");
selfProvider.getObject().connectWithRetry();
}

@Async
@Retryable(
retryFor = WebsocketException.class,
maxAttempts = 10,
backoff = @Backoff(delay = 1000, multiplier = 2, maxDelay = 60000)
)
public void connectWithRetry() {
log.info("[ Websocket ]: 업비트 웹소켓 연결 시도 중... URL: {}", PUBLIC_URL);
try {
// execute()의 리턴값인 Future를 대기(get)하여 타임아웃이나 연결 거부 시 예외를 발생시키도록 유도
webSocketClient.execute(upbitTickerHandler, PUBLIC_URL).get(5, TimeUnit.SECONDS);
} catch (Exception e) {
log.error("[ Websocket ]: 웹소켓 연결 실패: {}", e.getMessage());
// 예외를 던져서 @Retryable이 AOP 기반 백오프 및 재시도를 수행하도록 트리거
throw new WebsocketException(WebsocketErrorCode.UPBIT_WEBSOCKET_ERROR);
}
}
Comment on lines +53 to +63

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

🩺 Stability & Availability | 🟠 Major | ⚡ Quick win

타임아웃 시 진행 중인 연결 시도가 취소되지 않음.

Future.get(timeout)은 타임아웃 발생 시 대기만 중단할 뿐 내부적으로 진행 중인 연결 작업을 취소하지 않습니다. 타임아웃으로 WebsocketException이 발생해 재시도가 이루어지는 동안, 이전 execute() 호출이 백그라운드에서 나중에 성공하면 두 개 이상의 활성 웹소켓 연결이 동시에 존재할 수 있습니다. 이 경우 afterConnectionEstablished가 여러 번 호출되어 ConnectedEvent가 중복 발행되고, 티커 데이터가 중복으로 STOMP 브로드캐스트될 위험이 있습니다.

🔧 제안: 타임아웃 시 future를 명시적으로 취소
     public void connectWithRetry() {
         log.info("[ Websocket ]: 업비트 웹소켓 연결 시도 중... URL: {}", PUBLIC_URL);
+        java.util.concurrent.Future<WebSocketSession> future = null;
         try {
-            // execute()의 리턴값인 Future를 대기(get)하여 타임아웃이나 연결 거부 시 예외를 발생시키도록 유도
-            webSocketClient.execute(upbitTickerHandler, PUBLIC_URL).get(5, TimeUnit.SECONDS);
+            future = webSocketClient.execute(upbitTickerHandler, PUBLIC_URL);
+            future.get(5, TimeUnit.SECONDS);
         } catch (Exception e) {
             log.error("[ Websocket ]: 웹소켓 연결 실패: {}", e.getMessage());
+            if (future != null) {
+                future.cancel(true);
+            }
             throw new WebsocketException(WebsocketErrorCode.UPBIT_WEBSOCKET_ERROR);
         }
     }
📝 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
public void connectWithRetry() {
log.info("[ Websocket ]: 업비트 웹소켓 연결 시도 중... URL: {}", PUBLIC_URL);
try {
// execute()의 리턴값인 Future를 대기(get)하여 타임아웃이나 연결 거부 시 예외를 발생시키도록 유도
webSocketClient.execute(upbitTickerHandler, PUBLIC_URL).get(5, TimeUnit.SECONDS);
} catch (Exception e) {
log.error("[ Websocket ]: 웹소켓 연결 실패: {}", e.getMessage());
// 예외를 던져서 @Retryable이 AOP 기반 백오프 및 재시도를 수행하도록 트리거
throw new WebsocketException(WebsocketErrorCode.UPBIT_WEBSOCKET_ERROR);
}
}
public void connectWithRetry() {
log.info("[ Websocket ]: 업비트 웹소켓 연결 시도 중... URL: {}", PUBLIC_URL);
java.util.concurrent.Future<WebSocketSession> future = null;
try {
future = webSocketClient.execute(upbitTickerHandler, PUBLIC_URL);
future.get(5, TimeUnit.SECONDS);
} catch (Exception e) {
log.error("[ Websocket ]: 웹소켓 연결 실패: {}", e.getMessage());
if (future != null) {
future.cancel(true);
}
// 예외를 던져서 `@Retryable이` AOP 기반 백오프 및 재시도를 수행하도록 트리거
throw new WebsocketException(WebsocketErrorCode.UPBIT_WEBSOCKET_ERROR);
}
}
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@src/main/java/com/example/scoi/domain/websocket/WebsocketConnect.java` around
lines 53 - 63, connectWithRetry에서 webSocketClient.execute(...)가 반환하는 Future를 변수로
보관하고, get 타임아웃 발생 시 해당 Future를 명시적으로 cancel(true)하여 진행 중인 연결 작업을 중단하세요. 취소 후 기존
WebsocketException을 발생시켜 `@Retryable` 재시도를 유지하고, 이미 완료된 경우에는 불필요한 취소가 발생하지 않도록
처리하세요.


@Recover
public void recover(Exception e) {
log.error("[ Websocket ]: 웹소켓 재연결 최대 시도 횟수(10회) 초과");

// 2시간뒤 다시 시도
scheduler.schedule(() -> {
log.info("[ Websocket ]: 시스템 복구를 위해 웹소켓 재연결 사이클을 다시 시작합니다.");
selfProvider.getObject().connectWithRetry();
}, 2, TimeUnit.HOURS);
}

@EventListener
public void onConnected(ConnectedEvent event) {
log.info("[ Websocket ]: 이벤트 수신 - 웹소켓 연결 활성화 완료.");

if (watchdogTask != null && !watchdogTask.isCancelled()) {
watchdogTask.cancel(false);
}
// 10초마다 마지막 수신 시간 검사
watchdogTask = scheduler.scheduleAtFixedRate(() -> {
long idleTime = System.currentTimeMillis() - upbitTickerHandler.getLastMessageTime();
if (idleTime > 30_000) { // 30초 초과 시
log.error("[ Websocket ]: 30초간 업비트 응답이 없습니다...");
upbitTickerHandler.closeSession(); // DisconnectedEvent 유발
}
}, 10, 10, TimeUnit.SECONDS);
}

@EventListener
public void onDisconnected(DisconnectedEvent event) {
log.warn("[ Websocket ]: 이벤트 수신 - 웹소켓 연결 해제. 재연결 시도");
if (watchdogTask != null) {
watchdogTask.cancel(false);
}
selfProvider.getObject().connectWithRetry();
}

@PreDestroy
public void shutdown() {
log.info("[ Websocket ]: 워치독 스케줄러 종료 중...");
scheduler.shutdown();
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
package com.example.scoi.domain.websocket.code;

import com.example.scoi.global.apiPayload.code.BaseErrorCode;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import org.springframework.http.HttpStatus;

@Getter
@RequiredArgsConstructor
public enum WebsocketErrorCode implements BaseErrorCode {

UPBIT_WEBSOCKET_ERROR(HttpStatus.INTERNAL_SERVER_ERROR,
"WEBSOCKET500_1",
"웹소켓 연결 도중 에러가 발생했습니다."),
;

private final HttpStatus status;
private final String code;
private final String message;
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
package com.example.scoi.domain.websocket.event;

public interface WebsocketConnectionEvent {
record ConnectedEvent() implements WebsocketConnectionEvent {}
record DisconnectedEvent() implements WebsocketConnectionEvent {}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
package com.example.scoi.domain.websocket.exception;

import com.example.scoi.global.apiPayload.code.BaseErrorCode;
import com.example.scoi.global.apiPayload.exception.ScoiException;

public class WebsocketException extends ScoiException {
public WebsocketException(BaseErrorCode code) {
super(code);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -2,14 +2,19 @@

import com.example.scoi.domain.websocket.converter.WebSocketConverter;
import com.example.scoi.domain.websocket.dto.UpbitResDTO;
import com.example.scoi.domain.websocket.event.WebsocketConnectionEvent.ConnectedEvent;
import com.example.scoi.domain.websocket.event.WebsocketConnectionEvent.DisconnectedEvent;
import com.example.scoi.domain.websocket.service.WebSocketService;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.PropertyNamingStrategies;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.messaging.simp.SimpMessageSendingOperations;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import org.springframework.web.socket.BinaryMessage;
import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.WebSocketSession;
import org.springframework.web.socket.handler.BinaryWebSocketHandler;

Expand All @@ -19,18 +24,27 @@

@Component
@RequiredArgsConstructor
@Slf4j
public class UpbitTickerHandler extends BinaryWebSocketHandler {

private final SimpMessageSendingOperations simpMessageSendingOperations;
private final WebSocketService webSocketService;
private final ApplicationEventPublisher eventPublisher;

private static final ObjectMapper objectMapper = new ObjectMapper()
.setPropertyNamingStrategy(PropertyNamingStrategies.LOWER_CAMEL_CASE);

private volatile long lastMessageTime;
private volatile WebSocketSession currentSession;

@Override
public void afterConnectionEstablished(
WebSocketSession session
) throws IOException {
this.currentSession = session;
this.lastMessageTime = System.currentTimeMillis();
log.info("[ Websocket ]: 업비트 웹소켓 연결 성공.");
eventPublisher.publishEvent(new ConnectedEvent());
session.sendMessage(WebSocketConverter.toGetCoinPrice(List.of("KRW-USDT","KRW-USDC")));
}

Expand All @@ -39,6 +53,8 @@ public void handleBinaryMessage(
WebSocketSession session,
BinaryMessage message
) throws IOException {
this.lastMessageTime = System.currentTimeMillis(); // Heartbeat 갱신

String converted = new String(message.getPayload().array(), StandardCharsets.UTF_8);
converted = converted.replace("[","").replace("]","");
publish(converted);
Expand All @@ -50,4 +66,31 @@ public void handleBinaryMessage(
protected void publish(String message) {
simpMessageSendingOperations.convertAndSend("/topic/ticker", message);
}

@Override
public void afterConnectionClosed(WebSocketSession session, CloseStatus status) throws Exception {
log.warn("[ Websocket ]: 업비트 웹소켓 연결이 종료되었습니다. CloseStatus: {}", status);
eventPublisher.publishEvent(new DisconnectedEvent());
}

@Override
public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {
log.error("[ Websocket ]: 웹소켓 에러 발생: {}", exception.getMessage());
eventPublisher.publishEvent(new DisconnectedEvent());
}
Comment thread
rlawngjs0313 marked this conversation as resolved.

public long getLastMessageTime() {
return lastMessageTime;
}

public void closeSession() {
if (currentSession != null && currentSession.isOpen()) {
try {
log.warn("[ Websocket ]: 하프 오픈(Half-open) 상태 감지.");
currentSession.close();
} catch (IOException e) {
log.error("[ Websocket ]: 세션 강제 종료 중 에러 발생: {}", e.getMessage());
}
}
}
}
Loading
Loading