package cn.iocoder.yudao.module.hrm.controller.admin.training;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.iocoder.yudao.framework.common.pojo.CommonResult;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.hrm.controller.admin.training.vo.HrmTrainingPageReqVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.training.vo.HrmTrainingParticipantRespVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.training.vo.HrmTrainingRespVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.training.vo.HrmTrainingSaveReqVO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.training.HrmTrainingDO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.training.HrmTrainingParticipantDO;
|
import cn.iocoder.yudao.module.hrm.service.employee.HrmEmployeeService;
|
import cn.iocoder.yudao.module.hrm.service.training.HrmTrainingService;
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.Parameter;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import jakarta.annotation.Resource;
|
import jakarta.validation.Valid;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.validation.annotation.Validated;
|
import org.springframework.web.bind.annotation.*;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
|
import static cn.iocoder.yudao.framework.common.pojo.CommonResult.success;
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
|
import static cn.iocoder.yudao.module.hrm.dal.dataobject.training.HrmTrainingDO.*;
|
|
@Tag(name = "管理后台 - HRM 培训活动")
|
@RestController
|
@RequestMapping("/hrm/training")
|
@Validated
|
public class HrmTrainingController {
|
|
@Resource
|
private HrmTrainingService trainingService;
|
@Resource
|
private HrmEmployeeService employeeService;
|
|
@PostMapping("/create")
|
@Operation(summary = "创建培训活动（初始为草稿状态）")
|
@PreAuthorize("@ss.hasPermission('hrm:training:create')")
|
public CommonResult<Long> createTraining(@Valid @RequestBody HrmTrainingSaveReqVO createReqVO) {
|
return success(trainingService.createTraining(createReqVO));
|
}
|
|
@PutMapping("/update")
|
@Operation(summary = "更新培训活动（仅草稿/待开展可改，状态走 start/complete/cancel 流转）")
|
@PreAuthorize("@ss.hasPermission('hrm:training:update')")
|
public CommonResult<Boolean> updateTraining(@Valid @RequestBody HrmTrainingSaveReqVO updateReqVO) {
|
trainingService.updateTraining(updateReqVO);
|
return success(true);
|
}
|
|
@DeleteMapping("/delete")
|
@Operation(summary = "删除培训活动（级联删除参训人员；已完成/已取消不可删）")
|
@Parameter(name = "ids", description = "编号数组", required = true)
|
@PreAuthorize("@ss.hasPermission('hrm:training:delete')")
|
public CommonResult<Boolean> deleteTraining(@RequestParam("ids") List<Long> ids) {
|
trainingService.deleteTraining(ids);
|
return success(true);
|
}
|
|
@GetMapping("/get")
|
@Operation(summary = "获得培训活动详情（含参训人员列表）")
|
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
@PreAuthorize("@ss.hasPermission('hrm:training:query')")
|
public CommonResult<HrmTrainingRespVO> getTraining(@RequestParam("id") Long id) {
|
HrmTrainingDO training = trainingService.getTraining(id);
|
if (training == null) {
|
return success(null);
|
}
|
HrmTrainingRespVO respVO = BeanUtils.toBean(training, HrmTrainingRespVO.class);
|
fillDisplayNames(respVO);
|
respVO.setParticipants(buildParticipantList(trainingService.getParticipantsByTrainingId(id)));
|
return success(respVO);
|
}
|
|
@GetMapping("/page")
|
@Operation(summary = "获得培训活动分页（含参训人数）")
|
@PreAuthorize("@ss.hasPermission('hrm:training:query')")
|
public CommonResult<PageResult<HrmTrainingRespVO>> getTrainingPage(@Valid HrmTrainingPageReqVO pageReqVO) {
|
PageResult<HrmTrainingDO> pageResult = trainingService.getTrainingPage(pageReqVO);
|
if (CollUtil.isEmpty(pageResult.getList())) {
|
return success(PageResult.empty(pageResult.getTotal()));
|
}
|
Map<Long, Integer> participantCountMap = trainingService.getParticipantCountMap(
|
convertSet(pageResult.getList(), HrmTrainingDO::getId));
|
return success(BeanUtils.toBean(pageResult, HrmTrainingRespVO.class, vo -> {
|
fillDisplayNames(vo);
|
vo.setParticipantCount(participantCountMap.getOrDefault(vo.getId(), 0));
|
}));
|
}
|
|
@PostMapping("/start")
|
@Operation(summary = "开始培训（草稿 → 待开展）")
|
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
@PreAuthorize("@ss.hasPermission('hrm:training:update')")
|
public CommonResult<Boolean> startTraining(@RequestParam("id") Long id) {
|
trainingService.startTraining(id);
|
return success(true);
|
}
|
|
@PostMapping("/complete")
|
@Operation(summary = "完成培训（草稿/待开展 → 已完成；开启自动发证时对合格人员联动发放证书）")
|
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
@PreAuthorize("@ss.hasPermission('hrm:training:update')")
|
public CommonResult<Boolean> completeTraining(@RequestParam("id") Long id) {
|
trainingService.completeTraining(id);
|
return success(true);
|
}
|
|
@PostMapping("/cancel")
|
@Operation(summary = "取消培训（草稿/待开展 → 已取消）")
|
@Parameter(name = "id", description = "编号", required = true, example = "1024")
|
@PreAuthorize("@ss.hasPermission('hrm:training:update')")
|
public CommonResult<Boolean> cancelTraining(@RequestParam("id") Long id) {
|
trainingService.cancelTraining(id);
|
return success(true);
|
}
|
|
// ========== 私有方法 ==========
|
|
private void fillDisplayNames(HrmTrainingRespVO respVO) {
|
respVO.setTrainingTypeName(convertTrainingTypeName(respVO.getTrainingType()));
|
respVO.setStatusName(convertStatusName(respVO.getStatus()));
|
}
|
|
private List<HrmTrainingParticipantRespVO> buildParticipantList(List<HrmTrainingParticipantDO> participants) {
|
if (CollUtil.isEmpty(participants)) {
|
return new ArrayList<>();
|
}
|
Map<Long, HrmEmployeeDO> employeeMap = employeeService.getEmployeeList(
|
convertSet(participants, HrmTrainingParticipantDO::getEmployeeId)).stream()
|
.collect(java.util.stream.Collectors.toMap(HrmEmployeeDO::getId, e -> e, (a, b) -> a));
|
List<HrmTrainingParticipantRespVO> voList = BeanUtils.toBean(participants, HrmTrainingParticipantRespVO.class);
|
for (HrmTrainingParticipantRespVO vo : voList) {
|
HrmEmployeeDO employee = employeeMap.get(vo.getEmployeeId());
|
if (employee != null) {
|
vo.setEmployeeName(employee.getName());
|
vo.setEmployeeNo(employee.getEmployeeNo());
|
}
|
}
|
return voList;
|
}
|
|
private String convertTrainingTypeName(Integer trainingType) {
|
return switch (trainingType == null ? 0 : trainingType) {
|
case TYPE_SAFETY -> "安全培训";
|
case TYPE_SKILL -> "技能培训";
|
case TYPE_MANAGE -> "管理培训";
|
case TYPE_OTHER -> "其他";
|
default -> "";
|
};
|
}
|
|
private String convertStatusName(Integer status) {
|
return switch (status == null ? -1 : status) {
|
case STATUS_DRAFT -> "草稿";
|
case STATUS_PLANNED -> "待开展";
|
case STATUS_COMPLETED -> "已完成";
|
case STATUS_CANCELED -> "已取消";
|
default -> "";
|
};
|
}
|
|
}
|
