getVideoIdByRootSourceId

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/service/recommend/ArticleService.java

@@ -17,12 +17,14 @@ import com.tzld.longarticle.recommend.server.model.entity.aigc.PublishContent;
 import com.tzld.longarticle.recommend.server.model.entity.aigc.PublishContentOutput;
 import com.tzld.longarticle.recommend.server.model.entity.crawler.Article;
 import com.tzld.longarticle.recommend.server.model.entity.longArticle.ArticlePoolPromotionSource;
+import com.tzld.longarticle.recommend.server.model.entity.longArticle.LongArticlesRootSourceId;
 import com.tzld.longarticle.recommend.server.model.param.ArticleFindSourceParam;
 import com.tzld.longarticle.recommend.server.model.vo.RootPublishContentVO;
 import com.tzld.longarticle.recommend.server.repository.aigc.PublishAccountRepository;
 import com.tzld.longarticle.recommend.server.repository.aigc.PublishContentOutputRepository;
 import com.tzld.longarticle.recommend.server.repository.crawler.ArticleRepository;
 import com.tzld.longarticle.recommend.server.repository.longArticle.ArticlePoolPromotionSourceRepository;
+import com.tzld.longarticle.recommend.server.repository.longArticle.LongArticlesRootSourceIdRepository;
 import com.tzld.longarticle.recommend.server.util.DateUtils;
 import com.tzld.longarticle.recommend.server.util.TitleSimilarCheckUtil;
 import lombok.extern.slf4j.Slf4j;
@@ -64,6 +66,8 @@ public class ArticleService {
     LongArticleBaseMapper longArticleBaseMapper;
     @Autowired
     ArticlePoolPromotionSourceRepository articlePoolPromotionSourceRepository;
+    @Autowired
+    LongArticlesRootSourceIdRepository longArticlesRootSourceIdRepository;
 
     private final static ExecutorService pool = new CommonThreadPoolExecutor(
             32,
@@ -391,4 +395,12 @@ public class ArticleService {
         });
         return result;
     }
+
+    public Long getVideoIdByRootSourceId(String rootSourceId) {
+        LongArticlesRootSourceId longArticlesRootSourceId = longArticlesRootSourceIdRepository.getByRootSourceId(rootSourceId);
+        if (Objects.isNull(longArticlesRootSourceId)) {
+            return null;
+        }
+        return longArticlesRootSourceId.getVideoId();
+    }
 }

long-article-recommend-service/src/main/java/com/tzld/longarticle/recommend/server/web/recommend/ArticleController.java

@@ -39,4 +39,9 @@ public class ArticleController {
         return CommonResponse.success(service.getCooperateArticleIdByRootSourceId(rootSourceId));
     }
 
+    @GetMapping("/getVideoIdByRootSourceId")
+    public CommonResponse<Long> getVideoIdByRootSourceId(@RequestParam String rootSourceId) {
+        return CommonResponse.success(service.getVideoIdByRootSourceId(rootSourceId));
+    }
+
 }