是否完成考核

xieb

2023-12-15 7dd9c5ee5fa54e1637c545aa92bb7303b778323f

 src/main/java/org/springblade/modules/assessment/service/impl/AssessmentScoreServiceImpl.java

@@ -16,15 +16,26 @@
 */
package org.springblade.modules.assessment.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import lombok.AllArgsConstructor;
import org.springblade.modules.assessment.entity.AssessmentScoreEntity;
import org.springblade.modules.assessment.entity.AssessmentSetEntity;
import org.springblade.modules.assessment.entity.AssessmentTaskEntity;
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 org.springblade.modules.assessment.excel.AssessmentScoreExcel;
import org.springblade.modules.assessment.mapper.AssessmentScoreMapper;
import org.springblade.modules.assessment.service.IAssessmentScoreService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import org.springblade.core.mp.base.BaseServiceImpl;
import org.springframework.transaction.annotation.Transactional;

import java.util.Collection;
import java.util.List;

/**
@@ -34,7 +45,14 @@
 * @since 2023-12-13
 */
@Service
@AllArgsConstructor
public class AssessmentScoreServiceImpl extends BaseServiceImpl<AssessmentScoreMapper, AssessmentScoreEntity> implements IAssessmentScoreService {

   private final IAssessmentSetService assessmentSetService;
   private final IAssessmentSetDeptService assessmentSetDeptService;

   private final IAssessmentTaskService assessmentTaskService;


   @Override
   public IPage<AssessmentScoreVO> selectAssessmentScorePage(IPage<AssessmentScoreVO> page, AssessmentScoreVO assessmentScore) {
@@ -51,4 +69,31 @@
      return assessmentScoreList;
   }

   @Override
   @Transactional
   public boolean saveOrUpdate(AssessmentScoreEntity entity) {
      boolean ret = super.saveOrUpdate(entity);

      long setCount = 0l;
      if (entity.getType() == 0) {
         setCount = assessmentSetService.count();

      } else {
         setCount = assessmentSetDeptService.count();
      }

      QueryWrapper<AssessmentScoreEntity> queryWrapper = new QueryWrapper<>();
      queryWrapper.eq("type",entity.getType());
      queryWrapper.eq("assessment_task_id",entity.getAssessmentTaskId());
      long scoreCount = count(queryWrapper);

      if (setCount == scoreCount) {
         AssessmentTaskEntity taskEntity = new AssessmentTaskEntity();
         taskEntity.setId(entity.getAssessmentTaskId());
         taskEntity.setIsAssessmentOk(1);
         assessmentTaskService.updateById(taskEntity);
      }

      return ret;
   }
}