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/imp/SearchService.java |  149 ++++++++++++++++++++++++++-----------------------
 1 files changed, 80 insertions(+), 69 deletions(-)

diff --git a/src/main/java/com/yeshi/buwan/service/imp/SearchService.java b/src/main/java/com/yeshi/buwan/service/imp/SearchService.java
index 876d12e..7e0d0b3 100644
--- a/src/main/java/com/yeshi/buwan/service/imp/SearchService.java
+++ b/src/main/java/com/yeshi/buwan/service/imp/SearchService.java
@@ -1,32 +1,27 @@
 package com.yeshi.buwan.service.imp;
 
-import java.io.Serializable;
-import java.io.UnsupportedEncodingException;
-import java.util.*;
-
-import javax.annotation.Resource;
-
-import com.yeshi.buwan.dao.video.AlbumVideoMapDao;
+import com.yeshi.buwan.dao.*;
 import com.yeshi.buwan.domain.*;
-import com.yeshi.buwan.domain.video.AlbumVideoMap;
+import com.yeshi.buwan.domain.web.DetailSystemSelect;
+import com.yeshi.buwan.domain.web.HotSearchAdmin;
+import com.yeshi.buwan.dto.search.SolrResultDTO;
+import com.yeshi.buwan.dto.search.SolrVideoSearchFilter;
+import com.yeshi.buwan.service.inter.VideoResourceMapExtraInfoService;
 import com.yeshi.buwan.service.manager.SolrAlbumDataManager;
 import com.yeshi.buwan.util.*;
 import com.yeshi.buwan.util.factory.VideoInfoFactory;
 import com.yeshi.buwan.util.video.VideoConstant;
+import com.yeshi.buwan.vo.video.VideoListResultVO;
 import org.hibernate.HibernateException;
 import org.hibernate.Session;
 import org.springframework.cache.annotation.Cacheable;
 import org.springframework.orm.hibernate4.HibernateCallback;
 import org.springframework.stereotype.Service;
 
