YEL-98 [feat] 구독 연장 유도 필요 여부 확인 API 추가

Lee Euije · web-flow · commit 7d806cc01948 · 2023-08-15T05:24:25.000+09:00
diff --git a/src/main/java/com/yello/server/domain/purchase/controller/PurchaseController.java b/src/main/java/com/yello/server/domain/purchase/controller/PurchaseController.java
@@ -1,8 +1,21 @@
 package com.yello.server.domain.purchase.controller;
 
+import static com.yello.server.global.common.SuccessCode.USER_SUBSCRIBE_NEEDED_READ_SUCCESS;
+
+import com.yello.server.domain.purchase.dto.response.UserSubscribeNeededResponse;
 import com.yello.server.domain.purchase.service.PurchaseService;
+import com.yello.server.domain.user.entity.User;
+import com.yello.server.global.common.annotation.AccessTokenUser;
+import com.yello.server.global.common.dto.BaseResponse;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.media.Content;
+import io.swagger.v3.oas.annotations.media.Schema;
+import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import java.time.LocalDateTime;
 import lombok.RequiredArgsConstructor;
+import lombok.val;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
@@ -14,4 +27,15 @@ public class PurchaseController {
 
     private final PurchaseService purchaseService;
 
+    @Operation(summary = "구독 연장 유도 필요 여부 확인 API", responses = {
+        @ApiResponse(
+            responseCode = "200",
+            content = @Content(mediaType = "application/json", schema = @Schema(implementation = UserSubscribeNeededResponse.class))
+        )
+    })
+    @GetMapping("/subscribe_need")
+    public BaseResponse<UserSubscribeNeededResponse> getUserSubscribeNeeded(@AccessTokenUser User user) {
+        val data = purchaseService.getUserSubscribe(user, LocalDateTime.now());
+        return BaseResponse.success(USER_SUBSCRIBE_NEEDED_READ_SUCCESS, data);
+    }
 }
diff --git a/src/main/java/com/yello/server/domain/purchase/dto/response/UserSubscribeNeededResponse.java b/src/main/java/com/yello/server/domain/purchase/dto/response/UserSubscribeNeededResponse.java
@@ -0,0 +1,19 @@
+package com.yello.server.domain.purchase.dto.response;
+
+import com.yello.server.domain.user.entity.Subscribe;
+import com.yello.server.domain.user.entity.User;
+import lombok.Builder;
+
+@Builder
+public record UserSubscribeNeededResponse(
+    Subscribe subscribe,
+    Boolean isSubscribeNeeded
+) {
+
+    public static UserSubscribeNeededResponse of(User user, Boolean isSubscribeNeeded) {
+        return UserSubscribeNeededResponse.builder()
+            .subscribe(user.getSubscribe())
+            .isSubscribeNeeded(isSubscribeNeeded)
+            .build();
+    }
+}
diff --git a/src/main/java/com/yello/server/domain/purchase/entity/Purchase.java b/src/main/java/com/yello/server/domain/purchase/entity/Purchase.java
@@ -13,13 +13,15 @@
 import javax.persistence.ManyToOne;
 import lombok.AccessLevel;
 import lombok.AllArgsConstructor;
-import lombok.Builder;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
+import org.hibernate.annotations.DynamicInsert;
 
 @Entity
 @Getter
