绩效管理-班组组建

airport-admin/src/main/java/com/sundot/airport/web/controller/item/PerformanceIndicatorsController.java

@@ -4,11 +4,16 @@ import cn.hutool.core.collection.CollectionUtil;
 import cn.hutool.core.date.DateUtil;
 import com.sundot.airport.common.core.controller.BaseController;
 import com.sundot.airport.common.core.domain.AjaxResult;
+import com.sundot.airport.common.core.domain.entity.SysUser;
 import com.sundot.airport.common.dto.*;
+import com.sundot.airport.common.enums.RoleTypeEnum;
+import com.sundot.airport.common.utils.DeptUtils;
 import com.sundot.airport.common.utils.GeometricMeanUtils;
 import com.sundot.airport.common.utils.poi.ExcelUtil;
 import com.sundot.airport.system.domain.SysConfig;
+import com.sundot.airport.system.mapper.SysUserMapper;
 import com.sundot.airport.system.service.ISysConfigService;
+import com.sundot.airport.system.service.ISysDeptService;
 import io.swagger.annotations.Api;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -37,6 +42,12 @@ public class PerformanceIndicatorsController extends BaseController {
     @Autowired
     private PerformanceMetricsController performanceMetricsController;
 
+    @Autowired
+    private SysUserMapper userMapper;
+
+    @Autowired
+    private ISysDeptService sysDeptService;
+
 
     // 绩效矩阵参数的配置键名
     private static final String CONFIG_KEY_PARAM_A = "item.performance.matrix.param_a";
@@ -179,6 +190,260 @@ public class PerformanceIndicatorsController extends BaseController {
     }
 
     /**
+     * 班组组建
+     *
+     * @param param 组建参数
+     * @return 组建结果
+     */
+    @PostMapping("/team/build")
+    public AjaxResult teamBuild(@RequestBody TeamBuildParamDto param) {
+        try {
+            // 参数校验
+            if (param.getTeamCount() == null || param.getTeamCount() < 1 || param.getTeamCount() > 10) {
+                return error("重点班组数量必须在1-10之间");
+            }
+            if (param.getTargetSize() == null || param.getTargetSize() < 2) {
+                return error("目标班组人数至少为2人");
+            }
+            if (param.getLeaderCondition() == null || param.getLeaderCondition() < 1 || param.getLeaderCondition() > 3) {
+                return error("组建班组长条件必须为1-综合能力、2-管理能力、3-业务能力");
+            }
+
+            // 查询个人维度的绩效数据（包含班组长和安检员）
+            PerformanceMetricsParamDto metricsParam = new PerformanceMetricsParamDto();
+            metricsParam.setDimension(1); // 个人维度
+            metricsParam.setStartTime(DateUtil.parse(param.getStartTime(), "yyyy-MM-dd"));
+            metricsParam.setEndTime(DateUtil.parse(param.getEndTime(), "yyyy-MM-dd"));
+            metricsParam.setSortField("totalScore");
+            metricsParam.setSortOrder(2); // 降序
+
+            AjaxResult result = performanceMetricsController.getPerformanceMetrics(metricsParam);
+            List<PerformanceMetricsResultDto> personalMetrics = (List<PerformanceMetricsResultDto>) result.get("data");
+
+            if (personalMetrics == null || personalMetrics.isEmpty()) {
+                return error("未查询到绩效数据");
+            }
+
+            // 获取所有班组长和安检员的用户信息
+            Long topSiteId = DeptUtils.getTopSiteId(sysDeptService.selectDeptById(getDeptId()));
+            if (topSiteId == null) {
+                return error("无法找到有效的站点信息");
+            }
+
+            List<SysUser> allUsers = userMapper.selectUserByDeptId(topSiteId);
+            Map<Long, SysUser> userMap = allUsers.stream()
+                    .filter(user -> user.getRoles() != null)
+                    .filter(user -> user.getRoles().stream()
+                            .anyMatch(role -> RoleTypeEnum.banzuzhang.getCode().equals(role.getRoleKey())
+                                    || RoleTypeEnum.SecurityCheck.getCode().equals(role.getRoleKey())))
+                    .collect(Collectors.toMap(SysUser::getUserId, user -> user, (u1, u2) -> u1));
+
+            // 根据组建条件选择班组长
+            List<TeamBuildResultDto> teamResults = new ArrayList<>();
+            Set<Long> selectedUserIds = new HashSet<>(); // 记录已选中的用户
+
+            // 1. 选择班组长
+            List<TeamMemberDto> leaders = selectLeaders(param, personalMetrics, userMap);
+            if (leaders.size() < param.getTeamCount()) {
+                return error("班组长人数不足，无法组建" + param.getTeamCount() + "个重点班组");
+            }
+
+            // 2. 选择成员并分配
+            List<TeamMemberDto> members = selectMembers(param, personalMetrics, userMap, selectedUserIds);
+
+            // 3. 组建班组
+            int memberCount = param.getTargetSize() - 1; // 每个班组的成员数 = 目标人数 - 1个班组长
+            int totalMembersNeeded = param.getTeamCount() * memberCount; // 总共需要的成员数
+
+            if (members.size() < totalMembersNeeded) {
+                return error("成员人数不足，需要" + totalMembersNeeded + "人，但只有" + members.size() + "人");
+            }
+
+            // 4. 分配成员到各个班组
+            for (int i = 0; i < param.getTeamCount(); i++) {
+                TeamBuildResultDto teamResult = new TeamBuildResultDto();
+                teamResult.setTeamNo(i + 1);
+
+                List<TeamMemberDto> teamMembers = new ArrayList<>();
+                // 添加班组长
+                teamMembers.add(leaders.get(i));
+                selectedUserIds.add(leaders.get(i).getUserId());
+
+                // 随机分配成员
+                for (int j = 0; j < memberCount; j++) {
+                    int randomIndex = new Random().nextInt(members.size());
+                    TeamMemberDto member = members.remove(randomIndex);
+                    teamMembers.add(member);
+                    selectedUserIds.add(member.getUserId());
+                }
+
+                teamResult.setMembers(teamMembers);
+                teamResults.add(teamResult);
+            }
+
+            return success(teamResults);
+        } catch (Exception e) {
+            log.error("班组组建失败：{}", e.getMessage());
+            return error("班组组建失败：" + e.getMessage());
+        }
+    }
+
+    /**
+     * 根据条件选择班组长
+     */
+    private List<TeamMemberDto> selectLeaders(TeamBuildParamDto param,
+                                              List<PerformanceMetricsResultDto> personalMetrics,
+                                              Map<Long, SysUser> userMap) {
+        List<TeamMemberDto> leaders = new ArrayList<>();
+
+        // 筛选出班组长角色的用户绩效数据
+        List<PerformanceMetricsResultDto> leaderMetrics = personalMetrics.stream()
+                .filter(metric -> {
+                    SysUser user = userMap.get(metric.getId());
+                    return user != null && user.getRoles().stream()
+                            .anyMatch(role -> RoleTypeEnum.banzuzhang.getCode().equals(role.getRoleKey()));
+                })
+                .collect(Collectors.toList());
+
+        Integer leaderCondition = param.getLeaderCondition();
+        if (leaderCondition == 1) {
+            // 综合能力：个人总分和班组总分之和除以2
+            // 获取班组绩效数据
+            PerformanceMetricsParamDto teamParam = new PerformanceMetricsParamDto();
+            teamParam.setDimension(2);
+            teamParam.setStartTime(DateUtil.parse(param.getStartTime(), "yyyy-MM-dd"));
+            teamParam.setEndTime(DateUtil.parse(param.getEndTime(), "yyyy-MM-dd"));
+            teamParam.setSortField("totalScore");
+            teamParam.setSortOrder(2);
+
+            AjaxResult teamResult = performanceMetricsController.getPerformanceMetrics(teamParam);
+            List<PerformanceMetricsResultDto> teamMetrics = (List<PerformanceMetricsResultDto>) teamResult.get("data");
+            Map<Long, BigDecimal> teamScoreMap = teamMetrics.stream()
+                    .collect(Collectors.toMap(PerformanceMetricsResultDto::getId, PerformanceMetricsResultDto::getTotalScore, (s1, s2) -> s1));
+
+            // 计算综合得分
+            for (PerformanceMetricsResultDto metric : leaderMetrics) {
+                SysUser user = userMap.get(metric.getId());
+                if (user != null && user.getDeptId() != null) {
+                    BigDecimal teamScore = teamScoreMap.getOrDefault(user.getDeptId(), BigDecimal.ZERO);
+                    BigDecimal comprehensiveScore = metric.getTotalScore()
+                            .add(teamScore)
+                            .divide(new BigDecimal("2"), 2, BigDecimal.ROUND_HALF_UP);
+
+                    TeamMemberDto leader = new TeamMemberDto();
+                    leader.setUserId(user.getUserId());
+                    leader.setName(user.getNickName());
+                    leader.setUserName(user.getUserName());
+                    leader.setRole("班组长");
+                    leader.setScore(comprehensiveScore);
+                    leaders.add(leader);
+                }
+            }
+
+            // 按综合得分降序排序
+            leaders.sort((a, b) -> b.getScore().compareTo(a.getScore()));
+
+        } else if (leaderCondition == 2) {
+            // 管理能力：使用班组维度的绩效总分
+            PerformanceMetricsParamDto teamParam = new PerformanceMetricsParamDto();
+            teamParam.setDimension(2);
+            teamParam.setStartTime(DateUtil.parse(param.getStartTime(), "yyyy-MM-dd"));
+            teamParam.setEndTime(DateUtil.parse(param.getEndTime(), "yyyy-MM-dd"));
+            teamParam.setSortField("totalScore");
+            teamParam.setSortOrder(2);
+
+            AjaxResult teamResult = performanceMetricsController.getPerformanceMetrics(teamParam);
+            List<PerformanceMetricsResultDto> teamMetrics = (List<PerformanceMetricsResultDto>) teamResult.get("data");
+            Map<Long, BigDecimal> teamScoreMap = teamMetrics.stream()
+                    .collect(Collectors.toMap(PerformanceMetricsResultDto::getId, PerformanceMetricsResultDto::getTotalScore, (s1, s2) -> s1));
+
+            // 获取班组长所在的班组得分
+            for (PerformanceMetricsResultDto metric : leaderMetrics) {
+                SysUser user = userMap.get(metric.getId());
+                if (user != null && user.getDeptId() != null) {
+                    BigDecimal teamScore = teamScoreMap.getOrDefault(user.getDeptId(), BigDecimal.ZERO);
+
+                    TeamMemberDto leader = new TeamMemberDto();
+                    leader.setUserId(user.getUserId());
+                    leader.setName(user.getNickName());
+                    leader.setUserName(user.getUserName());
+                    leader.setRole("班组长");
+                    leader.setScore(teamScore);
+                    leaders.add(leader);
+                }
+            }
+
+            // 按班组得分降序排序
+            leaders.sort((a, b) -> b.getScore().compareTo(a.getScore()));
+
+        } else if (leaderCondition == 3) {
+            // 业务能力：使用个人的绩效总分
+            for (PerformanceMetricsResultDto metric : leaderMetrics) {
+                SysUser user = userMap.get(metric.getId());
+                if (user != null) {
+                    TeamMemberDto leader = new TeamMemberDto();
+                    leader.setUserId(user.getUserId());
+                    leader.setName(user.getNickName());
+                    leader.setUserName(user.getUserName());
+                    leader.setRole("班组长");
+                    leader.setScore(metric.getTotalScore());
+                    leaders.add(leader);
+                }
+            }
+
+            // 按个人总分降序排序
+            leaders.sort((a, b) -> b.getScore().compareTo(a.getScore()));
+        }
+
+        // 取前 teamCount 个班组长
+        int teamCount = param.getTeamCount();
+        return leaders.subList(0, Math.min(teamCount, leaders.size()));
+    }
+
+    /**
+     * 选择成员（安检员角色）
+     */
+    private List<TeamMemberDto> selectMembers(TeamBuildParamDto param,
+                                              List<PerformanceMetricsResultDto> personalMetrics,
+                                              Map<Long, SysUser> userMap,
+                                              Set<Long> selectedUserIds) {
+        List<TeamMemberDto> members = new ArrayList<>();
+
+        // 筛选出安检员角色的用户绩效数据
+        List<PerformanceMetricsResultDto> memberMetrics = personalMetrics.stream()
+                .filter(metric -> {
+                    SysUser user = userMap.get(metric.getId());
+                    return user != null && user.getRoles().stream()
+                            .anyMatch(role -> RoleTypeEnum.SecurityCheck.getCode().equals(role.getRoleKey()));
+                })
+                .collect(Collectors.toList());
+
+        // 转换为成员DTO
+        for (PerformanceMetricsResultDto metric : memberMetrics) {
+            if (selectedUserIds.contains(metric.getId())) {
+                continue; // 跳过已选中的用户
+            }
+
+            SysUser user = userMap.get(metric.getId());
+            if (user != null) {
+                TeamMemberDto member = new TeamMemberDto();
+                member.setUserId(user.getUserId());
+                member.setName(user.getNickName());
+                member.setUserName(user.getUserName());
+                member.setRole("安检员");
+                member.setScore(metric.getTotalScore());
+                members.add(member);
+            }
+        }
+
+        // 按个人总分降序排序
+        members.sort((a, b) -> b.getScore().compareTo(a.getScore()));
+
+        return members;
+    }
+
+
+    /**
      * 转换为人员导出 DTO 列表
      */
     private List<PerformanceMetricsPersonalExportDto> convertToPersonalExportList(List<PerformanceMetricsResultDto> dataList, Date startTime, Date endTime) {

airport-common/src/main/java/com/sundot/airport/common/dto/TeamBuildParamDto.java

@@ -0,0 +1,31 @@
+package com.sundot.airport.common.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+/**
+ * 班组组建参数DTO
+ *
+ * @author sundot
+ * @date 2026-04-07
+ */
+@Data
+@ApiModel("班组组建参数DTO")
+public class TeamBuildParamDto {
+
+    @ApiModelProperty("重点班组数量（1-10）")
+    private Integer teamCount;
+
+    @ApiModelProperty("目标班组人数（班组长1人 + 成员N人）")
+    private Integer targetSize;
+
+    @ApiModelProperty("组建班组长条件：1-综合能力，2-管理能力，3-业务能力")
+    private Integer leaderCondition;
+
+    @ApiModelProperty("开始时间")
+    private String startTime;
+
+    @ApiModelProperty("结束时间")
+    private String endTime;
+}

airport-common/src/main/java/com/sundot/airport/common/dto/TeamBuildResultDto.java

@@ -0,0 +1,24 @@
+package com.sundot.airport.common.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+import java.util.List;
+
+/**
+ * 班组组建结果DTO
+ *
+ * @author sundot
+ * @date 2026-04-07
+ */
+@Data
+@ApiModel("班组组建结果DTO")
+public class TeamBuildResultDto {
+
+    @ApiModelProperty("班组编号")
+    private Integer teamNo;
+
+    @ApiModelProperty("班组成员列表")
+    private List<TeamMemberDto> members;
+}

airport-common/src/main/java/com/sundot/airport/common/dto/TeamMemberDto.java

@@ -0,0 +1,33 @@
+package com.sundot.airport.common.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+import java.math.BigDecimal;
+
+/**
+ * 班组成员DTO
+ *
+ * @author sundot
+ * @date 2026-04-07
+ */
+@Data
+@ApiModel("班组成员DTO")
+public class TeamMemberDto {
+
+    @ApiModelProperty("用户ID")
+    private Long userId;
+
+    @ApiModelProperty("姓名")
+    private String name;
+
+    @ApiModelProperty("账号")
+    private String userName;
+
+    @ApiModelProperty("角色（班组长/安检员）")
+    private String role;
+
+    @ApiModelProperty("分数")
+    private BigDecimal score;
+}