媒体资源搜索

core/src/main/java/com/tzld/supply/job/ContentMediaSearchJob.java

@@ -68,7 +68,8 @@ public class ContentMediaSearchJob {
             if (mediaCount > 0) {
                 continue;
             }
-            imageMediaSearch(content.getId(), content.getTitle());
+            List<SpiderMediaItem> dataItems = spiderApiService.searchContentImage(content.getTitle());
+            imageMediaSearch(content.getId(), dataItems, "百度图片");
             if (StringUtils.isBlank(content.getKeyword())) {
                 continue;
             }
@@ -77,15 +78,15 @@ public class ContentMediaSearchJob {
                 continue;
             }
             for (String keyword : keywords) {
-                imageMediaSearch(content.getId(), keyword);
+                dataItems = spiderApiService.searchContentImage(keyword);
+                imageMediaSearch(content.getId(), dataItems, "百度图片");
             }
         }
 
         return ReturnT.SUCCESS;
     }
 
-    private void imageMediaSearch(Long contentId, String keyword) {
-        List<SpiderMediaItem> dataItems = spiderApiService.searchContentImage(keyword);
+    private void imageMediaSearch(Long contentId, List<SpiderMediaItem> dataItems, String sourceSite) {
         if (CollectionUtil.isEmpty(dataItems)) {
             return;
         }
@@ -100,7 +101,7 @@ public class ContentMediaSearchJob {
                 continue;
             }
             SpiderContentMedia media = buildMedia(contentId, SpiderContentMediaTypeEnum.IMAGE.getMsg(),
-                    "百度图片", mediaItem.getUrl(), null,
+                    sourceSite, mediaItem.getUrl(), null,
                     mediaItem.getTitle().substring(0, Math.min(mediaItem.getTitle().length(), 100)), fileUrl,
                     SpiderContentMediaStatusEnum.WAITING.getCode(), now);
             saveList.add(media);
@@ -168,14 +169,14 @@ public class ContentMediaSearchJob {
             if (mediaCount > 0) {
                 continue;
             }
-            videoMediaSearch(content.getId(), content.getTitle());
+            List<SpiderMediaItem> dataItems = spiderApiService.searchContentVideo(content.getTitle());
+            videoMediaSearch(content.getId(), dataItems, "好看视频");
         }
 
         return ReturnT.SUCCESS;
     }
 
-    private void videoMediaSearch(Long contentId, String keyword) {
-        List<SpiderMediaItem> dataItems = spiderApiService.searchContentVideo(keyword);
+    private void videoMediaSearch(Long contentId, List<SpiderMediaItem> dataItems, String sourceSite) {
         if (CollectionUtil.isEmpty(dataItems)) {
             return;
         }
@@ -196,7 +197,7 @@ public class ContentMediaSearchJob {
                 continue;
             }
             SpiderContentMedia media = buildMedia(contentId, SpiderContentMediaTypeEnum.VIDEO.getMsg(),
-                    "好看视频", mediaItem.getUrl(), mediaItem.getDuration() * 1000,
+                    sourceSite, mediaItem.getUrl(), mediaItem.getDuration() * 1000,
                     mediaItem.getTitle().substring(0, Math.min(mediaItem.getTitle().length(), 100)), fileUrl,
                     SpiderContentMediaStatusEnum.PASSED.getCode(), now);
             saveList.add(media);