Add endpoints for fetching thumbnail IDs and unique video IDs by genres

Author: Trinh-Tan-Dat

src/main/java/com/programming/videoService/controller/VideoController.java

@@ -130,6 +130,19 @@ public ResponseEntity<?> getVideoIdFromThumbnailId(@PathVariable String id) {
         return new ResponseEntity<>(videoService.getVideoIdFromThumbnailId(id), HttpStatus.OK);
     }
 
+    @GetMapping("/getThumbnailIdsByVideoId/{videoId}")
+    public ResponseEntity<?> getThumbnailIdsByVideoId(@PathVariable String videoId) {
+        try {
+            List<String> thumbnailIds = videoService.getThumbnailIdsByVideoId(videoId);
+            return new ResponseEntity<>(thumbnailIds, HttpStatus.OK);
+        } catch (Exception e) {
+            return new ResponseEntity<>("Error fetching thumbnail IDs: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
+
+    //hanle views
+
+
     @PutMapping("/updateViews/{id}")
     public ResponseEntity<?> updateViews(@PathVariable String id) {
         videoService.updateViews(id);
@@ -275,7 +288,31 @@ public ResponseEntity<?> getGenresByUserId(@PathVariable String userId) {
             return new ResponseEntity<>("Error fetching combined genres: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
         }
     }
+
 
+    // Hanle Recommend
+    @GetMapping("/getUniqueVideoIdsByGenres/{userId}")
+    public ResponseEntity<?> getUniqueVideoIdsByGenres(@PathVariable String userId) {
+        try {
+            List<String> uniqueVideoIds = videoService.getUniqueVideoIdsByGenres(userId);
+            return new ResponseEntity<>(uniqueVideoIds, HttpStatus.OK);
+        } catch (Exception e) {
+            return new ResponseEntity<>("Error fetching video IDs: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
+
+    @GetMapping("/getThumbnailsByUserGenres/{userId}")
+    public ResponseEntity<?> getThumbnailsByUserGenres(@PathVariable String userId) {
+        try {
+            List<String> thumbnailIds = videoService.getThumbnailIdsByUserGenres(userId);
+            return new ResponseEntity<>(thumbnailIds, HttpStatus.OK);
+        } catch (Exception e) {
+            return new ResponseEntity<>("Error fetching thumbnails: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
+
+
+
 
 
     //handle Report

src/main/java/com/programming/videoService/service/VideoService.java

@@ -39,14 +39,18 @@
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.LinkedHashMap;
+import java.util.LinkedHashSet;
 import java.util.List;
 import java.util.Map;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.data.domain.Sort;
 import java.util.Properties;
+import java.util.Set;
+
 import org.apache.kafka.common.serialization.StringSerializer;
 import org.apache.kafka.common.serialization.StringDeserializer;
 import java.time.Duration;
@@ -192,6 +196,19 @@ public String getVideoIdFromThumbnailId(String thumbnailId) {
         return gridFSFile.getMetadata().get("videoId").toString();
     }
 
+    public List<String> getThumbnailIdsByVideoId(String videoId) {
+        Query query = Query.query(Criteria.where("metadata.videoId").is(videoId));
+        List<GridFSFile> gridFSFiles = template.find(query).into(new ArrayList<>());
+    
+        List<String> thumbnailIds = new ArrayList<>();
+        for (GridFSFile file : gridFSFiles) {
+            thumbnailIds.add(file.getObjectId().toString());
+        }
+    
+        return thumbnailIds;
+    }
+    
+
     public VideoWithStream getVideoWithStream(String id) throws IOException {
         GridFSFile gridFSFile = template.findOne(new Query(Criteria.where("_id").is(id)));
         MDC.put("type", "videoservice");
@@ -203,6 +220,8 @@ public VideoWithStream getVideoWithStream(String id) throws IOException {
         return null;
     }
 
+    //Hanle view
+
     public void updateViews(String id) {
         Query query = new Query(Criteria.where("_id").is(id));
         Update update = new Update().inc("views", 1);
@@ -353,7 +372,7 @@ public Map<String, Integer> getGenresByUserId(String userId) {
         // 1. Lấy danh sách thumbnailId từ History
         List<String> thumbnailIds = getHistoryByUserId(userId);
 
-        Map<String, Integer> genreCounts = new LinkedHashMap<>();
+        Map<String, Integer> genreCounts = new HashMap<>();
 
         // 2. Duyệt qua từng thumbnailId
         for (String thumbnailId : thumbnailIds) {
@@ -374,12 +393,64 @@ public Map<String, Integer> getGenresByUserId(String userId) {
             }
         }
 
-        return genreCounts;
+        // 3. Sắp xếp genres theo số lượng giảm dần và giới hạn 12 genres
+        return genreCounts.entrySet().stream()
+            .sorted((entry1, entry2) -> Integer.compare(entry2.getValue(), entry1.getValue())) // Sắp xếp giảm dần theo giá trị
+            .limit(12) // Lấy tối đa 12 genres
+            .collect(LinkedHashMap::new, 
+                     (map, entry) -> map.put(entry.getKey(), entry.getValue()), 
+                     LinkedHashMap::putAll);
     }
+
+    //handle recommend
+    public List<String> getUniqueVideoIdsByGenres(String userId) {
+        // 1. Lấy danh sách genres và số lần xuất hiện từ API /getGenresByUserId/{userId}
+        Map<String, Integer> genreCounts = getGenresByUserId(userId);
+
+        Set<String> uniqueVideoIds = new LinkedHashSet<>();
+
+        // 2. Tìm videoId theo từng genre
+        for (String genre : genreCounts.keySet()) {
+            Query query = Query.query(Criteria.where("genres").is(genre));
+            List<DBObject> videoFiles = mongoTemplate.find(query, DBObject.class, "fs.files");
+
+            for (DBObject video : videoFiles) {
+                String videoId = video.get("_id").toString();
+                uniqueVideoIds.add(videoId);
+            }
+        }
+
+        // 3. Trả về danh sách videoId (duy nhất, không trùng lặp)
+        return new ArrayList<>(uniqueVideoIds);
+    }
+
+    public List<String> getThumbnailIdsByUserGenres(String userId) {
+        // Bước 1: Lấy genres từ userId
+        Map<String, Integer> genreCounts = getGenresByUserId(userId);
+    
+        // Dùng Set để đảm bảo không có `thumbnailId` trùng lặp
+        Set<String> uniqueThumbnailIds = new LinkedHashSet<>();
 
+        // Bước 2: Lấy danh sách videoId từ từng genre
+        for (String genre : genreCounts.keySet()) {
+            Query query = Query.query(Criteria.where("genres").is(genre));
+            List<DBObject> videos = mongoTemplate.find(query, DBObject.class, "fs.files");
+            for (DBObject video : videos) {
+                String videoId = video.get("_id").toString();
+    
+                // Bước 3: Lấy danh sách thumbnailId từ videoId
+                List<String> thumbnailIds = getThumbnailIdsByVideoId(videoId);
+                uniqueThumbnailIds.addAll(thumbnailIds);
+            }
+        }
+    
+        // Trả về danh sách thumbnailId duy nhất
+        return new ArrayList<>(uniqueThumbnailIds);
+    }
 
+   
 
-    //hanlde Report
+        //hanlde Report
     public void uploadReport(String videoId, String msg, String userId) {
         Report report = new Report(videoId, msg, userId);
         mongoTemplate.save(report);