Get genres user watched

Author: Trinh-Tan-Dat

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

@@ -21,6 +21,7 @@
 import java.io.InputStream;
 import java.nio.charset.StandardCharsets;
 import java.sql.Timestamp;
+import java.util.List;
 import java.util.Map;
 
 import org.slf4j.Logger;
@@ -259,6 +260,24 @@ public ResponseEntity<?> getHistoryByUserId(@PathVariable String userId) {
         return new ResponseEntity<>(videoService.getHistoryByUserId(userId), HttpStatus.OK);
     }
 
+    //Handle Genre
+    @GetMapping("/getGenresByVideoId/{videoId}")
+    public ResponseEntity<?> getGenresByVideoId(@PathVariable String videoId) {
+        return new ResponseEntity<>(videoService.getGenresByVideoId(videoId), HttpStatus.OK);
+    }
+
+    @GetMapping("/getGenresByUserId/{userId}")
+    public ResponseEntity<?> getGenresByUserId(@PathVariable String userId) {
+        try {
+            Map<String, Integer> combinedGenres = videoService.getGenresByUserId(userId);
+            return new ResponseEntity<>(combinedGenres, HttpStatus.OK);
+        } catch (Exception e) {
+            return new ResponseEntity<>("Error fetching combined genres: " + e.getMessage(), HttpStatus.INTERNAL_SERVER_ERROR);
+        }
+    }
+    
+
+
     //handle Report
     @PostMapping("/uploadReport")
     public ResponseEntity<?> uploadReport(@RequestParam("videoId") String videoId,

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

@@ -339,6 +339,46 @@ public List<String> getHistoryByUserId(String userId) {
         return thumbIds;
     }
 
+    //Hanle genres
+    public List<String> getGenresByVideoId(String videoId){
+        Query query = new Query(Criteria.where("_id").is(videoId));
+        DBObject dbObject = mongoTemplate.findOne(query, DBObject.class, "fs.files");
+        if (dbObject != null) {
+            return (List<String>) dbObject.get("genres");
+        }
+        return null;
+    }
+
+    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<>();
+    
+        // 2. Duyệt qua từng thumbnailId
+        for (String thumbnailId : thumbnailIds) {
+            try {
+                // Lấy videoId từ thumbnailId
+                String videoId = getVideoIdFromThumbnailId(thumbnailId);
+    
+                // Lấy genres từ videoId
+                List<String> genres = getGenresByVideoId(videoId);
+    
+                // Đếm số lần xuất hiện của từng genre
+                for (String genre : genres) {
+                    genreCounts.put(genre, genreCounts.getOrDefault(genre, 0) + 1);
+                }
+            } catch (Exception e) {
+                // Nếu lỗi, ghi log và tiếp tục
+                System.err.println("Error processing thumbnailId: " + thumbnailId + ", error: " + e.getMessage());
+            }
+        }
+    
+        return genreCounts;
+    }
+    
+    
+
     //hanlde Report
     public void uploadReport(String videoId, String msg, String userId) {
         Report report = new Report(videoId, msg, userId);