소스 검색

增加参数指定日期

wangyunpeng 9 달 전
부모
커밋
7d97adea6c

+ 1 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/crawler/GetOffVideoCrawlerRepository.java

@@ -9,5 +9,5 @@ import java.util.List;
 @Repository
 public interface GetOffVideoCrawlerRepository extends JpaRepository<GetOffVideoCrawler, Long> {
 
-    List<GetOffVideoCrawler> getByPublishTimeGreaterThan(Long publishTime);
+    List<GetOffVideoCrawler> getByPublishTimeBetween(Long startTime, Long endTime);
 }

+ 1 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/repository/longArticle/GetOffVideoArticleRepository.java

@@ -9,6 +9,6 @@ import java.util.List;
 @Repository
 public interface GetOffVideoArticleRepository extends JpaRepository<GetOffVideoArticle, Long> {
 
-    List<GetOffVideoArticle> getByPublishTimeGreaterThan(Long publishTime);
+    List<GetOffVideoArticle> getByPublishTimeBetween(Long startTime, Long endTime);
 
 }

+ 8 - 2
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/XxlJobService.java

@@ -96,7 +96,10 @@ public class XxlJobService {
     public ReturnT<String> migrateCrawlerRootSourceId(String param) {
         try {
             long timeStamp = DateUtils.getBeforeDayStart(1);
-            List<GetOffVideoCrawler> getOffVideoCrawlerList = getOffVideoCrawlerRepository.getByPublishTimeGreaterThan(timeStamp / 1000);
+            if (StringUtils.hasText(param)) {
+                timeStamp = DateUtils.getStartOfDay(param, "yyyyMMdd");
+            }
+            List<GetOffVideoCrawler> getOffVideoCrawlerList = getOffVideoCrawlerRepository.getByPublishTimeBetween(timeStamp, timeStamp + 86400);
             List<String> traceIds = getOffVideoCrawlerList.stream().map(GetOffVideoCrawler::getTraceId).distinct().collect(Collectors.toList());
             List<LongArticlesVideo> longArticlesVideoList = longArticlesVideoRepository.getByTraceIdIn(traceIds);
             for (LongArticlesVideo longArticlesVideo : longArticlesVideoList) {
@@ -150,7 +153,10 @@ public class XxlJobService {
     public ReturnT<String> migrateArticleRootSourceId(String param) {
         try {
             long timeStamp = DateUtils.getBeforeDayStart(1);
-            List<GetOffVideoArticle> getOffVideoArticleList = getOffVideoArticleRepository.getByPublishTimeGreaterThan(timeStamp / 1000);
+            if (StringUtils.hasText(param)) {
+                timeStamp = DateUtils.getStartOfDay(param, "yyyyMMdd");
+            }
+            List<GetOffVideoArticle> getOffVideoArticleList = getOffVideoArticleRepository.getByPublishTimeBetween(timeStamp, timeStamp + 86400);
             List<String> traceIds = getOffVideoArticleList.stream().map(GetOffVideoArticle::getTraceId).distinct().collect(Collectors.toList());
             List<LongArticlesMatchVideo> longArticlesMatchVideoList = longArticlesMatchVideoRepository.getByTraceIdIn(traceIds);
             for (LongArticlesMatchVideo longArticlesMatchVideo : longArticlesMatchVideoList) {

+ 1 - 1
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/util/DateUtils.java

@@ -223,6 +223,6 @@ public final class DateUtils {
         calendar.set(Calendar.SECOND, 0);
         calendar.set(Calendar.MILLISECOND, 0);
         calendar.add(Calendar.DAY_OF_MONTH, -days);
-        return calendar.getTime().getTime();
+        return calendar.getTime().getTime() / 1000;
     }
 }

+ 4 - 4
long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/web/XxlJobController.java

@@ -20,12 +20,12 @@ public class XxlJobController {
     }
 
     @GetMapping("/migrateCrawlerRootSourceId")
-    public void migrateCrawlerRootSourceId() {
-        service.migrateCrawlerRootSourceId(null);
+    public void migrateCrawlerRootSourceId(String dateStr) {
+        service.migrateCrawlerRootSourceId(dateStr);
     }
 
     @GetMapping("/migrateArticleRootSourceId")
-    public void migrateArticleRootSourceId() {
-        service.migrateArticleRootSourceId(null);
+    public void migrateArticleRootSourceId(String dateStr) {
+        service.migrateArticleRootSourceId(dateStr);
     }
 }