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