|
@@ -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;
|
|
|
}
|
|
|
|