Perf: [BADA-368] 추천 고도화

src/main/java/com/TwoSeaU/BaData/domain/trade/controller/MockController.java

@@ -0,0 +1,21 @@
+package com.TwoSeaU.BaData.domain.trade.controller;
+
+import com.TwoSeaU.BaData.domain.trade.service.PostService;
+import com.TwoSeaU.BaData.global.response.ApiResponse;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequiredArgsConstructor
+@RequestMapping("/api/v1/trades")
+public class MockController {
+    private final PostService postService;
+
+    @PostMapping("/mock/posts")
+    public ResponseEntity<ApiResponse<String>> createMockPost() {
+        return ResponseEntity.ok().body(ApiResponse.success(postService.generateGifticons()));
+    }
+}

src/main/java/com/TwoSeaU/BaData/domain/trade/controller/PostController.java

@@ -37,6 +37,15 @@ public ResponseEntity<ApiResponse<CursorPageResponse<PostResponse>>> getPosts(
         return ResponseEntity.ok().body(ApiResponse.success(postSearchService.searchPosts(query, user == null ? null : user.getUsername(), cursor, size)));
     }
 
+    @GetMapping("/posts/rdb")
+    public ResponseEntity<ApiResponse<CursorPageResponse<PostResponse>>> getPostsByRDB(
+            @RequestParam(required = false) Long cursor,
+            @RequestParam(defaultValue = "10") int size,
+            @RequestParam(required = false) String query,
+            @AuthenticationPrincipal User user) {
+        return ResponseEntity.ok().body(ApiResponse.success(postSearchService.searchPostsByRDB(query, user == null ? null : user.getUsername(), cursor, size)));
+    }
+
     @GetMapping("/posts/{userId}/{isSold}")
     public ResponseEntity<ApiResponse<CursorPageResponse<PostResponse>>> getPostsByUserId(
             @RequestParam(required = false) Long cursor,

src/main/java/com/TwoSeaU/BaData/domain/trade/controller/RecommendController.java

@@ -3,7 +3,10 @@
 import com.TwoSeaU.BaData.domain.trade.dto.response.PostsResponse;
 import com.TwoSeaU.BaData.domain.trade.dto.response.SaveRecommendLikesResponse;
 import com.TwoSeaU.BaData.domain.trade.service.LikeService;
-import com.TwoSeaU.BaData.domain.trade.service.RecommendService;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.GlobalRecommendService;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.doubleVector.RecommendServiceDouble;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.floatVector.RecommendServiceFloat;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.pgVector.RecommendServicePg;
 import com.TwoSeaU.BaData.global.response.ApiResponse;
 import lombok.RequiredArgsConstructor;
 import org.springframework.http.ResponseEntity;
@@ -15,16 +18,40 @@
 @RequiredArgsConstructor
 @RequestMapping("/api/v1/trades/posts/recommend")
 public class RecommendController {
-    private final RecommendService recommendService;
+    private final RecommendServiceDouble doubleRecommendServiceDouble;
+    private final RecommendServiceFloat floatRecommendServiceFloat;
+    private final RecommendServicePg pgvectorRecommendServicePg;
+    private final GlobalRecommendService globalRecommendService;
     private final LikeService likeService;
 
+    @GetMapping("/double")
+    public ResponseEntity<ApiResponse<PostsResponse>> recommendPostsByDouble(@AuthenticationPrincipal User user, @RequestParam(defaultValue = "true") boolean isStart) {
+        return ResponseEntity.ok().body(ApiResponse.success(doubleRecommendServiceDouble.recommendPosts(user == null ? null : user.getUsername(), isStart)));
+    }
+
+    @GetMapping("/float")
+    public ResponseEntity<ApiResponse<PostsResponse>> recommendPostsByFloat(@AuthenticationPrincipal User user, @RequestParam(defaultValue = "true") boolean isStart) {
+        return ResponseEntity.ok().body(ApiResponse.success(floatRecommendServiceFloat.recommendPostsByFloat(user == null ? null : user.getUsername(), isStart)));
+    }
+
+    //@GetMapping("/pgvector")
     @GetMapping
-    public ResponseEntity<ApiResponse<PostsResponse>> recommendPosts(@AuthenticationPrincipal User user, @RequestParam(defaultValue = "true") boolean isStart) {
-        return ResponseEntity.ok().body(ApiResponse.success(recommendService.recommendPosts(user == null ? null : user.getUsername(), isStart)));
+    public ResponseEntity<ApiResponse<PostsResponse>> recommendPostsByPgvector(@AuthenticationPrincipal User user, @RequestParam(defaultValue = "true") boolean isStart) {
+        return ResponseEntity.ok().body(ApiResponse.success(pgvectorRecommendServicePg.recommendPostsBypgVector(user == null ? null : user.getUsername(), isStart)));
     }
 
     @PostMapping("/likes/{postId}")
     public ResponseEntity<ApiResponse<SaveRecommendLikesResponse>> likeRecommendation(@AuthenticationPrincipal User user, @PathVariable Long postId) {
         return ResponseEntity.ok().body(ApiResponse.success(likeService.likesAtRecommendation(user == null ? null : user.getUsername(), postId)));
     }
+
+    @PatchMapping("/vector/update/all/double")
+    public ResponseEntity<ApiResponse<String>> updateAllPostForDouble() {
+        return ResponseEntity.ok().body(ApiResponse.success(globalRecommendService.updateAllDoubleVector()));
+    }
+
+    @PatchMapping("/vector/update/all/float")
+    public ResponseEntity<ApiResponse<String>> updateAllPostVectorForPgvector() {
+        return ResponseEntity.ok().body(ApiResponse.success(globalRecommendService.updateAllFloatAndPgVector()));
+    }
 }

src/main/java/com/TwoSeaU/BaData/domain/trade/entity/Gifticon.java

@@ -23,7 +23,9 @@ public class Gifticon extends Post{
     @Column(nullable = false)
     private String partner;
 
-    private double[] vector;
+    private double[] doubleVector;
+
+    private float[] floatVector;
 
     @ManyToOne(fetch = FetchType.EAGER)
     @JoinColumn(name = "category_id", nullable = false)
@@ -34,16 +36,21 @@ public class Gifticon extends Post{
     public Gifticon(final User user, final String title, final String comment, final BigDecimal price,
                     final LocalDate deadLine, final String postImage, final Boolean isSold,
                     final String couponNumber, final String partner, final GifticonCategory category,
-                    final double[] vector) {
+                    final double[] doubleVector, final float[] floatVector) {
         super(user, title, comment, price, deadLine, postImage, isSold);
         this.couponNumber = couponNumber;
         this.partner = partner;
         this.category = category;
-        this.vector = vector;
+        this.doubleVector = doubleVector;
+        this.floatVector = floatVector;
+    }
+
+    public void updateDoubleVector(final double[] doubleVector) {
+        this.doubleVector = doubleVector;
     }
 
-    public void updateVector(final double[] vector) {
-        this.vector = vector;
+    public void updateFloatVector(final float[] floatVector) {
+        this.floatVector = floatVector;
     }
 
     public void updateBarcodeViewTime(final LocalDateTime barcodeViewTime) {

src/main/java/com/TwoSeaU/BaData/domain/trade/entity/PostVector.java

@@ -0,0 +1,25 @@
+package com.TwoSeaU.BaData.domain.trade.entity;
+
+import jakarta.persistence.*;
+import lombok.AccessLevel;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Entity
+@Getter
+@NoArgsConstructor(access = AccessLevel.PRIVATE)
+@AllArgsConstructor(access = AccessLevel.PRIVATE)
+@Table(name = "post_vector")
+public class PostVector {
+    @Id
+    private Long id;
+
+    @Column(name = "vector_pg", columnDefinition = "vector(57)")
+    private float[] vectorPg;
+
+    @OneToOne(fetch = FetchType.LAZY)
+    @MapsId
+    @JoinColumn(name = "id")
+    private Post post;
+}

src/main/java/com/TwoSeaU/BaData/domain/trade/service/PostSearchService.java

@@ -56,6 +56,16 @@ public CursorPageResponse<PostResponse> searchPosts(final String query, final St
         return searchByTitleAndComment(query, username, cursor, size);
     }
 
+    public CursorPageResponse<PostResponse> searchPostsByRDB(final String query, final String username, final Long cursor, final int size) {
+
+        if (query != null && !query.isEmpty()) {
+            log.info("event-keyword-search, {}", query);
+            saveSearchHistoryIfUserExists(username, query);
+        }
+
+        return postRepository.searchPostsByKeyword(query, username, cursor, size);
+    }
+
     private void saveSearchHistoryIfUserExists(final String username, final String query) {
         if (username == null) {
             return;

src/main/java/com/TwoSeaU/BaData/domain/trade/service/PostService.java

@@ -11,6 +11,9 @@
 import com.TwoSeaU.BaData.domain.trade.enums.PaymentStatus;
 import com.TwoSeaU.BaData.domain.trade.exception.TradeException;
 import com.TwoSeaU.BaData.domain.trade.repository.*;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.doubleVector.PostVectorizerDouble;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.floatVector.PostVectorizerFloat;
+import com.TwoSeaU.BaData.domain.trade.service.recommend.pgVector.VectorUtilsPg;
 import com.TwoSeaU.BaData.domain.user.entity.User;
 import com.TwoSeaU.BaData.domain.user.exception.UserException;
 import com.TwoSeaU.BaData.domain.user.repository.UserRepository;
@@ -22,9 +25,17 @@
 import jakarta.servlet.http.HttpServletResponse;
 import jakarta.transaction.Transactional;
 import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.math.BigDecimal;
+import java.time.LocalDate;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Stream;
+
+@Slf4j
 @Service
 @RequiredArgsConstructor
 @Transactional
@@ -38,10 +49,14 @@ public class PostService {
     private final PostLikesRepository postLikesRepository;
     private final PaymentRepository paymentRepository;
     private final ELAService elaService;
-    private final PostVectorizer postVectorizer;
+    private final PostVectorizerDouble postVectorizerDouble;
+    private final PostVectorizerFloat postVectorizerFloat;
+    private final VectorUtilsPg vectorUtilsPg;
     private final S3ImageService s3ImageService;
     private final OCRService ocrService;
     private final PostDocumentRepository postDocumentRepository;
+    private final JdbcRepository jdbcRepository;
+    private final PartnerRepository partnerRepository;
 
     private static final String COOKIE_NAME = "View_Post";
 
@@ -81,7 +96,15 @@ public SavePostResponse createGifticonPost(final SaveGifticonPostRequest saveGif
         }
 
         final String imageUrl = s3ImageService.saveImage(saveGifticonPostRequest.getFile(), "trades/gifticon/", saveGifticonPostRequest.getFile().getOriginalFilename());
-        final double[] vector = postVectorizer.vectorizePost(
+
+        final double[] doubleVector = postVectorizerDouble.vectorizePost(
+                saveGifticonPostRequest.getPrice(),
+                saveGifticonPostRequest.getDeadLine(),
+                category,
+                saveGifticonPostRequest.getPartner()
+        );
+
+        final float[] floatVector = postVectorizerFloat.vectorizePost(
                 saveGifticonPostRequest.getPrice(),
                 saveGifticonPostRequest.getDeadLine(),
                 category,
@@ -99,13 +122,19 @@ public SavePostResponse createGifticonPost(final SaveGifticonPostRequest saveGif
                 saveGifticonPostRequest.getCouponNumber(),
                 saveGifticonPostRequest.getPartner(),
                 category,
-                vector
+                doubleVector,
+                floatVector
+                //floatVector
+                //byteVector
         );
 
         final Gifticon savedGifticon = gifticonRepository.save(gifticon);
         final PostDocument postDocument = PostDocument.from(savedGifticon);
         postDocumentRepository.save(postDocument);
 
+        //TODO: 여기
+        jdbcRepository.insertPostVector(savedGifticon.getId(), floatVector);
+
         return SavePostResponse.builder()
                 .postId(savedGifticon.getId())
                 .build();
@@ -260,7 +289,14 @@ public SavePostResponse modifyPostGifticon(final Long postId, final UpdateGiftic
             throw new GeneralException(TradeException.GIFTICON_NOT_FOUND);
         }
 
-        final double[] vector = postVectorizer.vectorizePost(
+        final double[] doubleVector = postVectorizerDouble.vectorizePost(
+                updateGifticonPostRequest.getPrice(),
+                gifticon.getDeadLine(),
+                gifticon.getCategory(),
+                gifticon.getPartner()
+        );
+
+        final float[] floatVector = postVectorizerFloat.vectorizePost(
                 updateGifticonPostRequest.getPrice(),
                 gifticon.getDeadLine(),
                 gifticon.getCategory(),
@@ -272,14 +308,83 @@ public SavePostResponse modifyPostGifticon(final Long postId, final UpdateGiftic
                 updateGifticonPostRequest.getPrice()
         );
 
-        gifticon.updateVector(vector);
+        gifticon.updateDoubleVector(doubleVector);
+        gifticon.updateFloatVector(floatVector);
+        jdbcRepository.updatePostVector(gifticon.getId(), floatVector);
 
         final PostDocument postDocument = PostDocument.from(gifticon);
         postDocumentRepository.save(postDocument);
 
         return SavePostResponse.of(post.getId());
     }
 
+    public String generateGifticons() {
+        List<GifticonCategory> categories = gifticonCategoryRepository.findAll();
+        User user = userRepository.findById(2L)
+                .orElseThrow(() -> new GeneralException(UserException.USER_NOT_FOUND));
+
+        List<LocalDate> dates = Stream.iterate(LocalDate.now().plusDays(0), date -> date.plusDays(2))
+                .limit(60)
+                .toList();
+
+        List<BigDecimal> prices = new ArrayList<>();
+        for (int price = 1000; price <= 30000; price += 1000) {
+            prices.add(BigDecimal.valueOf(price));
+        }
+
+        int index = 1;
+
+        for (GifticonCategory category : categories) {
+            log.info("category : " + category.getCategoryName());
+            List<Partner> partners = partnerRepository.findByCategoryId(category.getId());
+
+            for (Partner partner : partners) {
+                log.info("partner : " + partner.getPartner());
+
+                for (LocalDate date : dates) {
+                    for (BigDecimal price : prices) {
+                        final double[] doubleVector = postVectorizerDouble.vectorizePost(
+                                price,
+                                date,
+                                category,
+                                partner.getPartner()
+                        );
+
+                        final float[] floatVector = postVectorizerFloat.vectorizePost(
+                                price,
+                                date,
+                                category,
+                                partner.getPartner()
+                        );
+
+                        Gifticon gifticon = new Gifticon(
+                                user,
+                                partner.getPartner() + " 싸게 팔아요",
+                                null,
+                                price,
+                                date,
+                                "temp.png",
+                                false,
+                                index+"",
+                                partner.getPartner(),
+                                category,
+                                doubleVector,
+                                floatVector
+                        );
+
+                        gifticonRepository.save(gifticon);
+                        jdbcRepository.insertPostVector(gifticon.getId(), floatVector);
+                        log.info("post id : " + index);
+
+                        index++;
+                    }
+                }
+            }
+        }
+
+        return "success";
+    }
+
     private Post validateAndGetPost(final Long postId, final String username) {
         final User user = getUserByUsername(username);
         final Post post = getPostById(postId);