修复课程列表推荐问题

community-service/src/main/java/com/tanpu/community/service/HotRecommendService.java

@@ -22,12 +22,17 @@ import com.tanpu.community.dao.mapper.community.HotRecommendMapper;
 import com.tanpu.community.dao.mapper.community.HotRecommendUserMapper;
 import com.tanpu.community.feign.fatools.FeignClientForFatools;
 import com.tanpu.community.util.ConvertUtil;
+import com.tanpu.community.util.HttpServletHelper;
 import org.apache.commons.collections4.CollectionUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
+import javax.validation.constraints.NotEmpty;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 import java.util.stream.Collectors;
 
 /**
@@ -44,6 +49,12 @@ public class HotRecommendService {
 
     @Resource
     private FeignClientForFatools feignClientForFatools;
+    @Resource
+    private FollowRelService followRelService;
+
+    @Resource
+    private HttpServletHelper httpServletHelper;
+
 
     /**
      * 首页热门推荐列表
@@ -84,15 +95,29 @@ public class HotRecommendService {
         LambdaQueryWrapper<HotRecommend> queryWrapper = new LambdaQueryWrapper<>();
         queryWrapper.eq(HotRecommend::getHotRecommendId, req.getHotRecommendId());
         HotRecommend hotRecommend = hotRecommendMapper.selectOne(queryWrapper);
-        List<FollowQo> list = null;
+        Page<FollowQo> page = null;
         if (hotRecommend.getType() == 1) {
-            return this.getChiefInvAdv(req);
+            page = this.getChiefInvAdv(req);
         } else if (hotRecommend.getType() == 2) {
-            return this.getCorpInfo(req);
+            page = this.getCorpInfo(req);
         } else if (hotRecommend.getType() == 3) {
-            return this.getUserInfo(req);
+            page = this.getUserInfo(req);
+        }
+        if (StringUtils.isNotEmpty(httpServletHelper.getCurrentUserId())) {
+            page.setContent(judgeFollowed(page.getContent(), httpServletHelper.getCurrentUserId()));
         }
-        return null;
+        return page;
+    }
+
+    //判断返回列表中的用户是否被当前用户关注
+    public List<FollowQo> judgeFollowed(List<FollowQo> followQos, @NotEmpty String followerId) {
+        Set<String> idolSet = new HashSet<>(followRelService.queryIdolsByFansId(followerId));
+        return followQos.stream().map(o -> {
+            if (idolSet.contains(o.getUserId())) {
+                o.setFollowed(true);
+            }
+            return o;
+        }).collect(Collectors.toList());
     }
 
     private Page<FollowQo> getCorpInfo(UserInfoCorpReq req) {