feat: 결제 구현 (#97)

* feat: 결제 관련 서비스, 컨트롤러, dto 추가

* feat: 결제 로직 수정

* feat: 컨트롤러 로직 수정

* feat: 취소 기능 적용

Author: ense333

src/main/java/com/threestar/trainus/domain/coupon/user/entity/UserCoupon.java

@@ -4,6 +4,8 @@
 
 import com.threestar.trainus.domain.user.entity.User;
 import com.threestar.trainus.global.entity.BaseDateEntity;
+import com.threestar.trainus.global.exception.domain.ErrorCode;
+import com.threestar.trainus.global.exception.handler.BusinessException;
 
 import jakarta.persistence.Entity;
 import jakarta.persistence.EnumType;
@@ -15,6 +17,7 @@
 import jakarta.persistence.ManyToOne;
 import jakarta.persistence.Table;
 import jakarta.persistence.UniqueConstraint;
+import jakarta.persistence.Version;
 import lombok.AccessLevel;
 import lombok.AllArgsConstructor;
 import lombok.Builder;
@@ -53,10 +56,29 @@ public class UserCoupon extends BaseDateEntity {
 
 	private LocalDateTime expirationDate;
 
+	@Version
+	private Long version;
+
 	public UserCoupon(User user, Coupon coupon, LocalDateTime expirationDate) {
 		this.user = user;
 		this.coupon = coupon;
 		this.expirationDate = expirationDate;
 		this.status = CouponStatus.ACTIVE;
 	}
+
+	public void use() {
+		if (this.status != CouponStatus.ACTIVE) {
+			throw new BusinessException(ErrorCode.INVALID_COUPON);
+		}
+		this.status = CouponStatus.INACTIVE;
+		this.useDate = LocalDateTime.now();
+	}
+
+	public void restore() {
+		if (this.status == CouponStatus.ACTIVE) {
+			throw new BusinessException(ErrorCode.INVALID_COUPON_RESTORE);
+		}
+		this.status = CouponStatus.ACTIVE;
+		this.useDate = null;
+	}
 }

src/main/java/com/threestar/trainus/domain/payment/controller/PaymentController.java

@@ -0,0 +1,78 @@
+package com.threestar.trainus.domain.payment.controller;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import com.threestar.trainus.domain.payment.dto.CancelPaymentRequest;
+import com.threestar.trainus.domain.payment.dto.ConfirmPaymentRequest;
+import com.threestar.trainus.domain.payment.dto.PaymentClient;
+import com.threestar.trainus.domain.payment.dto.PaymentRequestDto;
+import com.threestar.trainus.domain.payment.dto.PaymentResponseDto;
+import com.threestar.trainus.domain.payment.dto.SaveAmountRequest;
+import com.threestar.trainus.domain.payment.dto.TossPaymentResponseDto;
+import com.threestar.trainus.domain.payment.service.PaymentService;
+import com.threestar.trainus.global.unit.BaseResponse;
+
+import jakarta.servlet.http.HttpSession;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+@RestController
+@RequestMapping("/api/v1/payments")
+@RequiredArgsConstructor
+public class PaymentController {
+
+	private final PaymentClient tossPaymentClient;
+	private final PaymentService paymentService;
+
+	@PostMapping("/prepare")
+	public ResponseEntity<BaseResponse<PaymentResponseDto>> preparePayment(HttpSession session,
+		@RequestBody PaymentRequestDto request) {
+		Long userId = (Long)session.getAttribute("LOGIN_USER");
+		PaymentResponseDto response = paymentService.preparePayment(request, userId);
+		//여기에서 쿠폰 가격 정해서 답변 내보내고 이를 통해 아래 호출
+		return BaseResponse.ok("결제 정보 준비 완료", response, HttpStatus.OK);
+	}
+
+	@PostMapping("/saveAmount")
+	public ResponseEntity<BaseResponse<Void>> saveAmount(HttpSession session, @RequestBody SaveAmountRequest request) {
+		session.setAttribute(request.getOrderId(), request.getAmount());
+		return BaseResponse.ok("Payment temp save Successful", null, HttpStatus.OK);
+	}
+
+	@PostMapping("/verifyAmount")
+	public ResponseEntity<BaseResponse<Void>> verifyAmount(HttpSession session,
+		@RequestBody SaveAmountRequest request) {
+		Integer amount = (Integer)session.getAttribute(request.getOrderId());
+		log.info("amount = {}", amount);
+
+		try {
+			if (amount == null || !amount.equals(request.getAmount())) {
+				return BaseResponse.error("결제 금액 정보가 유효하지 않습니다", null, HttpStatus.BAD_REQUEST);
+			}
+			return BaseResponse.ok("Payment is valid", null, HttpStatus.OK);
+		} finally {
+			session.removeAttribute(request.getOrderId());
+		}
+	}
+
+	@PostMapping("/confirm")
+	public ResponseEntity<BaseResponse<Void>> confirm(@RequestBody ConfirmPaymentRequest request) {
+		TossPaymentResponseDto tossResponse = tossPaymentClient.confirmPayment(request);
+		paymentService.processConfirm(tossResponse);
+		return BaseResponse.ok("결제 성공", null, HttpStatus.OK);
+	}
+
+	@PostMapping("/cancel")
+	public ResponseEntity<BaseResponse<Void>> cancel(@RequestBody CancelPaymentRequest request) {
+		TossPaymentResponseDto tossResponse = tossPaymentClient.cancelPayment(request);
+		paymentService.processCancel(tossResponse);
+		return BaseResponse.ok("결제 취소 성공", null, HttpStatus.OK);
+	}
+}

src/main/java/com/threestar/trainus/domain/payment/dto/CancelPaymentRequest.java

@@ -0,0 +1,9 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import lombok.Data;
+
+@Data
+public class CancelPaymentRequest {
+	private String paymentKey;
+	private String cancelReason;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/ConfirmPaymentRequest.java

@@ -0,0 +1,10 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import lombok.Data;
+
+@Data
+public class ConfirmPaymentRequest {
+	private int amount;
+	private String orderId;
+	private String paymentKey;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/PaymentClient.java

@@ -0,0 +1,71 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+
+import org.apache.http.HttpHeaders;
+import org.springframework.http.HttpStatusCode;
+import org.springframework.http.MediaType;
+import org.springframework.stereotype.Component;
+import org.springframework.web.client.RestClient;
+
+import com.threestar.trainus.global.exception.domain.ErrorCode;
+import com.threestar.trainus.global.exception.handler.BusinessException;
+
+import lombok.extern.slf4j.Slf4j;
+
+@Slf4j
+@Component
+public class PaymentClient {
+	private static final String BASIC_DELIMITER = ":";
+	private static final String AUTH_HEADER_PREFIX = "Basic ";
+
+	private final PaymentProperties paymentProperties;
+	private final RestClient restClient;
+
+	public PaymentClient(PaymentProperties paymentProperties) {
+		this.paymentProperties = paymentProperties;
+		this.restClient = RestClient.builder()
+			.baseUrl(paymentProperties.getBaseUrl())
+			.defaultHeader(HttpHeaders.AUTHORIZATION, createPaymentAuthHeader(paymentProperties))
+			.build();
+	}
+
+	private String createPaymentAuthHeader(PaymentProperties paymentProperties) {
+		log.info("secret-key : {}, url ={} baseUrl = {}, baseUrl2 = {}", paymentProperties.getSecretKey(),
+			paymentProperties.getBaseUrl(), paymentProperties.getConfirmEndPoint(),
+			paymentProperties.getCancelEndPoint());
+		byte[] encodedBytes = Base64.getEncoder()
+			.encode((paymentProperties.getSecretKey() + BASIC_DELIMITER).getBytes(StandardCharsets.UTF_8));
+		log.info("new Header = {}", AUTH_HEADER_PREFIX + new String(encodedBytes));
+		return AUTH_HEADER_PREFIX + new String(encodedBytes);
+	}
+
+	public TossPaymentResponseDto confirmPayment(ConfirmPaymentRequest request) {
+		log.info("Sending Toss confirm request: orderId={}, amount={}, paymentKey={}",
+			request.getOrderId(), request.getAmount(), request.getPaymentKey());
+
+		return restClient.post()
+			.uri(paymentProperties.getConfirmEndPoint())
+			.contentType(MediaType.APPLICATION_JSON)
+			.body(request)
+			.retrieve()
+			.onStatus(HttpStatusCode::isError, (req, res) -> {
+				throw new BusinessException(ErrorCode.CONFIRM_PAYMENT_FAILED);
+			})
+			.body(TossPaymentResponseDto.class);
+	}
+
+	public TossPaymentResponseDto cancelPayment(CancelPaymentRequest request) {
+		return restClient.post()
+			.uri(String.format(paymentProperties.getCancelEndPoint(), request.getPaymentKey()))
+			.contentType(MediaType.APPLICATION_JSON)
+			.body(request)
+			.retrieve()
+			.onStatus(HttpStatusCode::isError, (req, res) -> {
+				throw new BusinessException(ErrorCode.CANCEL_PAYMENT_FAILED);
+			})
+			.body(TossPaymentResponseDto.class);
+	}
+
+}

src/main/java/com/threestar/trainus/domain/payment/dto/PaymentProperties.java

@@ -0,0 +1,19 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.stereotype.Component;
+
+import lombok.Getter;
+import lombok.Setter;
+
+@Getter
+@Setter
+@Component
+@ConfigurationProperties(prefix = "payment")
+public class PaymentProperties {
+
+	private String secretKey;
+	private String baseUrl;
+	private String confirmEndPoint;
+	private String cancelEndPoint;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/PaymentRequestDto.java

@@ -0,0 +1,12 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import com.threestar.trainus.domain.payment.entity.PaymentMethod;
+
+import lombok.Data;
+
+@Data
+public class PaymentRequestDto {
+	private Long lessonId;
+	private Long userCouponId;
+	private PaymentMethod paymentMethod;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/PaymentResponseDto.java

@@ -0,0 +1,19 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import java.time.LocalDateTime;
+
+import com.threestar.trainus.domain.payment.entity.PaymentMethod;
+
+import lombok.Builder;
+import lombok.Data;
+
+@Builder
+@Data
+public class PaymentResponseDto {
+	private String orderId;
+	private String lessonTitle;
+	private Integer originPrice;
+	private Integer payPrice;
+	private PaymentMethod paymentMethod;
+	private LocalDateTime expiredAt;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/SaveAmountRequest.java

@@ -0,0 +1,9 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import lombok.Data;
+
+@Data
+public class SaveAmountRequest {
+	private String orderId;
+	private Integer amount;
+}

src/main/java/com/threestar/trainus/domain/payment/dto/TossPaymentResponseDto.java

@@ -0,0 +1,19 @@
+package com.threestar.trainus.domain.payment.dto;
+
+import java.time.LocalDateTime;
+
+import com.threestar.trainus.domain.payment.entity.PaymentStatus;
+
+import lombok.Getter;
+
+@Getter
+public class TossPaymentResponseDto {
+	private String paymentKey;
+	private String orderId;
+	private String orderName;
+	private String status;
+	private String requestedAt;
+	private String approvedAt;
+	private int totalAmount;
+	private String method;
+}