From 0a18a8cb0a7a57bf1f82df425251334c57f8c39a Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期六, 23 一月 2021 15:40:37 +0800 Subject: [PATCH] 新增手机号/qq/微信绑定 --- src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java | 85 ++++++++++++++++++++++++++++++++++++------ 1 files changed, 72 insertions(+), 13 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 3949aa9..cc55f50 100644 --- a/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java +++ b/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java @@ -1,30 +1,42 @@ package com.yeshi.buwan.service.manager; +import com.yeshi.buwan.domain.ResourceVideo; import com.yeshi.buwan.domain.SolrVideo; import com.yeshi.buwan.domain.VideoInfo; +import com.yeshi.buwan.domain.VideoResource; import com.yeshi.buwan.domain.video.AlbumVideoMap; import com.yeshi.buwan.dto.search.SolrResultDTO; +import com.yeshi.buwan.dto.search.SolrVideoSearchFilter; +import com.yeshi.buwan.service.imp.ResourceVideoService; import com.yeshi.buwan.service.imp.VideoInfoService; import com.yeshi.buwan.service.inter.juhe.AlbumVideoMapService; +import com.yeshi.buwan.util.StringUtil; import com.yeshi.buwan.util.factory.SolrVideoFactory; import org.apache.solr.client.solrj.response.UpdateResponse; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; import org.springframework.data.domain.Sort; import org.springframework.data.solr.core.SolrTemplate; import org.springframework.data.solr.core.query.Criteria; import org.springframework.data.solr.core.query.Query; import org.springframework.data.solr.core.query.SimpleQuery; +import org.springframework.data.solr.core.query.SolrDataQuery; import org.springframework.data.solr.core.query.result.ScoredPage; import org.springframework.stereotype.Component; import javax.annotation.Resource; import java.util.ArrayList; +import java.util.HashMap; import java.util.List; +import java.util.Map; /** * 鎼滅储寮曟搸-涓撹緫鏁版嵁绠$悊 */ @Component public class SolrAlbumDataManager { + + private final Logger logger = LoggerFactory.getLogger(SolrAlbumDataManager.class); @Resource private SolrTemplate solrTemplate; @@ -35,9 +47,12 @@ @Resource private VideoInfoService videoInfoService; + @Resource + private ResourceVideoService resourceVideoService; + public void saveOrUpdate(VideoInfo videoInfo) { - UpdateResponse updateResponse = solrTemplate.saveBean(SolrVideoFactory.create(videoInfo)); + UpdateResponse updateResponse = solrTemplate.saveBean(SolrVideoFactory.create(videoInfo, videoInfo.getResourceList())); if (updateResponse.getStatus() == 0) { solrTemplate.commit(); } else { @@ -48,7 +63,7 @@ public void saveOrUpdate(List<VideoInfo> videoInfoList) { List<SolrVideo> solrVideoList = new ArrayList<>(); for (VideoInfo vi : videoInfoList) { - solrVideoList.add(SolrVideoFactory.create(vi)); + solrVideoList.add(SolrVideoFactory.create(vi, vi.getResourceList())); } UpdateResponse updateResponse = solrTemplate.saveBeans(solrVideoList); if (updateResponse.getStatus() == 0) { @@ -58,7 +73,22 @@ } } + /** + *鍒犻櫎鎵�鏈� + */ + public void clear() { + Query query = new SimpleQuery("name:*"); + UpdateResponse updateResponse = solrTemplate.delete(query); + if (updateResponse.getStatus() == 0) { + solrTemplate.commit(); + } else { + solrTemplate.rollback(); + } + } + public void deleteById(String id) { + + UpdateResponse updateResponse = solrTemplate.deleteById(id); if (updateResponse.getStatus() == 0) { solrTemplate.commit(); @@ -72,13 +102,18 @@ return solrVideo; } - public SolrResultDTO findByKey(String key, Integer contentType, Integer videoType, int page, int pageSize) { - Query query = new SimpleQuery("name:\"" + key + "\""); + public SolrResultDTO find(SolrVideoSearchFilter filter, int page, int pageSize) { + logger.info(filter.getKey() + "#" + page); + Query query = new SimpleQuery("name:\"" + filter.getKey() + "\""); - Criteria criteria = new Criteria("contenttype").is(contentType); + Criteria criteria = new Criteria("contenttype").is(filter.getContentType()); - if (videoType != null) - criteria = criteria.and("root_video_type").is(videoType); + if (filter.getVideoType() != null) { + criteria = criteria.and("root_video_type").is(filter.getVideoType()); + } + if (filter.getResourceIds() != null && filter.getResourceIds().length > 0) { + criteria = criteria.and("resourceIds").contains(filter.getResourceIds()); + } /** 娣诲姞鏉′欢 */ if (criteria != null) @@ -88,7 +123,11 @@ query.setOffset((page - 1) * pageSize); /** 璁剧疆姣忛〉鏄剧ず璁板綍鏁帮紝榛樿10 */ query.setRows(pageSize); - query.addSort(new Sort(Sort.Direction.DESC, "year")); + if (StringUtil.isNullOrEmpty(filter.getSortKey())) { + query.addSort(new Sort(Sort.Direction.DESC, "year")); + } else { + query.addSort(new Sort(Sort.Direction.DESC, filter.getSortKey())); + } ScoredPage<SolrVideo> result = solrTemplate.queryForPage(query, SolrVideo.class); System.out.println("鎬昏褰曟暟锛�" + result.getTotalElements()); List<SolrVideo> list = result.getContent(); @@ -126,6 +165,23 @@ videoIds.add(map.getVideoId()); } List<VideoInfo> videoList = videoInfoService.listByVideoIds(videoIds); + + //鑾峰彇璧勬簮ID + List<ResourceVideo> rvList = resourceVideoService.getResourceList(videoList); + Map<String, List<VideoResource>> map = new HashMap<>(); + for (ResourceVideo rv : rvList) { + if (rv.getVideo() != null && rv.getResource() != null) { + if (map.get(rv.getVideo().getId()) == null) + map.put(rv.getVideo().getId(), new ArrayList<>()); + map.get(rv.getVideo().getId()).add(rv.getResource()); + } + } + for (VideoInfo vi : videoList) { + if (map.get(vi.getId()) != null) { + vi.setResourceList(map.get(vi.getId())); + } + } + saveOrUpdate(videoList); } } @@ -134,12 +190,15 @@ 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); + if (video != null) { + List<ResourceVideo> rvList = resourceVideoService.getResourceList(video.getId()); + List<VideoResource> vrList = new ArrayList<>(); + for (ResourceVideo rv : rvList) + vrList.add(rv.getResource()); + video.setResourceList(vrList); + saveOrUpdate(video); + } } } -- Gitblit v1.8.0