From 5d2cfd4562f74496a5263db3451c1e78ff53aead Mon Sep 17 00:00:00 2001
From: heheng <475597332@qq.com>
Date: 星期四, 27 三月 2025 08:44:16 +0800
Subject: [PATCH] 单据增加查询返回字段

---
 assess-system/src/main/java/com/gkhy/assess/system/service/impl/AssPlanPersonServiceImpl.java |   34 ++++++++++++++++++++++++++++------
 1 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/assess-system/src/main/java/com/gkhy/assess/system/service/impl/AssPlanPersonServiceImpl.java b/assess-system/src/main/java/com/gkhy/assess/system/service/impl/AssPlanPersonServiceImpl.java
index e641012..5b9beeb 100644
--- a/assess-system/src/main/java/com/gkhy/assess/system/service/impl/AssPlanPersonServiceImpl.java
+++ b/assess-system/src/main/java/com/gkhy/assess/system/service/impl/AssPlanPersonServiceImpl.java
@@ -1,21 +1,27 @@
 package com.gkhy.assess.system.service.impl;
 
+import cn.hutool.core.util.ObjectUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.gkhy.assess.common.enums.DeleteFlagEnum;
 import com.gkhy.assess.common.exception.ApiException;
 import com.gkhy.assess.common.utils.StringUtils;
 import com.gkhy.assess.system.domain.AssPlanPerson;
+import com.gkhy.assess.system.domain.SysUser;
 import com.gkhy.assess.system.enums.PlayRoleEnum;
 import com.gkhy.assess.system.mapper.AssPlanPersonMapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.gkhy.assess.system.mapper.SysUserMapper;
 import com.gkhy.assess.system.service.AssPlanPersonService;
 import com.gkhy.assess.system.service.AssProjectService;
 import com.gkhy.assess.system.utils.ShiroUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
+import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
+import java.util.stream.Collectors;
 
 /**
  * <p>
@@ -29,6 +35,8 @@
 public class AssPlanPersonServiceImpl extends ServiceImpl<AssPlanPersonMapper, AssPlanPerson> implements AssPlanPersonService {
     @Autowired
     private AssProjectService projectService;
+    @Autowired
+    private SysUserMapper sysUserMapper;
     @Override
     public List<AssPlanPerson> getByProjectId(Long projectId) {
         projectService.checkUserAllowed(projectId);
@@ -44,24 +52,31 @@
         return baseMapper.insert(planPerson);
     }
 
-
     @Override
     public void checkPersonUnique(AssPlanPerson planPerson) {
+        if(planPerson.getPlayRole()==null){
+            planPerson.setPlayRole(PlayRoleEnum.PROJECT_MEMBER.getCode());
+        }
         LambdaQueryWrapper<AssPlanPerson> lambdaQueryWrapper = Wrappers.<AssPlanPerson>lambdaQuery()
                 .eq(AssPlanPerson::getPersonId, planPerson.getPersonId())
                 .eq(AssPlanPerson::getDelFlag, DeleteFlagEnum.UN_DELETE)
-                .eq(AssPlanPerson::getProjectId, planPerson.getProjectId());
+                .eq(AssPlanPerson::getProjectId, planPerson.getProjectId())
+                .eq(AssPlanPerson::getPlayRole,planPerson.getPlayRole());
         if(planPerson.getId()!=null){
             lambdaQueryWrapper.ne(AssPlanPerson::getId,planPerson.getId());
         }
         AssPlanPerson pPerson= getOne(lambdaQueryWrapper);
         if(pPerson!=null){
             PlayRoleEnum playRoleEnum=PlayRoleEnum.getInfoByCode(pPerson.getPlayRole());
+            SysUser sysUser=sysUserMapper.selectById(planPerson.getPersonId());
+            if(sysUser==null){
+                throw new ApiException("选择的用户不存在");
+            }
             String message=playRoleEnum!=null?playRoleEnum.getInfo():"";
             if(StringUtils.isBlank(message)){
-                message="该用户已存在";
+                message=String.format("用户<%s>已担任职务",sysUser.getName());
             }else{
-                message="该用户已担任"+message;
+                message=String.format("用户<%s>已担任%s",sysUser.getName(),message);
             }
             throw new ApiException(message);
         }
@@ -73,6 +88,11 @@
                 .eq(AssPlanPerson::getPersonId,personId)
                 .eq(AssPlanPerson::getDelFlag,DeleteFlagEnum.UN_DELETE.getCode())
                 .eq(AssPlanPerson::getProjectId,projectId));
+    }
+
+    @Override
+    public int getMajorPlanPersonCountByProject(Long projectId) {
+        return baseMapper.getMajorPlanPersonCountByProject(projectId);
     }
 
     @Override
@@ -96,10 +116,12 @@
     }
 
     @Override
-    public int deletePlanPersonByPersonId(Long personId) {
+    public int deletePlanPersonByPersonId(Long personId,Long projectId,Integer playCode) {
         return baseMapper.update(null,Wrappers.<AssPlanPerson>lambdaUpdate()
                 .set(AssPlanPerson::getDelFlag,DeleteFlagEnum.DELETED.getCode())
                 .set(AssPlanPerson::getUpdateBy,ShiroUtils.getSysUser().getUsername())
-                .eq(AssPlanPerson::getPersonId,personId));
+                .eq(AssPlanPerson::getPersonId,personId)
+                .eq(AssPlanPerson::getProjectId,projectId)
+                .eq(AssPlanPerson::getPlayRole,playCode));
     }
 }

--
Gitblit v1.9.2