부산대 BE_정지민 4주차 과제(3단계)

src/main/java/gift/payment/application/PaymentService.java

@@ -0,0 +1,44 @@
+package gift.payment.application;
+
+import gift.product.domain.Product;
+import gift.product.domain.ProductOption;
+import gift.product.domain.WishList;
+import gift.product.domain.WishListProduct;
+import gift.product.exception.ProductException;
+import gift.product.infra.ProductRepository;
+import gift.product.infra.WishListRepository;
+import gift.util.ErrorCode;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.Objects;
+
+@Service
+public class PaymentService {
+
+    private final WishListRepository wishListRepository;
+    private final ProductRepository productRepository;
+
+    public PaymentService(WishListRepository wishListRepository, ProductRepository productRepository) {
+        this.wishListRepository = wishListRepository;
+        this.productRepository = productRepository;
+    }
+
+    @Transactional
+    public void processPayment(Long userId, Long wishListId) {
+        WishList wishList = wishListRepository.findById(wishListId);
+
+        if (!Objects.equals(wishList.getUser().getId(), userId)) {
+            throw new ProductException(ErrorCode.NOT_USER_OWNED);
+        }
+
+        for (WishListProduct product : wishList.getWishListProducts()) {
+            Product wishListProduct = product.getProduct();
+            ProductOption productOption = product.getProductOption();
+            productOption.decreaseQuantity(product.getQuantity());
+
+            productRepository.save(wishListProduct);
+        }
+        wishListRepository.delete(wishList);
+    }
+}

src/main/java/gift/payment/controller/PaymentController.java

@@ -0,0 +1,33 @@
+package gift.payment.presentation;
+
+import gift.payment.application.PaymentService;
+import gift.util.CommonResponse;
+import gift.util.annotation.JwtAuthenticated;
+import org.springframework.http.ResponseEntity;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.web.bind.annotation.*;
+
+@RestController
+@RequestMapping("/payment")
+public class PaymentController {
+
+    private final PaymentService paymentService;
+
+    public PaymentController(PaymentService paymentService) {
+        this.paymentService = paymentService;
+    }
+
+    @PostMapping("/process/{wishListId}")
+    @JwtAuthenticated
+    public ResponseEntity<?> processPayment(@PathVariable Long wishListId) {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+        Long userId = Long.valueOf(authentication.getName());
+
+        paymentService.processPayment(userId, wishListId);
+        return ResponseEntity.ok(new CommonResponse<>(
+                null, "결제 완료", true
+        ));
+    }
+
+}

src/main/java/gift/product/application/CategoryService.java

@@ -25,7 +25,7 @@ public void addCategory(CreateCategoryRequest request) {
         if (categoryRepository.findByName(request.getName()) != null) {
             throw new IllegalArgumentException("이미 존재하는 카테고리입니다.");
         }
-        Category category = new Category(request.getName(), request.getDescription(), request.getImageUrl(), request.getColor());
+        Category category = new Category(request);
 
         categoryRepository.save(category);
     }

src/main/java/gift/product/application/ProductService.java

@@ -10,7 +10,6 @@
 import java.util.List;
 
 @Service
