From 81db7b3b070c003e6f5f0d1c757ab30b6f42c944 Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期六, 10 十月 2020 18:49:41 +0800
Subject: [PATCH] 百度内容联盟

---
 src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java |   46 +++++++++++++++++++++++++++++++++++++++++++---
 1 files changed, 43 insertions(+), 3 deletions(-)

diff --git a/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java b/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java
index 8fe3f86..be0cdfb 100644
--- a/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java
+++ b/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java
@@ -1,7 +1,11 @@
 package com.yeshi.buwan.service.manager;
 
+import com.yeshi.buwan.dao.video.AlbumVideoMapDao;
 import com.yeshi.buwan.domain.SolrVideo;
 import com.yeshi.buwan.domain.VideoInfo;
+import com.yeshi.buwan.domain.video.AlbumVideoMap;
+import com.yeshi.buwan.service.imp.VideoInfoService;
+import com.yeshi.buwan.service.inter.juhe.AlbumVideoMapService;
 import com.yeshi.buwan.util.factory.SolrVideoFactory;
 import org.apache.solr.client.solrj.response.UpdateResponse;
 import org.springframework.data.solr.core.SolrTemplate;
@@ -23,6 +27,12 @@
 
     @Resource
     private SolrTemplate solrTemplate;
+
+    @Resource
+    private AlbumVideoMapService albumVideoMapService;
+
+    @Resource
+    private VideoInfoService videoInfoService;
 
 
     public void saveOrUpdate(VideoInfo videoInfo) {
@@ -63,9 +73,9 @@
 
     public List<SolrVideo> findByKey(String key, Integer contentType, Integer videoType, int page) {
         int pageSize = 20;
-        Query query = new SimpleQuery("name:\"" + key+"\"");
+        Query query = new SimpleQuery("name:\"" + key + "\"");
 
-        Criteria criteria =new Criteria("contenttype").is(contentType);
+        Criteria criteria = new Criteria("contenttype").is(contentType);
 
         if (videoType != null)
             criteria = criteria.and("root_video_type").is(videoType);
@@ -85,7 +95,7 @@
     }
 
     public List<String> getSuggestKeyList(String key) {
-        Query query = new SimpleQuery("name:\""+key+"\"");
+        Query query = new SimpleQuery("name:\"" + key + "\"");
         /** 璁剧疆鍒嗛〉寮�濮嬭褰曟暟(绗竴椤�) 榛樿0 */
         query.setOffset(0);
         /** 璁剧疆姣忛〉鏄剧ず璁板綍鏁帮紝榛樿10 */
@@ -101,4 +111,34 @@
     }
 
 
+    /**
+     * 鍚屾鎵�鏈変笓杈�
+     */
+    public void syncAllAlbum() {
+        long count = albumVideoMapService.countAll();
+        int pageSize = 100;
+        int pageCount = (int) (count % pageSize == 0 ? count / pageSize : count / pageSize + 1);
+        for (int i = 3; i < pageCount; i++) {
+            List<String> videoIds = new ArrayList<>();
+            List<AlbumVideoMap> albumVideoMapList = albumVideoMapService.listAll(i + 1, pageSize);
+            for (AlbumVideoMap map : albumVideoMapList) {
+                videoIds.add(map.getVideoId());
+            }
+            List<VideoInfo> videoList = videoInfoService.listByVideoIds(videoIds);
+            saveOrUpdate(videoList);
+        }
+    }
+
+    public void syncAlbum(String videoId) {
+        AlbumVideoMap map = albumVideoMapService.selectByVideoId(videoId);
+        if (map == null)
+            return;
+
+        List<VideoInfo> videoList = new ArrayList<>();
+        VideoInfo video = videoInfoService.getVideoInfo(videoId);
+        if (video != null)
+            videoList.add(video);
+        saveOrUpdate(videoList);
+    }
+
 }

--
Gitblit v1.8.0