From 4011b8d0b377af33e2bc435f7726329630d706cb Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期二, 02 三月 2021 18:28:52 +0800 Subject: [PATCH] 搜索引擎接口更新 --- src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java | 46 +++++++++++++++++++++++----------------------- 1 files changed, 23 insertions(+), 23 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 606760e..936f88c 100644 --- a/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java +++ b/src/main/java/com/yeshi/buwan/service/manager/SolrAlbumDataManager.java @@ -12,27 +12,27 @@ import com.yeshi.buwan.service.inter.juhe.AlbumVideoMapService; import com.yeshi.buwan.util.StringUtil; import com.yeshi.buwan.util.factory.SolrVideoFactory; -import net.sf.ehcache.search.expression.BaseCriteria; 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.*; +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.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; +import java.util.*; /** * 鎼滅储寮曟搸-涓撹緫鏁版嵁绠$悊 */ @Component public class SolrAlbumDataManager { + + private final static String CORE_NAME = "buwan_album"; private final Logger logger = LoggerFactory.getLogger(SolrAlbumDataManager.class); @@ -50,11 +50,11 @@ public void saveOrUpdate(VideoInfo videoInfo) { - UpdateResponse updateResponse = solrTemplate.saveBean(SolrVideoFactory.create(videoInfo, videoInfo.getResourceList())); + UpdateResponse updateResponse = solrTemplate.saveBean(CORE_NAME, SolrVideoFactory.create(videoInfo, videoInfo.getResourceList())); if (updateResponse.getStatus() == 0) { - solrTemplate.commit(); + solrTemplate.commit(CORE_NAME); } else { - solrTemplate.rollback(); + solrTemplate.rollback(CORE_NAME); } } @@ -63,11 +63,11 @@ for (VideoInfo vi : videoInfoList) { solrVideoList.add(SolrVideoFactory.create(vi, vi.getResourceList())); } - UpdateResponse updateResponse = solrTemplate.saveBeans(solrVideoList); + UpdateResponse updateResponse = solrTemplate.saveBeans(CORE_NAME, solrVideoList); if (updateResponse.getStatus() == 0) { - solrTemplate.commit(); + solrTemplate.commit(CORE_NAME); } else { - solrTemplate.rollback(); + solrTemplate.rollback(CORE_NAME); } } @@ -84,17 +84,17 @@ // } // } public void deleteById(String id) { - UpdateResponse updateResponse = solrTemplate.deleteById(id); + UpdateResponse updateResponse = solrTemplate.deleteByIds(CORE_NAME, id); if (updateResponse.getStatus() == 0) { - solrTemplate.commit(); + solrTemplate.commit(CORE_NAME); } else { - solrTemplate.rollback(); + solrTemplate.rollback(CORE_NAME); } } public SolrVideo findOne(String id) { - SolrVideo solrVideo = solrTemplate.getById(id, SolrVideo.class); - return solrVideo; + Optional<SolrVideo> solrVideo = solrTemplate.getById(CORE_NAME, id, SolrVideo.class); + return solrVideo.get(); } public SolrResultDTO find(SolrVideoSearchFilter filter, int page, int pageSize) { @@ -116,11 +116,11 @@ } if (filter.getActor() != null) { - criteria = criteria.and("mainactor").expression("\""+filter.getActor()+"\""); + criteria = criteria.and("mainactor").expression("\"" + filter.getActor() + "\""); } if (filter.getDirector() != null) { - criteria = criteria.and("director").expression("\""+filter.getDirector()+"\""); + criteria = criteria.and("director").expression("\"" + filter.getDirector() + "\""); } if (filter.getResourceIds() != null && filter.getResourceIds().length > 0) { @@ -133,7 +133,7 @@ query.addCriteria(criteria); /** 璁剧疆鍒嗛〉寮�濮嬭褰曟暟(绗竴椤�) 榛樿0 */ - query.setOffset((page - 1) * pageSize); + query.setOffset((page - 1) * pageSize * 1L); /** 璁剧疆姣忛〉鏄剧ず璁板綍鏁帮紝榛樿10 */ query.setRows(pageSize); if (StringUtil.isNullOrEmpty(filter.getSortKey())) { @@ -141,7 +141,7 @@ } else { query.addSort(new Sort(Sort.Direction.DESC, filter.getSortKey())); } - ScoredPage<SolrVideo> result = solrTemplate.queryForPage(query, SolrVideo.class); + ScoredPage<SolrVideo> result = solrTemplate.queryForPage(CORE_NAME, query, SolrVideo.class); System.out.println("鎬昏褰曟暟锛�" + result.getTotalElements()); List<SolrVideo> list = result.getContent(); return new SolrResultDTO(list, (int) result.getTotalElements()); @@ -150,10 +150,10 @@ public List<String> getSuggestKeyList(String key) { Query query = new SimpleQuery("name:\"" + key + "\""); /** 璁剧疆鍒嗛〉寮�濮嬭褰曟暟(绗竴椤�) 榛樿0 */ - query.setOffset(0); + query.setOffset(0L); /** 璁剧疆姣忛〉鏄剧ず璁板綍鏁帮紝榛樿10 */ query.setRows(10); - ScoredPage<SolrVideo> result = solrTemplate.queryForPage(query, SolrVideo.class); + ScoredPage<SolrVideo> result = solrTemplate.queryForPage(CORE_NAME, query, SolrVideo.class); System.out.println("鎬昏褰曟暟锛�" + result.getTotalElements()); List<SolrVideo> list = result.getContent(); List<String> keyList = new ArrayList<>(); -- Gitblit v1.8.0