From 7dd9c5ee5fa54e1637c545aa92bb7303b778323f Mon Sep 17 00:00:00 2001
From: xieb <vip_xiaobin810@163.com>
Date: Fri, 15 Dec 2023 14:58:24 +0800
Subject: [PATCH] 是否完成考核

---
 src/main/java/org/springblade/modules/assessment/service/impl/AssessmentScoreServiceImpl.java |   45 +++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 45 insertions(+), 0 deletions(-)

diff --git a/src/main/java/org/springblade/modules/assessment/service/impl/AssessmentScoreServiceImpl.java b/src/main/java/org/springblade/modules/assessment/service/impl/AssessmentScoreServiceImpl.java
index c34f461..0ae4068 100644
--- a/src/main/java/org/springblade/modules/assessment/service/impl/AssessmentScoreServiceImpl.java
+++ b/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;
+	}
 }

--
Gitblit v1.9.3