Skip to content

Commit

Permalink
✨ feat : 모든 api 호출에 대한 로깅 추가, 토큰화 금액 추가
Browse files Browse the repository at this point in the history
  • Loading branch information
@na-rong
na-rong committed Dec 5, 2024
1 parent a240381 commit 1962b5a
Showing 1 changed file with 120 additions and 89 deletions.
209 changes: 120 additions & 89 deletions src/main/java/com/picktartup/wallet/aop/WalletLoggingAspect.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,16 +3,24 @@
import com.picktartup.wallet.dto.PaymentDto;
import com.picktartup.wallet.dto.TransactionDto;
import com.picktartup.wallet.dto.WalletDto;
import com.picktartup.wallet.entity.TokenTransaction;
import com.picktartup.wallet.utils.TokenUtils;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.ResponseEntity;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import jakarta.servlet.http.HttpServletRequest;
import java.time.LocalDateTime;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

@Slf4j
@Aspect
Expand All @@ -25,158 +33,181 @@ public class WalletLoggingAspect {
/**
* 공통 로그 기록 메서드
*/
private void logBusinessAction(String action, Map<String, Object> additionalFields) {
Map<String, Object> logData = new HashMap<>();
logData.put("log_type", "business");
logData.put("action", action);
logData.put("timestamp", LocalDateTime.now().toString());
logData.put("service_name", serviceName); // 서비스 이름 추가
logData.putAll(additionalFields);

log.info("{}", logData); // JSON 형태로 기록
private void logApiCall(Map<String, Object> logData, boolean success, int httpStatus, long responseTime, String errorMessage) {
logData.put("status", success ? "success" : "failed");
logData.put("http_status", httpStatus);
logData.put("response_time_ms", responseTime);
if (errorMessage != null) {
logData.put("error_message", errorMessage);
}
log.info("{}", logData);
}

/**
* 결제 웹훅 모니터링
* 공통 API 호출 로그
*/
@Around("execution(* com.picktartup.wallet.service.TokenService.mintTokenFromPayment(..))")
public Object monitorPayment(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
PaymentDto.CompletedEvent payment = (PaymentDto.CompletedEvent) args[0];
@Around("execution(* com.picktartup.wallet.controller.*.*(..))")
public Object logAllApiCalls(ProceedingJoinPoint joinPoint) throws Throwable {
long startTime = System.currentTimeMillis();
HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()).getRequest();

Map<String, Object> logData = new HashMap<>();
logData.put("http_method", request.getMethod());
logData.put("uri", request.getRequestURI());
logData.put("api_path", request.getRequestURI().replaceAll("/\\d+", "/{id}"));
logData.put("api_name", ((MethodSignature) joinPoint.getSignature()).getMethod().getName());
logData.put("client_ip", request.getRemoteAddr());
logData.put("request_id", UUID.randomUUID().toString());
logData.put("timestamp", LocalDateTime.now().toString());
logData.put("service_name", serviceName);

try {
Object result = joinPoint.proceed();
long responseTime = System.currentTimeMillis() - startTime;

logBusinessAction("payment_completed", Map.of(
"amount", payment.getAmount(),
"transaction_id", payment.getTransactionId(),
"status", "success"
));
// 성공 로그
int httpStatus = (result instanceof ResponseEntity) ?
((ResponseEntity<?>) result).getStatusCodeValue() : 200;
logApiCall(logData, true, httpStatus, responseTime, null);

return result;
} catch (Exception e) {
logBusinessAction("payment_failed", Map.of(
"amount", payment.getAmount(),
"transaction_id", payment.getTransactionId(),
"status", "failed",
"error_message", e.getMessage()
));
long responseTime = System.currentTimeMillis() - startTime;

// 실패 로그
logApiCall(logData, false, 500, responseTime, e.getMessage());
throw e;
}
}

/**
* 지갑 잔고 조회 모니터링
* 결제 웹훅 로직 (토큰화 금액 포함)
*/
@Around("execution(* com.picktartup.wallet.service.WalletService.getWalletBalanceByUserId(..))")
public Object monitorBalanceCheck(ProceedingJoinPoint joinPoint) throws Throwable {
@Around("execution(* com.picktartup.wallet.service.TokenService.mintTokenFromPayment(..))")
public Object monitorPayment(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
Long userId = (Long) args[0];
PaymentDto.CompletedEvent payment = (PaymentDto.CompletedEvent) args[0];

Map<String, Object> logData = new HashMap<>();
logData.put("log_type", "business");
logData.put("action", "payment_completed");
logData.put("amount", payment.getAmount()); // 결제 금액
logData.put("transaction_id", payment.getTransactionId()); // 주문번호
logData.put("service_name", serviceName);
logData.put("timestamp", LocalDateTime.now().toString());

// 서비스에서 전달된 토큰화 금액 사용
try {
Object result = joinPoint.proceed();

logBusinessAction("check_wallet_balance", Map.of(
"user_id", userId,
"status", "success"
));

// 성공 로그
if (result instanceof TokenTransaction) {
TokenTransaction transaction = (TokenTransaction) result;
logData.put("tokenized_amount", transaction.getTokenAmount()); // 토큰화 금액
logData.put("tokenized_amount_wei", TokenUtils.toWei(transaction.getTokenAmount().doubleValue())); // Wei 변환
}
logApiCall(logData, true, 200, 0, null); // 성공 시 HTTP 200
return result;
} catch (Exception e) {
logBusinessAction("check_wallet_balance_failed", Map.of(
"user_id", userId,
"status", "failed",
"error_message", e.getMessage()
));
logApiCall(logData, false, 500, 0, e.getMessage()); // 실패 시 HTTP 500
throw e;
}

}


/**
* 송금 패턴 모니터링
* 잔고 조회 로직
*/
@Around("execution(* com.picktartup.wallet.service.TokenService.transferToAdmin(..))")
public Object monitorTransfer(ProceedingJoinPoint joinPoint) throws Throwable {
@Around("execution(* com.picktartup.wallet.service.WalletService.getWalletBalanceByUserId(..))")
public Object monitorBalanceCheck(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
TransactionDto.Request request = (TransactionDto.Request) args[0];
Long userId = (Long) args[0];
long startTime = System.currentTimeMillis();

Map<String, Object> logData = new HashMap<>();
logData.put("log_type", "business");
logData.put("action", "check_wallet_balance");
logData.put("user_id", userId);
logData.put("timestamp", LocalDateTime.now().toString());
logData.put("service_name", serviceName);

try {
Object result = joinPoint.proceed();
long responseTime = System.currentTimeMillis() - startTime;

logBusinessAction("transfer_to_admin", Map.of(
"user_id", request.getUserId(),
"amount", request.getAmount(),
"status", "success"
));

// 성공 로그
logApiCall(logData, true, 200, responseTime, null);
return result;
} catch (Exception e) {
logBusinessAction("transfer_to_admin_failed", Map.of(
"user_id", request.getUserId(),
"amount", request.getAmount(),
"status", "failed",
"error_message", e.getMessage()
));
long responseTime = System.currentTimeMillis() - startTime;

// 실패 로그
logApiCall(logData, false, 500, responseTime, e.getMessage());
throw e;
}
}

/**
* 지갑 상태 변경 모니터링
* 잔고 업데이트 로직
*/
@Around("execution(* com.picktartup.wallet.service.WalletService.updateWalletStatus(..))")
public Object monitorWalletStatus(ProceedingJoinPoint joinPoint) throws Throwable {
@Around("execution(* com.picktartup.wallet.service.WalletService.updateBalance(..))")
public Object monitorBalanceUpdate(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
Long walletId = (Long) args[0];
WalletDto.UpdateStatus.Request request = (WalletDto.UpdateStatus.Request) args[1];
Long userId = (Long) args[0];
long startTime = System.currentTimeMillis();

Map<String, Object> logData = new HashMap<>();
logData.put("log_type", "business");
logData.put("action", "update_wallet_balance");
logData.put("user_id", userId);
logData.put("timestamp", LocalDateTime.now().toString());
logData.put("service_name", serviceName);

try {
Object result = joinPoint.proceed();
long responseTime = System.currentTimeMillis() - startTime;

logBusinessAction("update_wallet_status", Map.of(
"wallet_id", walletId,
"new_status", request.getStatus(),
"status", "success"
));

// 성공 로그
logApiCall(logData, true, 200, responseTime, null);
return result;
} catch (Exception e) {
logBusinessAction("update_wallet_status_failed", Map.of(
"wallet_id", walletId,
"new_status", request.getStatus(),
"status", "failed",
"error_message", e.getMessage()
));
long responseTime = System.currentTimeMillis() - startTime;

// 실패 로그
logApiCall(logData, false, 500, responseTime, e.getMessage());
throw e;
}
}

/**
* 잔고 업데이트 모니터링
*/
@Around("execution(* com.picktartup.wallet.service.WalletService.updateBalance(..))")
public Object monitorBalanceUpdate(ProceedingJoinPoint joinPoint) throws Throwable {
@Around("execution(* com.picktartup.wallet.service.TokenService.transferToAdmin(..))")
public Object monitorTokenRefund(ProceedingJoinPoint joinPoint) throws Throwable {
Object[] args = joinPoint.getArgs();
Long userId = (Long) args[0];
TransactionDto.Request request = (TransactionDto.Request) args[0];

Map<String, Object> logData = new HashMap<>();
logData.put("log_type", "business");
logData.put("action", "token_refund");
logData.put("user_id", request.getUserId());
logData.put("transaction_id", request.getTransactionId());
logData.put("requested_token_amount", request.getAmount());
logData.put("timestamp", LocalDateTime.now().toString());
logData.put("service_name", serviceName);

try {
Object result = joinPoint.proceed();

logBusinessAction("update_wallet_balance", Map.of(
"user_id", userId,
"status", "success"
));
if (result instanceof TransactionDto.Response) {
TransactionDto.Response response = (TransactionDto.Response) result;
logData.put("status", "success");
logData.put("refunded_cash_amount", response.getAmount());
logData.put("transaction_hash", response.getTransactionHash());
}

logApiCall(logData, true, 200, 0, null);
return result;
} catch (Exception e) {
logBusinessAction("update_wallet_balance_failed", Map.of(
"user_id", userId,
"status", "failed",
"error_message", e.getMessage()
));
logApiCall(logData, false, 500, 0, e.getMessage());
throw e;
}
}

}

0 comments on commit 1962b5a

Please sign in to comment.