wangyunpeng 11 місяців тому
батько
коміт
41ce149e1c

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

@@ -1,8 +1,10 @@
 package com.tzld.longarticle.recommend.server.service;
 
+import com.tzld.longarticle.recommend.server.model.ArticleSortResponseData;
 import com.tzld.longarticle.recommend.server.model.Content;
 import com.tzld.longarticle.recommend.server.model.RecommendRequest;
 import com.tzld.longarticle.recommend.server.model.RecommendResponse;
+import com.tzld.longarticle.recommend.server.service.rank.RankItem;
 import com.tzld.longarticle.recommend.server.service.rank.RankParam;
 import com.tzld.longarticle.recommend.server.service.rank.RankResult;
 import com.tzld.longarticle.recommend.server.service.rank.RankService;
@@ -18,6 +20,7 @@ import org.springframework.stereotype.Service;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.Objects;
+import java.util.stream.Collectors;
 
 /**
  * @author dyp
@@ -40,8 +43,14 @@ public class RecommendService {
         RankResult rankResult = rankService.rank(convertToRankParam(param, recallResult));
         // MergeResult mergeResult = mergeService.merge(convertToMergeParam(param, rankResult));
 
-        RecommendResponse response = new RecommendResponse();
+        List<Content> contentList = rankResult.getContents().stream().map(RankItem::getContent).collect(Collectors.toList());
+        ArticleSortResponseData data = new ArticleSortResponseData();
+        data.setRank_list(contentList);
 
+        RecommendResponse response = new RecommendResponse();
+        response.setCode(0);
+        response.setMsg("success");
+        response.setData(data);
         return response;
     }