xieb
2023-12-14 63a4dd5239676ff135d05bfa31362a3fb36af3c1
src/main/java/org/springblade/modules/assessment/wrapper/AssessmentScoreWrapper.java
@@ -16,11 +16,22 @@
 */
package org.springblade.modules.assessment.wrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import org.springblade.core.mp.support.BaseEntityWrapper;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.core.tool.utils.SpringUtil;
import org.springblade.modules.assessment.entity.AssessmentScoreEntity;
import org.springblade.modules.assessment.entity.AssessmentSetDeptEntity;
import org.springblade.modules.assessment.entity.AssessmentSetEntity;
import org.springblade.modules.assessment.service.IAssessmentScoreService;
import org.springblade.modules.assessment.service.IAssessmentSetDeptService;
import org.springblade.modules.assessment.service.IAssessmentSetService;
import org.springblade.modules.assessment.service.IAssessmentTaskService;
import org.springblade.modules.assessment.vo.AssessmentScoreVO;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;
/**
 * 考核评分 包装类,返回视图层所需的字段
@@ -29,6 +40,18 @@
 * @since 2023-12-13
 */
public class AssessmentScoreWrapper extends BaseEntityWrapper<AssessmentScoreEntity, AssessmentScoreVO>  {
   private static final IAssessmentSetService assessmentSetService;
   private static final IAssessmentSetDeptService assessmentSetDeptService;
   private static final IAssessmentScoreService assessmentScoreService;
   private static final IAssessmentTaskService assessmentTaskService;
   static {
      assessmentSetService = SpringUtil.getBean(IAssessmentSetService.class);
      assessmentSetDeptService = SpringUtil.getBean(IAssessmentSetDeptService.class);
      assessmentScoreService = SpringUtil.getBean(IAssessmentScoreService.class);
      assessmentTaskService = SpringUtil.getBean(IAssessmentTaskService.class);
   }
   public static AssessmentScoreWrapper build() {
      return new AssessmentScoreWrapper();
@@ -43,8 +66,25 @@
      //assessmentScoreVO.setCreateUserName(createUser.getName());
      //assessmentScoreVO.setUpdateUserName(updateUser.getName());
      //考核数量
      QueryWrapper<AssessmentSetEntity> AssessmentSetEntityWrapper = new QueryWrapper<>();
      AssessmentSetEntityWrapper.eq("user_id",assessmentScoreVO.getBeId());
      long userAssCount = assessmentSetService.count(AssessmentSetEntityWrapper);
      //已经考核数量
      QueryWrapper<AssessmentScoreEntity> wrapper = new QueryWrapper<>();
      wrapper.eq("be_id",assessmentScoreVO.getBeId());
      long assCount = assessmentScoreService.count(wrapper);
      assessmentScoreVO.setIsAssessmentOk(userAssCount == assCount);
      assessmentScoreVO.setAssessmentTaskVO(AssessmentTaskWrapper.build().entityVO(assessmentTaskService.getById(assessmentScoreVO.getAssessmentTaskId())));
      return assessmentScoreVO;
   }
   @Override
   public List<AssessmentScoreVO> listVO(List<AssessmentScoreEntity> list) {
      return list.stream().map(this::entityVO).collect(Collectors.toList());
   }
}