xieb
2024-01-09 d0c4197a12052490df33d9882dad00d9d64f901e
src/main/java/org/springblade/modules/evaluate/wrapper/EvaluateTaskWrapper.java
@@ -24,8 +24,12 @@
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.evaluate.dto.EvaluateTaskDTO;
import org.springblade.modules.evaluate.entity.EvaluateCandidateResultEntity;
import org.springblade.modules.evaluate.entity.EvaluateTaskCategoryEntity;
import org.springblade.modules.evaluate.entity.EvaluateTaskEntity;
import org.springblade.modules.evaluate.entity.EvaluateTaskReferrerEntity;
import org.springblade.modules.evaluate.service.IEvaluateCandidateResultService;
import org.springblade.modules.evaluate.service.IEvaluateTaskCategoryService;
import org.springblade.modules.evaluate.service.IEvaluateTaskReferrerService;
import org.springblade.modules.evaluate.vo.EvaluateTaskVO;
@@ -42,8 +46,14 @@
   private static final IEvaluateTaskReferrerService evaluateTaskReferrerService;
   private static final IEvaluateTaskCategoryService evaluateTaskCategoryService;
   private static final IEvaluateCandidateResultService evaluateCandidateResultService;
   static {
      evaluateTaskReferrerService = SpringUtil.getBean(IEvaluateTaskReferrerService.class);
      evaluateTaskCategoryService = SpringUtil.getBean(IEvaluateTaskCategoryService.class);
      evaluateCandidateResultService = SpringUtil.getBean(IEvaluateCandidateResultService.class);
   }
   public static EvaluateTaskWrapper build() {
@@ -67,7 +77,12 @@
      return dto;
   }
   public IPage<EvaluateTaskVO> h5PageVO(IPage<EvaluateTaskEntity> pages) {
   /**
    * h5返回自己第一轮投票的人
    * @param pages
    * @return
    */
   public IPage<EvaluateTaskVO> h5ReferremPageVO(IPage<EvaluateTaskEntity> pages) {
      List<EvaluateTaskVO> records = listVO(pages.getRecords());
      for (EvaluateTaskVO vo: records) {
@@ -86,5 +101,35 @@
      return pageVo;
   }
   /**
    * 第二轮候选人投票
    * @return
    */
   public IPage<EvaluateTaskVO> h5CandidatePageVO(IPage<EvaluateTaskEntity> pages) {
      List<EvaluateTaskVO> records = listVO(pages.getRecords());
      for (EvaluateTaskVO vo: records) {
         QueryWrapper queryWrapper = new QueryWrapper();
         queryWrapper.eq("create_user", AuthUtil.getUserId());
         queryWrapper.eq("evaluate_task_id", vo.getId());
         List<EvaluateCandidateResultEntity> entities = evaluateCandidateResultService.list(queryWrapper);
         QueryWrapper qw = new QueryWrapper();
         qw.eq("evaluate_task_id", vo.getId());
         List<EvaluateTaskCategoryEntity> evaluateTaskCategoryEntities = evaluateTaskCategoryService.list(qw);
         int evaluateCount = evaluateTaskCategoryEntities.stream().mapToInt(EvaluateTaskCategoryEntity::getPeopleNum).sum(); // 计算总数
         vo.setIsVote(entities.size() == evaluateCount);
         if (entities.size() > 0)
            vo.setEvaluateCandidateResultVOS(EvaluateCandidateResultWrapper.build().listVO(entities));
      }
      IPage<EvaluateTaskVO> pageVo = new Page<>(pages.getCurrent(), pages.getSize(), pages.getTotal());
      pageVo.setRecords(records);
      return pageVo;
   }
}