score数据处理

long-article-recommend-service/src/test/java/com/tzld/longarticle/recommend/server/DataDashboardTest.java

@@ -18,7 +18,9 @@ import org.springframework.http.HttpHeaders;
 
 import javax.annotation.Resource;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
+import java.util.stream.Collectors;
 
 @SpringBootTest(classes = Application.class)
 @Slf4j
@@ -75,10 +77,16 @@ public class DataDashboardTest {
     @Test
     public void datastatScore() {
         List<DatastatScore> scoreList = datastatScoreRepository.getBySourceIdIsNotNull();
+        List<ArticleCategory> categoryList = articleCategoryRepository.getByStatusAndVersion(1, 2);
+        Map<String, ArticleCategory> categoryMap = categoryList.stream().collect(Collectors.toMap(ArticleCategory::getProduceContentId, category -> category));
+        Map<String, ArticleCategory> titleMap = categoryList.stream().collect(Collectors.toMap(ArticleCategory::getTitle, category -> category));
         for (DatastatScore score : scoreList) {
-            ArticleCategory category = articleCategoryRepository.getByProduceContentIdAndVersion(score.getSourceId(), 2);
+            ArticleCategory category = categoryMap.get(score.getSourceId());
             if (Objects.isNull(category)) {
-                continue;
+                category = titleMap.get(score.getTitle());
+                if (Objects.isNull(category)) {
+                    continue;
+                }
             }
             score.setCategory(category.getCategory());
             datastatScoreRepository.save(score);