-@Builder
+@SuperBuilder
+@DynamicInsert
 @AllArgsConstructor
 @NoArgsConstructor(access = AccessLevel.PROTECTED)
 public class Purchase extends AuditingTimeEntity {
diff --git a/src/main/java/com/yello/server/domain/purchase/repository/PurchaseJpaRepository.java b/src/main/java/com/yello/server/domain/purchase/repository/PurchaseJpaRepository.java
@@ -1,8 +1,15 @@
 package com.yello.server.domain.purchase.repository;
 
+import com.yello.server.domain.purchase.entity.ProductType;
 import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.user.entity.User;
+import java.util.List;
+import java.util.Optional;
 import org.springframework.data.jpa.repository.JpaRepository;
 
 public interface PurchaseJpaRepository extends JpaRepository<Purchase, Long> {
 
+    List<Purchase> findAllByUser(User user);
+
+    Optional<Purchase> findTopByUserAndProductTypeOrderByCreatedAtDesc(User user, ProductType productType);
 }
diff --git a/src/main/java/com/yello/server/domain/purchase/repository/PurchaseRepository.java b/src/main/java/com/yello/server/domain/purchase/repository/PurchaseRepository.java
@@ -1,5 +1,18 @@
 package com.yello.server.domain.purchase.repository;
 
+import com.yello.server.domain.purchase.entity.ProductType;
+import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.user.entity.User;
+import java.util.List;
+import java.util.Optional;
+
 public interface PurchaseRepository {
 
+    Purchase save(Purchase purchase);
+
+    Optional<Purchase> findById(Long purchaseId);
+
+    List<Purchase> findAllByUser(User user);
+
+    Optional<Purchase> findTopByUserAndProductTypeOrderByCreatedAtDesc(User user, ProductType productType);
 }
diff --git a/src/main/java/com/yello/server/domain/purchase/repository/PurchaseRepositoryImpl.java b/src/main/java/com/yello/server/domain/purchase/repository/PurchaseRepositoryImpl.java
@@ -1,10 +1,36 @@
 package com.yello.server.domain.purchase.repository;
 
+import com.yello.server.domain.purchase.entity.ProductType;
+import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.user.entity.User;
+import java.util.List;
+import java.util.Optional;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Repository;
 
 @Repository
 @RequiredArgsConstructor
 public class PurchaseRepositoryImpl implements PurchaseRepository {
 
+    public final PurchaseJpaRepository purchaseJpaRepository;
+
+    @Override
+    public Purchase save(Purchase purchase) {
+        return purchaseJpaRepository.save(purchase);
+    }
+
+    @Override
+    public Optional<Purchase> findById(Long purchaseId) {
+        return purchaseJpaRepository.findById(purchaseId);
+    }
+
+    @Override
+    public List<Purchase> findAllByUser(User user) {
+        return purchaseJpaRepository.findAllByUser(user);
+    }
+
+    @Override
+    public Optional<Purchase> findTopByUserAndProductTypeOrderByCreatedAtDesc(User user, ProductType productType) {
+        return purchaseJpaRepository.findTopByUserAndProductTypeOrderByCreatedAtDesc(user, productType);
+    }
 }
diff --git a/src/main/java/com/yello/server/domain/purchase/service/PurchaseService.java b/src/main/java/com/yello/server/domain/purchase/service/PurchaseService.java
@@ -1,11 +1,21 @@
 package com.yello.server.domain.purchase.service;
 
+import com.yello.server.domain.purchase.dto.response.UserSubscribeNeededResponse;
+import com.yello.server.domain.purchase.entity.ProductType;
+import com.yello.server.domain.purchase.entity.Purchase;
 import com.yello.server.domain.purchase.repository.PurchaseRepository;
+import com.yello.server.domain.user.entity.Subscribe;
+import com.yello.server.domain.user.entity.User;
 import com.yello.server.domain.user.repository.UserRepository;
+import java.time.Duration;
+import java.time.LocalDateTime;
+import java.util.Optional;
+import lombok.Builder;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
+@Builder
 @Service
 @RequiredArgsConstructor
 @Transactional(readOnly = true)
@@ -14,4 +24,13 @@ public class PurchaseService {
     private final UserRepository userRepository;
     private final PurchaseRepository purchaseRepository;
 
+    public UserSubscribeNeededResponse getUserSubscribe(User user, LocalDateTime time) {
+        final Optional<Purchase> mostRecentPurchase = purchaseRepository.findTopByUserAndProductTypeOrderByCreatedAtDesc(
+            user, ProductType.YELLO_PLUS);
+        final Boolean isSubscribeNeeded = user.getSubscribe() == Subscribe.CANCELED
+            && mostRecentPurchase.isPresent()
+            && Duration.between(mostRecentPurchase.get().getCreatedAt(), time).getSeconds() < 1 * 24 * 60 * 60;
+        
+        return UserSubscribeNeededResponse.of(user, isSubscribeNeeded);
+    }
 }
diff --git a/src/main/java/com/yello/server/domain/user/service/UserService.java b/src/main/java/com/yello/server/domain/user/service/UserService.java
@@ -4,6 +4,7 @@
 import com.yello.server.domain.cooldown.repository.CooldownRepository;
 import com.yello.server.domain.friend.entity.Friend;
 import com.yello.server.domain.friend.repository.FriendRepository;
+import com.yello.server.domain.purchase.repository.PurchaseRepository;
 import com.yello.server.domain.user.dto.response.UserDetailResponse;
 import com.yello.server.domain.user.dto.response.UserResponse;
 import com.yello.server.domain.user.entity.User;
@@ -26,6 +27,7 @@ public class UserService {
     private final VoteRepository voteRepository;
     private final CooldownRepository cooldownRepository;
     private final TokenRepository tokenRepository;
+    private final PurchaseRepository purchaseRepository;
 
     public UserDetailResponse findMyProfile(Long userId) {
         final User user = userRepository.getById(userId);
diff --git a/src/main/java/com/yello/server/global/common/SuccessCode.java b/src/main/java/com/yello/server/global/common/SuccessCode.java
@@ -18,6 +18,7 @@ public enum SuccessCode {
     YELLOID_VALIDATION_SUCCESS(OK, "옐로 아이디 중복 여부 조회에 성공했습니다."),
     READ_FRIEND_SUCCESS(OK, "친구 조회에 성공했습니다."),
     READ_USER_SUCCESS(OK, "유저 조회에 성공했습니다."),
+    USER_SUBSCRIBE_NEEDED_READ_SUCCESS(OK, "유저 구독 연장 유도 필요 여부 확인 조회에 성공하였습니다."),
     ADD_FRIEND_SUCCESS(OK, "친구 추가에 성공했습니다."),
     SHUFFLE_FRIEND_SUCCESS(OK, "친구 셔플에 성공했습니다."),
     CHECK_KEYWORD_SUCCESS(OK, "키워드 확인에 성공했습니다."),
diff --git a/src/test/java/com/yello/server/small/domain/purchase/FakePurchaseRepository.java b/src/test/java/com/yello/server/small/domain/purchase/FakePurchaseRepository.java
@@ -0,0 +1,64 @@
+package com.yello.server.small.domain.purchase;
+
+import com.yello.server.domain.purchase.entity.ProductType;
+import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.purchase.repository.PurchaseRepository;
+import com.yello.server.domain.user.entity.User;
+import java.util.ArrayList;
+import java.util.Comparator;
+import java.util.List;
+import java.util.Objects;
+import java.util.Optional;
+
+public class FakePurchaseRepository implements PurchaseRepository {
+
+    private final List<Purchase> data = new ArrayList<>();
+    private Long id = 0L;
+
+
+    @Override
+    public Purchase save(Purchase purchase) {
+        if (purchase.getId() != null && purchase.getId() > id) {
+            id = purchase.getId();
+        }
+
+        final Purchase newPurchase = Purchase.builder()
+            .id(purchase.getId() == null ? ++id : purchase.getId())
+            .price(purchase.getPrice())
+            .user(purchase.getUser())
+            .gateway(purchase.getGateway())
+            .productType(purchase.getProductType())
+            .createdAt(purchase.getCreatedAt())
+            .updatedAt(purchase.getUpdatedAt())
+            .build();
+
+        data.add(newPurchase);
+        return newPurchase;
+    }
+
+    @Override
+    public Optional<Purchase> findById(Long purchaseId) {
+        return data.stream()
+            .filter(purchase -> Objects.equals(purchase.getId(), purchaseId))
+            .findFirst();
+    }
+
+    @Override
+    public List<Purchase> findAllByUser(User user) {
+        return data.stream()
+            .filter(purchase -> purchase.getUser().equals(user))
+            .toList();
+    }
+
+    @Override
+    public Optional<Purchase> findTopByUserAndProductTypeOrderByCreatedAtDesc(User user, ProductType productType) {
+        return data.stream()
+            .filter(purchase -> {
+                    return purchase.getUser().getId().equals(user.getId())
+                        && purchase.getProductType().equals(productType);
+                }
+            )
+            .sorted(Comparator.comparing(Purchase::getCreatedAt).reversed())
+            .findFirst();
+    }
+}
diff --git a/src/test/java/com/yello/server/small/domain/purchase/PurchaseTest.java b/src/test/java/com/yello/server/small/domain/purchase/PurchaseTest.java
@@ -0,0 +1,76 @@
+package com.yello.server.small.domain.purchase;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import com.yello.server.domain.group.entity.School;
+import com.yello.server.domain.purchase.dto.response.UserSubscribeNeededResponse;
+import com.yello.server.domain.purchase.entity.Gateway;
+import com.yello.server.domain.purchase.entity.ProductType;
+import com.yello.server.domain.purchase.entity.Purchase;
+import com.yello.server.domain.purchase.repository.PurchaseRepository;
+import com.yello.server.domain.purchase.service.PurchaseService;
+import com.yello.server.domain.user.entity.Gender;
+import com.yello.server.domain.user.entity.Social;
+import com.yello.server.domain.user.entity.Subscribe;
+import com.yello.server.domain.user.entity.User;
+import com.yello.server.domain.user.repository.UserRepository;
+import com.yello.server.small.domain.user.FakeUserRepository;
+import java.time.LocalDateTime;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+public class PurchaseTest {
+
+    private final UserRepository userRepository = new FakeUserRepository();
+    private final PurchaseRepository purchaseRepository = new FakePurchaseRepository();
+    private PurchaseService purchaseService;
+
+    @BeforeEach
+    void init() {
+        this.purchaseService = PurchaseService.builder()
+            .userRepository(userRepository)
+            .purchaseRepository(purchaseRepository)
+            .build();
+
+        School school = School.builder()
+            .schoolName("Test School")
+            .departmentName("Testing")
+            .build();
+        final User user = User.builder()
+            .id(1L)
+            .recommendCount(0L).name("test")
+            .yelloId("yelloworld").gender(Gender.MALE)
+            .point(200).social(Social.KAKAO)
+            .profileImage("test image").uuid("1234")
+            .deletedAt(null).group(school)
+            .groupAdmissionYear(20).email("test@test.com")
+            .subscribe(Subscribe.CANCELED)
+            .build();
+        userRepository.save(user);
+
+        purchaseRepository.save(Purchase.builder()
+            .id(1L)
+            .price(1000)
+            .user(user)
+            .gateway(Gateway.GOOGLE)
+            .createdAt(LocalDateTime.now())
+            .updatedAt(LocalDateTime.now())
+            .productType(ProductType.YELLO_PLUS)
+            .build());
+    }
+
+    @Test
+    void 구독_연장_유도_필요_여부_확인_조회에_성공합니다() {
+        // given
+        Long userId = 1L;
+
+        // when
+        final User user = userRepository.getById(userId);
+        final UserSubscribeNeededResponse response = purchaseService.getUserSubscribe(user,
+            LocalDateTime.now().plusDays(1).minusMinutes(1));
+
+        // then
+        assertThat(response.subscribe()).isEqualTo(user.getSubscribe());
+        assertThat(response.isSubscribeNeeded()).isEqualTo(true);
+    }
+}
diff --git a/src/test/java/com/yello/server/small/domain/user/FakeUserRepository.java b/src/test/java/com/yello/server/small/domain/user/FakeUserRepository.java
@@ -36,6 +36,7 @@ public User save(User user) {
             .group(user.getGroup())
             .groupAdmissionYear(user.getGroupAdmissionYear())
             .email(user.getEmail())
+            .subscribe(user.getSubscribe())
             .build();
 
         data.add(newUser);
