From f13ed98e1de0ec7b85ed179212cc095f63480eed Mon Sep 17 00:00:00 2001
From: admin <weikou2014>
Date: 星期四, 25 三月 2021 18:18:20 +0800
Subject: [PATCH] 全网搜优酷兼容

---
 src/main/java/com/yeshi/buwan/controller/parser/UserParser.java |  102 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 102 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/yeshi/buwan/controller/parser/UserParser.java b/src/main/java/com/yeshi/buwan/controller/parser/UserParser.java
index 6a71487..9bcbf7c 100644
--- a/src/main/java/com/yeshi/buwan/controller/parser/UserParser.java
+++ b/src/main/java/com/yeshi/buwan/controller/parser/UserParser.java
@@ -2,19 +2,25 @@
 
 import com.yeshi.buwan.domain.*;
 import com.yeshi.buwan.domain.Collection;
+import com.yeshi.buwan.domain.solr.SolrAlbumVideo;
 import com.yeshi.buwan.domain.system.DetailSystem;
 import com.yeshi.buwan.domain.user.LoginUser;
+import com.yeshi.buwan.domain.video.InternetSearchVideo;
 import com.yeshi.buwan.dto.log.BaseLog;
+import com.yeshi.buwan.dto.search.SolrResultDTO;
+import com.yeshi.buwan.dto.search.SolrVideoSearchFilter;
 import com.yeshi.buwan.dto.user.LoginInfoDto;
 import com.yeshi.buwan.exception.user.LoginUserException;
 import com.yeshi.buwan.exception.user.RegisterUserException;
 import com.yeshi.buwan.pptv.PPTVUtil;
 import com.yeshi.buwan.service.imp.*;
+import com.yeshi.buwan.service.inter.juhe.InternetSearchVideoService;
 import com.yeshi.buwan.service.manager.SolrAlbumVideoDataManager;
 import com.yeshi.buwan.util.*;
 import com.yeshi.buwan.util.JuHe.VideoResourceUtil;
 import com.yeshi.buwan.util.annotation.RequireUid;
 import com.yeshi.buwan.util.email.MailSenderUtil;
+import com.yeshi.buwan.util.factory.VideoInfoFactory;
 import com.yeshi.buwan.util.log.LoggerUtil;
 import com.yeshi.buwan.util.log.UserActiveLogFactory;
 import com.yeshi.buwan.util.video.VideoCategoryConstant;
@@ -449,10 +455,68 @@
     }
 
 
+    @Resource
+    private InternetSearchVideoService internetSearchVideoService;
+
+    private void guessLikeForInternetSearch(AcceptData acceptData, String videoId, PrintWriter out) {
+        InternetSearchVideo internetSearchVideo = internetSearchVideoService.selectByPrimaryKey(videoId);
+        if (internetSearchVideo != null) {
+            int rootType = internetSearchVideo.getRootType();
+            SolrVideoSearchFilter filter = new SolrVideoSearchFilter();
+            filter.setVideoType(rootType);
+            filter.setResourceIds(new String[]{PPTVUtil.RESOURCE_ID + ""});
+            filter.setSortKey("watchcount");
+            SolrResultDTO dto = solrDataManager.find(filter, 1, 20);
+            List<SolrAlbumVideo> solrAlbumVideoList = new ArrayList<>();
+            solrAlbumVideoList.addAll(dto.getVideoList());
+            List<VideoInfo> list = new ArrayList<>();
+            for (int i = 0; i < 4; i++) {
+                if (solrAlbumVideoList.size() > 0) {
+                    int p = (int) (Math.random() * solrAlbumVideoList.size());
+                    SolrAlbumVideo solrAlbumVideo = solrAlbumVideoList.get(p);
+                    list.add(VideoInfoFactory.create(solrAlbumVideo));
+                    solrAlbumVideoList.remove(p);
+                }
+            }
+
+            if (list.size() < 4) {
+                filter.setResourceIds(null);
+                dto = solrDataManager.find(filter, 1, 20);
+                solrAlbumVideoList.clear();
+                solrAlbumVideoList.addAll(dto.getVideoList());
+                for (int i = list.size(); i < 4; i++) {
+                    if (solrAlbumVideoList.size() > 0) {
+                        int p = (int) (Math.random() * solrAlbumVideoList.size());
+                        SolrAlbumVideo solrAlbumVideo = solrAlbumVideoList.get(p);
+                        list.add(VideoInfoFactory.create(solrAlbumVideo));
+                        solrAlbumVideoList.remove(p);
+                    }
+                }
+            }
+
+
+            JSONObject object = new JSONObject();
+            object.put("count", list.size() + "");
+            JSONArray array = new JSONArray();
+
+            for (int i = 0; i < list.size(); i++) {
+                array.add(StringUtil.outPutResultJson(list.get(i)));
+            }
+
+            object.put("data", array);
+            out.print(JsonUtil.loadTrueJson(object.toString()));
+        }
+
+    }
+
     // 鐚滀綘鍠滄
     public void guessLike(AcceptData acceptData, HttpServletRequest request, PrintWriter out) {
 
         String videoId = request.getParameter("VideoId");
+        if (!NumberUtil.isNumeric(videoId)) {
+            guessLikeForInternetSearch(acceptData, videoId, out);
+            return;
+        }
 
         DetailSystem ds = systemService.getDetailSystemByPackage(acceptData.getPackageName());
 
@@ -625,6 +689,39 @@
 
     }
 
+
+    private void getRelativeVideosForInternetSearch(AcceptData acceptData, String videoId, PrintWriter out) {
+        InternetSearchVideo internetSearchVideo = internetSearchVideoService.selectByPrimaryKey(videoId);
+        if (internetSearchVideo != null) {
+            SolrVideoSearchFilter filter = new SolrVideoSearchFilter();
+            filter.setKey(internetSearchVideo.getName());
+            filter.setFuzzy(true);
+            SolrResultDTO dto = solrDataManager.find(filter, 1, 20);
+            List<SolrAlbumVideo> solrAlbumVideoList = new ArrayList<>();
+            solrAlbumVideoList.addAll(dto.getVideoList());
+            List<VideoInfo> list = new ArrayList<>();
+            for (int i = 0; i < 4; i++) {
+                if (i < solrAlbumVideoList.size()) {
+                    SolrAlbumVideo solrAlbumVideo = solrAlbumVideoList.get(i);
+                    list.add(VideoInfoFactory.create(solrAlbumVideo));
+                }
+            }
+
+
+            JSONObject object = new JSONObject();
+            object.put("count", list.size() + "");
+            JSONArray array = new JSONArray();
+
+            for (int i = 0; i < list.size(); i++) {
+                array.add(StringUtil.outPutResultJson(list.get(i)));
+            }
+
+            object.put("data", array);
+            out.print(JsonUtil.loadTrueJson(object.toString()));
+        }
+
+    }
+
     /**
      * 鐩稿叧瑙嗛
      *
@@ -638,6 +735,11 @@
             return;
         }
 
+        if (!NumberUtil.isNumeric(videoId)) {
+            getRelativeVideosForInternetSearch(acceptData, videoId, out);
+            return;
+        }
+
         List<Long> resourceList = videoResouceUtil.getAvailableResourceIds(acceptData.getDetailSystem(), acceptData.getVersion());
 
         DetailSystem ds = systemService.getDetailSystemByPackage(acceptData.getPackageName());

--
Gitblit v1.8.0