|
@@ -12,10 +12,7 @@ import lombok.extern.slf4j.Slf4j;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
|
|
|
-import java.util.ArrayList;
|
|
|
-import java.util.Collections;
|
|
|
-import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
+import java.util.*;
|
|
|
|
|
|
/**
|
|
|
* @author dyp
|
|
@@ -71,7 +68,10 @@ public class FwhColdStartRankStrategy implements RankStrategy {
|
|
|
|
|
|
// 3 选文章
|
|
|
List<Content> result = new ArrayList<>();
|
|
|
- int size = param.getSize() * param.getUserGroupIds().size();
|
|
|
+ int size = param.getSize();
|
|
|
+ if (Objects.nonNull(param.getUserGroupIds())) {
|
|
|
+ size = param.getSize() * param.getUserGroupIds().size();
|
|
|
+ }
|
|
|
result.addAll(contents.subList(0, Math.min(contents.size(), size)));
|
|
|
|
|
|
return new RankResult(result);
|