-@Transactional(readOnly = true)
 public class ProductService {
 
     private final ProductRepository productRepository;
@@ -28,8 +27,7 @@ public ProductService(ProductRepository productRepository, CategoryService categ
     @Transactional
     public Long saveProduct(CreateProductRequestDTO createProductRequestDTO) {
         Category category = categoryService.getCategoryByName(createProductRequestDTO.getCategory());
-        Product product = new Product(createProductRequestDTO.getName(), createProductRequestDTO.getPrice(),
-                createProductRequestDTO.getImageUrl(), category);
+        Product product = new Product(createProductRequestDTO, category);
         validateProduct(product);
 
         return productRepository.save(product).getId();
@@ -39,14 +37,8 @@ public Long saveProduct(CreateProductRequestDTO createProductRequestDTO) {
     public void addProductOption(Long id, CreateProductOptionRequestDTO createProductOptionRequestDTO) {
         Product product = productRepository.findById(id);
 
-        productRepository.findProductOptionsByProductId(id).forEach(productOption -> {
-            if (productOption.getName().equals(createProductOptionRequestDTO.getName())) {
-                throw new ProductException(ErrorCode.DUPLICATED_OPTION_NAME);
-            }
-        });
+        product.addProductOption(createProductOptionRequestDTO);
 
-        ProductOption productOption = productRepository.saveProductOption(new ProductOption(createProductOptionRequestDTO.getName(),
-                createProductOptionRequestDTO.getQuantity(), product));
         productRepository.save(product);
     }
 

src/main/java/gift/product/application/WishListService.java

@@ -1,9 +1,6 @@
 package gift.product.application;
 
-import gift.product.domain.Product;
-import gift.product.domain.ProductOption;
-import gift.product.domain.WishList;
-import gift.product.domain.WishListProduct;
+import gift.product.domain.*;
 import gift.product.exception.ProductException;
 import gift.product.infra.ProductRepository;
 import gift.product.infra.WishListRepository;
@@ -51,23 +48,23 @@ public Page<WishList> getProductsInWishList(Long userId, int page, int size, Str
     }
 
     @Transactional
-    public void addProductToWishList(Long userId, Long wishlistId, Long productId, Long optionId) {
-        WishList wishList = findById(wishlistId);
-        if (!Objects.equals(wishList.getUser().getId(), userId)) {
+    public void addProductToWishList(AddWishListRequest request) {
+        WishList wishList = findById(request.getWishlistId());
+        if (!Objects.equals(wishList.getUser().getId(), request.getUserId())) {
             throw new ProductException(ErrorCode.NOT_USER_OWNED);
         }
 
-        Product product = productRepository.findById(productId);
-        ProductOption productOption = productRepository.getProductWithOption(productId, optionId);
+        Product product = productRepository.findById(request.getProductId());
+        ProductOption productOption = productRepository.getProductWithOption(request.getProductId(), request.getOptionId());
 
-        wishList.addWishListProduct(new WishListProduct(wishList, product, productOption));
+        wishList.addWishListProduct(new WishListProduct(wishList, product, productOption, request.getQuantity()));
 
 
         wishListRepository.save(wishList);
     }
 
     public WishList findById(Long id) {
-        return wishListRepository.findById(id).orElseThrow(() -> new ProductException(ErrorCode.WISHLIST_NOT_FOUND));
+        return wishListRepository.findById(id);
     }
 
     @Transactional

src/main/java/gift/product/domain/AddWishListRequest.java

@@ -0,0 +1,37 @@
+package gift.product.domain;
+
+public class AddWishListRequest {
+    private Long userId;
+    private Long wishlistId;
+    private Long productId;
+    private Long optionId;
+    private Long quantity;
+
+    public Long getUserId() {
+        return userId;
+    }
+
+    public Long getWishlistId() {
+        return wishlistId;
+    }
+
+    public Long getProductId() {
+        return productId;
+    }
+
+    public Long getOptionId() {
+        return optionId;
+    }
+
+    public Long getQuantity() {
+        return quantity;
+    }
+
+    public AddWishListRequest(Long userId, Long wishlistId, Long productId, Long optionId, Long quantity) {
+        this.userId = userId;
+        this.wishlistId = wishlistId;
+        this.productId = productId;
+        this.optionId = optionId;
+        this.quantity = quantity;
+    }
+}

src/main/java/gift/product/domain/Category.java

@@ -29,11 +29,11 @@ public class Category {
     public Category() {
     }
 
-    public Category(String name, String description, String imageUrl, String color) {
-        this.name = name;
-        this.description = description;
-        this.imageUrl = imageUrl;
-        this.color = color;
+    public Category(CreateCategoryRequest request) {
+        this.name = request.getName();
+        this.description = request.getDescription();
+        this.imageUrl = request.getImageUrl();
+        this.color = request.getColor();
     }
 
     public Long getId() {

src/main/java/gift/product/domain/Product.java

@@ -32,10 +32,10 @@ public class Product {
     public Product() {
     }
 
-    public Product(String name, Double price, String imageUrl, Category category) {
-        this.name = name;
-        this.price = price;
-        this.imageUrl = imageUrl;
+    public Product(CreateProductRequestDTO createProductRequestDTO, Category category) {
+        this.name = createProductRequestDTO.getName();
+        this.price = createProductRequestDTO.getPrice();
+        this.imageUrl = createProductRequestDTO.getImageUrl();
         this.category = category;
     }
 
@@ -71,19 +71,12 @@ public void setImageUrl(String imageUrl) {
         this.imageUrl = imageUrl;
     }
 
-    public List<WishListProduct> getWishListProducts() {
-        return wishListProducts;
-    }
-
-    public void setWishListProducts(List<WishListProduct> wishListProducts) {
-        this.wishListProducts = wishListProducts;
-    }
-
-    public void addProductOption(String name, Long quentity) {
-        ProductOption productOption = new ProductOption(name, quentity, this);
+    public void addProductOption(CreateProductOptionRequestDTO createProductOptionRequestDTO) {
+        ProductOption productOption = new ProductOption(createProductOptionRequestDTO.getName(), createProductOptionRequestDTO.getQuantity(), this);
         productOptions.add(productOption);
     }
 
+
     public void addProductOption(ProductOption productOption) {
         productOptions.add(productOption);
     }

src/main/java/gift/product/domain/ProductOption.java

@@ -15,6 +15,10 @@ public class ProductOption {
     @JoinColumn(name = "product_id")
     private Product product;
 
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "wishlist_product_id")
+    private WishListProduct wishListProduct;
+
     public ProductOption(String name, Long quentity, Product product) {
         this.name = name;
         this.quentity = quentity;
@@ -24,7 +28,22 @@ public ProductOption(String name, Long quentity, Product product) {
     public ProductOption() {
     }
 
+    public Long getId() {
+        return id;
+    }
+
+    public Long getQuentity() {
+        return quentity;
+    }
+
     public String getName() {
         return name;
     }
+
+    public void decreaseQuantity(Long quentity) {
+        if (this.quentity < quentity) {
+            throw new IllegalArgumentException("재고가 부족합니다.");
+        }
+        this.quentity -= quentity;
+    }
 }

src/main/java/gift/product/domain/WishList.java

@@ -19,7 +19,7 @@ public class WishList {
     @JoinColumn(name = "user_id")
     private User user;
 
-    @OneToMany(mappedBy = "wishList", cascade = CascadeType.ALL, orphanRemoval = true)
+    @OneToMany(mappedBy = "wishList")
     private List<WishListProduct> wishListProducts = new ArrayList<>();
 
     private LocalDateTime createdAt;
@@ -71,4 +71,8 @@ public LocalDateTime getCreatedAt() {
     public void setCreatedAt(LocalDateTime createdAt) {
         this.createdAt = createdAt;
     }
+
+    public void clearProducts() {
+        wishListProducts.clear();
+    }
 }

src/main/java/gift/product/domain/WishListProduct.java

@@ -22,13 +22,16 @@ public class WishListProduct {
     @JoinColumn(name = "product_option_id")
     private ProductOption productOption;
 
+    private Long quantity;
+
     public WishListProduct() {
     }
 
-    public WishListProduct(WishList wishList, Product product, ProductOption productOption) {
+    public WishListProduct(WishList wishList, Product product, ProductOption productOption, Long quantity) {
         this.wishList = wishList;
         this.product = product;
         this.productOption = productOption;
+        this.quantity = quantity;
     }
 
     public Long getId() {
@@ -54,4 +57,13 @@ public Product getProduct() {
     public void setProduct(Product product) {
         this.product = product;
     }
+
+
+    public ProductOption getProductOption() {
+        return productOption;
+    }
+
+    public Long getQuantity() {
+        return quantity;
+    }
 }

src/main/java/gift/product/infra/WishListRepository.java

@@ -0,0 +1,38 @@
+package gift.product.infra;
+
+import gift.product.domain.WishList;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.Pageable;
+import org.springframework.stereotype.Repository;
+
+import java.util.Optional;
+
+@Repository
+public class WishListRepository {
+    private final WishListJpaRepository wishListJpaRepository;
+
+    public WishListRepository(WishListJpaRepository wishListJpaRepository) {
+        this.wishListJpaRepository = wishListJpaRepository;
+    }
+
+    public WishList findByUserId(Long userId) {
+        return wishListJpaRepository.findByUserId(userId)
+                .orElseThrow(() -> new IllegalArgumentException("해당 ID의 위시리스트가 존재하지 않습니다."));
+    }
+
+    public Page<WishList> findByUserId(Long userId, Pageable pageable) {
+        return wishListJpaRepository.findByUserId(userId, pageable);
+    }
+
+    public void save(WishList wishList) {
+        wishListJpaRepository.save(wishList);
+    }
+
+    public WishList findById(Long id) {
+        return wishListJpaRepository.findById(id).orElseThrow(() -> new IllegalArgumentException("해당 ID의 위시리스트가 존재하지 않습니다."));
+    }
+
+    public void delete(WishList wishList) {
+        wishListJpaRepository.delete(wishList);
+    }
+}