-import com.yeshi.buwan.dao.DetailSystemDao;
-import com.yeshi.buwan.dao.HotSearchDao;
-import com.yeshi.buwan.dao.SearchDao;
-import com.yeshi.buwan.dao.SuperHotSearchDao;
-import com.yeshi.buwan.dao.VideoBanQuanVideoDao;
-import com.yeshi.buwan.dao.VideoInfoDao;
-import com.yeshi.buwan.domain.web.DetailSystemSelect;
-import com.yeshi.buwan.domain.web.HotSearchAdmin;
+import javax.annotation.Resource;
+import java.io.Serializable;
+import java.io.UnsupportedEncodingException;
+import java.util.*;
 
 @Service
 public class SearchService {
@@ -41,22 +36,20 @@
     @Resource
     private SuperHotSearchDao superHotSearchDao;
     @Resource
-    private ConfigService configService;
-    @Resource
     private VideoBanQuanVideoDao videoBanQuanVideoDao;
     @Resource
     private DetailSystemDao detailSystemDao;
     @Resource
-    private AlbumVideoMapDao albumVideoMapDao;
+    private SolrAlbumDataManager solrAlbumDataManager;
 
     @Resource
-    private SolrAlbumDataManager solrAlbumDataManager;
+    private VideoResourceMapExtraInfoService videoResourceMapExtraInfoService;
 
     @SuppressWarnings("rawtypes")
     @Cacheable(value = "userCache", key = "'suggestSearch'+'-'+#key+'-'+#system")
     public List<String> suggestSearch(String key, String system) {
         if (StringUtil.isNullOrEmpty(key) || key.startsWith("%"))
-            return new ArrayList<String>();
+            return new ArrayList<>();
 
         List<String> list;
         Session session = null;
@@ -67,7 +60,7 @@
             sql = "select name as result from wk_video_video where name like ?";
         }
 
-        list = new ArrayList<String>();
+        list = new ArrayList<>();
 
         List li = null;
         try {
@@ -93,7 +86,7 @@
         }
 
         Iterator<String> it = set.iterator();
-        list = new ArrayList<String>();
+        list = new ArrayList<>();
         while (it.hasNext()) {
             list.add(it.next());
             if (list.size() > 10)
@@ -135,7 +128,7 @@
                     .append(") order by REPLACE(v.name,?,'') ,v.watchCount desc").toString();
         else
             sql = "from VideoInfo v where v.show='1' and (v.name like ?) order by REPLACE(v.name,?,''),v.watchCount desc";// 鏀逛负sql
-        List<Serializable> list = new ArrayList<Serializable>();
+        List<Serializable> list = new ArrayList<>();
         BaiduVideoUtil util = new BaiduVideoUtil();
         List<VideoInfo> localList = videoInfoDao.list(sql, (page - 1) * Constant.pageCount, Constant.pageCount,
                 new String[]{(new StringBuilder("")).append(key).append("%").toString(), key});
@@ -196,7 +189,7 @@
         searchDao.create(sh);
         // 鏌ヨ鏁版嵁搴�
 
-        List<VideoInfo> list = new ArrayList<VideoInfo>();
+        List<VideoInfo> list = new ArrayList<>();
         List<VideoInfo> localList = null;
         try {
             long startt = System.currentTimeMillis();
@@ -220,7 +213,7 @@
                 sql = sql.substring(0, sql.length() - 9);
 
             List rlist = videoInfoDao.sqlList(sql);
-            List<Integer> pList = new ArrayList<Integer>();
+            List<Integer> pList = new ArrayList<>();
             for (int i = 0; i < localList.size(); i++) {
                 if (Integer.parseInt(rlist.get(i) + "") < 1) {
                     pList.add(i);
@@ -250,8 +243,8 @@
 
 
     @Cacheable(value = "userCache", key = "'searchNew'+'-'+#detailSystem+'-'+#key+'-'+#page+'-'+#videoType+'-'+#system+'-'+#cacheMD5")
-    public List<VideoInfo> searchNew(String detailSystem, String ip, String uid, String key, int page, int videoType,
-                                     String system, List<Long> resourceList, String cacheMD5) {
+    public VideoListResultVO searchNew(String detailSystem, String ip, String uid, String key, int page, int videoType,
+                                       String system, List<Long> resourceList, String cacheMD5) {
         String resourceWhere = "";
         for (Long re : resourceList) {
             resourceWhere += " rv.resourceid=" + re + " or";
@@ -273,7 +266,9 @@
         searchDao.create(sh);
         // 鏌ヨ鏁版嵁搴�
 
-        List<VideoInfo> list = new ArrayList<VideoInfo>();
+        //涓撹緫鏁伴噺
+        long albumCount = 0L;
+        List<VideoInfo> list = new ArrayList<>();
         List<VideoInfo> localList = null;
         try {
             long startt = System.currentTimeMillis();
@@ -281,43 +276,56 @@
 
             localList = new ArrayList<>();
             //鍏堟悳绱笓杈�
-            List<SolrVideo> solrVideoList = solrAlbumDataManager.findByKey(key, 1, videoType == 0 ? null : videoType, page);
+            int pageSize = 20;
+            //涓撹緫瑙嗛闆嗗悎
             Set<String> solrAlbumVids = new HashSet<>();
-            for (SolrVideo sv : solrVideoList) {
-                VideoInfo video = VideoInfoFactory.create(sv);
-                video.setShowType(1);
-                if (!StringUtil.isNullOrEmpty(video.getMainActor())) {
-                    video.setMainActor("涓绘紨锛�" + video.getMainActor());
-                }
 
-                //璁剧疆tag
-                String tag = "";
-                if (!StringUtil.isNullOrEmpty(sv.getYear())) {
-                    tag += sv.getYear() + "/";
-                }
+            SolrVideoSearchFilter filter = new SolrVideoSearchFilter();
+            filter.setKey(key);
+            filter.setVideoType(videoType == 0 ? null : videoType);
+            filter.setContentType(1);
 
-                if (video.getVideoType() != null) {
-                    String vt = VideoConstant.getMainCategoryName(sv.getRootVideoType());
-                    if (!StringUtil.isNullOrEmpty(vt)) {
-                        tag += vt + "/";
+            SolrResultDTO solrResultDTO = solrAlbumDataManager.find(filter, page, pageSize);
+            if (solrResultDTO != null) {
+                albumCount = solrResultDTO.getTotalCount();
+
+                for (SolrVideo sv : solrResultDTO.getVideoList()) {
+                    VideoInfo video = VideoInfoFactory.create(sv);
+                    video.setShowType(1);
+                    if (!StringUtil.isNullOrEmpty(video.getMainActor())) {
+                        video.setMainActor("涓绘紨锛�" + video.getMainActor());
                     }
+
+                    //璁剧疆tag
+                    String tag = "";
+                    if (!StringUtil.isNullOrEmpty(sv.getYear())) {
+                        tag += sv.getYear() + "/";
+                    }
+
+                    if (video.getVideoType() != null) {
+                        String vt = VideoConstant.getMainCategoryName(sv.getRootVideoType());
+                        if (!StringUtil.isNullOrEmpty(vt)) {
+                            tag += vt + "/";
+                        }
+                    }
+
+                    if (!StringUtil.isNullOrEmpty(video.getArea())) {
+                        tag += video.getArea() + "/";
+                    }
+
+                    if (tag.endsWith("/"))
+                        tag = tag.substring(0, tag.length() - 1);
+                    video.setTag(tag);
+
+                    localList.add(video);
+                    solrAlbumVids.add(video.getId());
                 }
-
-                if (!StringUtil.isNullOrEmpty(video.getArea())) {
-                    tag += video.getArea() + "/";
-                }
-
-                if (tag.endsWith("/"))
-                    tag = tag.substring(0, tag.length() - 1);
-                video.setTag(tag);
-
-                localList.add(video);
-                solrAlbumVids.add(video.getId());
             }
+            int commonSolrPage = page - solrResultDTO.getTotalCount() / pageSize;
 
-            if (videoType == 0) {
+            if (videoType == 0 && commonSolrPage > 0) {
                 //鎼滅储鍘熷鐨�
-                List<VideoInfo> solrList = SolrUtil.search(key, page);
+                List<VideoInfo> solrList = SolrUtil.search(key, commonSolrPage);
                 for (VideoInfo vi : solrList) {
                     if (!solrAlbumVids.contains(vi.getId())) {
                         localList.add(vi);
@@ -336,8 +344,8 @@
             if (sql.endsWith("union all"))
                 sql = sql.substring(0, sql.length() - 9);
 
-            List rlist = videoInfoDao.sqlList(sql);
-            List<Integer> pList = new ArrayList<Integer>();
+            List rlist = StringUtil.isNullOrEmpty(sql) ? new ArrayList() : videoInfoDao.sqlList(sql);
+            List<Integer> pList = new ArrayList<>();
             for (int i = 0; i < localList.size(); i++) {
                 if (Integer.parseInt(rlist.get(i) + "") < 1) {
                     pList.add(i);
@@ -363,7 +371,10 @@
             }
         }
 
-        return list;
+        //鑾峰彇闄勫姞淇℃伅
+        list = videoResourceMapExtraInfoService.batchExtraInfo(list, resourceList);
+
+        return new VideoListResultVO(list, videoType == 0 ? albumCount : 1000L);
     }
 
     public long searchCount(String key, String system) {
@@ -382,7 +393,7 @@
     @SuppressWarnings({"rawtypes"})
     @Cacheable(value = "userCache", key = "'getHotSearchList'+'-'+#system")
     public List<String> getHotSearchList(String system) {
-        List<String> list = new ArrayList<String>();
+        List<String> list = new ArrayList<>();
         try {
             List li = searchDao.sqlList(
                     "SELECT h.`name` FROM wk_video_super_hotsearch sh LEFT JOIN wk_video_hotsearch h ON sh.`hotsearchid`=h.`id`  WHERE sh.`detailsystem`=? ORDER BY h.`orderby` DESC",
@@ -455,16 +466,16 @@
     }
 
     @SuppressWarnings("unchecked")
-    public List<HotSearchAdmin> getHotSearchAdmin(String key, int detailSystem, int page) {
-        List<HotSearchAdmin> zhiBoClassList = new ArrayList<HotSearchAdmin>();
+    public List<HotSearchAdmin> getHotSearchAdmin(String key, String systemId, int detailSystem, int page) {
+        List<HotSearchAdmin> zhiBoClassList = new ArrayList<>();
         try {
-            List<DetailSystem> detailSystemList = detailSystemDao.list("from DetailSystem");
+            List<DetailSystem> detailSystemList = detailSystemDao.list("from DetailSystem ds where ds.system.id=" + systemId);
             String sql = "";
             if (detailSystem > 0)
                 sql = "select sh.hotSearch from SuperHotSearch sh where sh.hotSearch.name like ? and sh.detailSystem.id="
                         + detailSystem + " order by  sh.createtime desc";
             else
-                sql = "from HotSearch zb where zb.name like ? order by zb.createtime desc";
+                sql = "from HotSearch zb where zb.name like ? and zb.system.id=" + systemId + " order by zb.createtime desc";
 
             List<HotSearch> list = hotSearchDao.list(sql, (page - 1) * Constant.pageCount, Constant.pageCount,
                     new Serializable[]{"%" + key + "%"});
@@ -472,7 +483,7 @@
                 List<DetailSystem> detailSystemS = detailSystemDao
                         .list("select vb.detailSystem from SuperHotSearch vb where vb.hotSearch.id=" + vb.getId());
 
-                List<DetailSystemSelect> dssList = new ArrayList<DetailSystemSelect>();
+                List<DetailSystemSelect> dssList = new ArrayList<>();
 
                 for (DetailSystem ds : detailSystemList) {
                     DetailSystemSelect dss = new DetailSystemSelect();
@@ -502,13 +513,13 @@
         return zhiBoClassList;
     }
 
-    public long getHotSearchAdminCount(String key, int detailSystem) {
+    public long getHotSearchAdminCount(String key,String systemId, int detailSystem) {
         String sql = "";
         if (detailSystem > 0)
             sql = "select count(*) from  (select count(*) from wk_video_super_hotsearch zb left join wk_video_hotsearch c on c.id=zb.hotsearchid where zb.detailsystem="
                     + detailSystem + " and c.name like '%" + key + "%' group by zb.hotsearchid) s";
         else
-            sql = "select count(*) from wk_video_hotsearch h where h.name like '%" + key + "%'";
+            sql = "select count(*) from wk_video_hotsearch h where h.name like '%" + key + "%' and h.system="+systemId;
 
         return videoBanQuanVideoDao.getCountSQL(sql);
     }

--
Gitblit v1.8.0