From 66437129579d47f19b0085d9324ce6a80ec207c9 Mon Sep 17 00:00:00 2001
From: "Lihong@123456" <1090591095@qq.com>
Date: Fri, 30 Oct 2020 11:16:34 +0800
Subject: [PATCH] 10-30

---
 .../ccsens/mt/service/ProvinceService.java    | 33 ++++++++++---------
 .../main/java/com/ccsens/util/CodeEnum.java   |  4 ++-
 2 files changed, 21 insertions(+), 16 deletions(-)

diff --git a/mt/src/main/java/com/ccsens/mt/service/ProvinceService.java b/mt/src/main/java/com/ccsens/mt/service/ProvinceService.java
index f702d402..3d270ea9 100644
--- a/mt/src/main/java/com/ccsens/mt/service/ProvinceService.java
+++ b/mt/src/main/java/com/ccsens/mt/service/ProvinceService.java
@@ -438,22 +438,25 @@ public class ProvinceService implements IProvinceService {
 
     @Override
     public void updateLeader(ProvinceCompeteDto.ModifyLeader param) {
-        Long coachId = param.getCoachId();
-//        Long companyId = param.getCompanyId();
-//        CompeteCoachExample competeCoachExample = new CompeteCoachExample();
-//        competeCoachExample.createCriteria().andCompanyIdEqualTo(companyId);
-//        List<CompeteCoach> competeCoachList = competeCoachMapper.selectByExample(competeCoachExample);
-        if(ObjectUtil.isNull(coachId)){
-            CompeteCoach competeCoach=new CompeteCoach();
-            competeCoach.setId(snowflake.nextId());
-            BeanUtil.copyProperties(param,competeCoach);
-            competeCoachMapper.insertSelective(competeCoach);
-        }else{
-            CompeteCoach competeCoach=competeCoachMapper.selectByPrimaryKey(coachId);
-            if(ObjectUtil.isNotNull(competeCoach)){
-                BeanUtil.copyProperties(param,competeCoach);
-                competeCoachMapper.updateByPrimaryKeySelective(competeCoach);
+        if(ObjectUtil.isNull(param.getCoachId())){
+            CompeteCoachExample competeCoachExample = new CompeteCoachExample();
+            competeCoachExample.createCriteria().andCompanyIdEqualTo(param.getCompanyId()).andIdentityEqualTo((byte) param.getIdentity());
+            long count = competeCoachMapper.countByExample(competeCoachExample);
+            if (param.getIdentity() == 0 && count >= 2){
+                throw  new BaseException(CodeEnum.COACH_NUM_FULL);
+            }
+            if (param.getIdentity() == 1 && count >=3){
+                throw new BaseException(CodeEnum.COACH_NUM_FULL);
             }
+            CompeteCoach competeCoach1 = new CompeteCoach();
+            competeCoach1.setId(snowflake.nextId());
+            BeanUtil.copyProperties(param, competeCoach1);
+            competeCoachMapper.insertSelective(competeCoach1);
+        }else {
+            CompeteCoach competeCoach2 = new CompeteCoach();
+            competeCoach2.setId(snowflake.nextId());
+            BeanUtil.copyProperties(param, competeCoach2);
+            competeCoachMapper.insertSelective(competeCoach2);
         }
     }
 
diff --git a/util/src/main/java/com/ccsens/util/CodeEnum.java b/util/src/main/java/com/ccsens/util/CodeEnum.java
index 880a3c59..98ab9657 100644
--- a/util/src/main/java/com/ccsens/util/CodeEnum.java
+++ b/util/src/main/java/com/ccsens/util/CodeEnum.java
@@ -162,7 +162,9 @@ public enum CodeEnum {
     DINGDING_EXCEPTION(132,"钉钉接口调用异常。",true),
     URL_ERROR(133, "请求路径转换异常", true),
     JOIN_PROJECT_NUM_FULL(134, "可参赛次数已满", true),
-    PHOTO_FILE_EXCEED_2M(134, "图片大小不能超过2M", true),
+    PHOTO_FILE_EXCEED_2M(135, "图片大小不能超过2M", true),
+    COACH_NUM_FULL(136,"该单位领队或教练人数已达上限",true)
+
 
     ;