gunshi-project-ss/src/main/java/com/gunshi/project/hsz/controller/PersonnelPlanLogController....

268 lines
11 KiB
Java

package com.gunshi.project.hsz.controller;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.gunshi.core.result.R;
import com.gunshi.project.hsz.entity.so.PersonnelPlanLogPage;
import com.gunshi.project.hsz.entity.vo.PersonnelPlanLogStatisticsVo;
import com.gunshi.project.hsz.model.PersonnelPlan;
import com.gunshi.project.hsz.model.PersonnelPlanLog;
import com.gunshi.project.hsz.service.FileAssociationsService;
import com.gunshi.project.hsz.service.PersonnelPlanLogService;
import com.gunshi.project.hsz.service.PersonnelPlanService;
import com.gunshi.project.hsz.util.DateUtil;
import com.gunshi.project.hsz.validate.markers.Insert;
import com.gunshi.project.hsz.validate.markers.Update;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.tags.Tag;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import java.io.Serializable;
import java.time.LocalDate;
import java.util.*;
/**
* Description:
* Created by XuSan on 2024/9/23.
*
* @author XuSan
* @version 1.0
*/
@Tag(name = "培训记录表")
@RestController
@RequestMapping(value = "/personnelPlanLog")
public class PersonnelPlanLogController extends AbstractCommonFileController{
@Autowired
private PersonnelPlanLogService service;
@Autowired
private PersonnelPlanService planService;
@Autowired
private FileAssociationsService fileService;
@Operation(summary = "新增")
@PostMapping("/insert")
public R<PersonnelPlanLog> insert(@Validated(Insert.class) @RequestBody PersonnelPlanLog dto) {
if (Objects.nonNull(dto.getStm()) && Objects.nonNull(dto.getEtm()) && dto.getStm().compareTo(dto.getEtm()) >= 0){
throw new IllegalArgumentException("开始时间不能大于结束时间");
}
LambdaQueryChainWrapper<PersonnelPlanLog> query = service.lambdaQuery()
.eq(PersonnelPlanLog::getPlanDate, dto. getPlanDate())
.eq(PersonnelPlanLog::getName, dto.getName());
if (query.count() > 0){
throw new IllegalArgumentException("当前培训日期标题名称重复");
}
if (Objects.nonNull(dto.getPlanId()) && planService.lambdaQuery().eq(PersonnelPlan::getId, dto.getPlanId()).count() == 0) {
throw new IllegalArgumentException("培训计划不存在");
}
dto.setId(IdWorker.getId());
dto.setCreateTime(new Date());
boolean result = service.save(dto);
if (result){
fileService.saveFile(dto.getFiles1(), getGroupId(), String.valueOf( dto.getId()),"1");
fileService.saveFile(dto.getFiles2(), getGroupId(), String.valueOf( dto.getId()),"2");
}
return R.ok(result ? dto : null);
}
@Operation(summary = "修改")
@PostMapping("/update")
public R<PersonnelPlanLog> update(@Validated(Update.class) @RequestBody PersonnelPlanLog dto) {
if (Objects.nonNull(dto.getStm()) && Objects.nonNull(dto.getEtm()) && dto.getStm().compareTo(dto.getEtm()) >= 0){
throw new IllegalArgumentException("开始时间不能大于结束时间");
}
LambdaQueryChainWrapper<PersonnelPlanLog> query = service.lambdaQuery()
.ne(PersonnelPlanLog::getId, dto.getId())
.eq(PersonnelPlanLog::getPlanDate, dto. getPlanDate())
.eq(PersonnelPlanLog::getName, dto.getName());
if (query.count() > 0){
throw new IllegalArgumentException("当前培训日期标题名称重复");
}
if (Objects.nonNull(dto.getPlanId()) && planService.lambdaQuery().eq(PersonnelPlan::getId, dto.getPlanId()).count() == 0) {
throw new IllegalArgumentException("培训计划不存在");
}
dto.setCreateTime(null);
dto.setCreateBy(null);
dto.setCreateName(null);
dto.setUpdateTime(new Date());
boolean result = service.updateById(dto);
if (result){
fileService.saveFile(dto.getFiles1(), getGroupId(), String.valueOf( dto.getId()),"1");
fileService.saveFile(dto.getFiles2(), getGroupId(), String.valueOf( dto.getId()),"2");
}
return R.ok(result ? dto : null);
}
@Operation(summary = "删除")
@GetMapping("/del/{id}")
public R<Boolean> del(@Schema(name = "id") @PathVariable("id") Serializable id) {
return R.ok(service.removeById(id));
}
@Operation(summary = "获取详情(包括文件信息)")
@GetMapping("/get/{id}")
public R<PersonnelPlanLog> get(@Schema(name = "id") @PathVariable("id") Serializable id) {
PersonnelPlanLog o = service.getById(id);
if (Objects.isNull(o)) {
throw new IllegalArgumentException("当前数据不存在");
}
o.setFiles1(fileService.getFiles(getGroupId(),String.valueOf(o.getId()),"1"));
o.setFiles2(fileService.getFiles(getGroupId(),String.valueOf(o.getId()),"2"));
return R.ok(o);
}
@Operation(summary = "列表")
@PostMapping("/list")
public R<List<PersonnelPlanLog>> list() {
return R.ok(service.lambdaQuery()
.list());
}
@Operation(summary = "统计")
@GetMapping("/statistics/{year}")
public R<PersonnelPlanLogStatisticsVo> statistics(@PathVariable("year") Integer year) {
if (year < 1970 || year > LocalDate.now().getYear()) {
throw new IllegalArgumentException("年份不合法");
}
//plan planLog
Date stm = DateUtil.convertStringToDate(year + "-01-01 00:00:00");
Date etm = DateUtil.convertStringToDate(year + "-12-31 23:59:59");
List<PersonnelPlanLog> planLogs = service.lambdaQuery()
.between(PersonnelPlanLog::getPlanDate, stm, etm)
.list();
List<PersonnelPlan> plans = planService.lambdaQuery()
.between(PersonnelPlan::getStm, stm, etm)
.list();
PersonnelPlanLogStatisticsVo vo = new PersonnelPlanLogStatisticsVo();
// Map<Integer, PersonnelPlanLogStatisticsVo.EchartsData> map1 = Maps.newHashMap();
// Calendar calendar = Calendar.getInstance();
// int month = calendar.get(Calendar.MONTH) + 1;
if (CollectionUtils.isNotEmpty(planLogs)) {
vo.setNumberOfPeriods1(planLogs.size());
vo.setNumberOfPeriods2(plans.stream().mapToInt(PersonnelPlan::getNum).sum());
vo.setPersonNum1(planLogs.stream().mapToInt(PersonnelPlanLog::getNumPeople).sum());
vo.setPersonNum2(plans.stream()
.map(
item -> item.getNumPeople() * item.getNum()
).mapToInt(Integer::intValue).sum()
);
Calendar calendar = Calendar.getInstance();
List<PersonnelPlanLogStatisticsVo.EchartsData> logStatPerMonth = new ArrayList<>();
for (int i = 0; i < 12; i++) {
int month = i; //calendar的月份是从0开始
List<PersonnelPlanLog> logsPerMonth = planLogs.stream().filter(log -> {
Date planDate = log.getPlanDate();
calendar.setTime(planDate);
return month == calendar.get(Calendar.MONTH);
}).toList();
//month num1实际人数 num2实际期数
PersonnelPlanLogStatisticsVo.EchartsData chartItem = new PersonnelPlanLogStatisticsVo.EchartsData();
chartItem.setMonth(i + 1); //calendar的月份是从0开始
chartItem.setNum1(logsPerMonth.stream().mapToInt(PersonnelPlanLog::getNumPeople).sum());
chartItem.setNum2(logsPerMonth.size());
logStatPerMonth.add(chartItem);
// PersonnelPlanLogStatisticsVo.EchartsData echartsData1 = map1.get(i);
// if (Objects.isNull(echartsData1)) {
// echartsData1 = new PersonnelPlanLogStatisticsVo.EchartsData();
// }
// int finalI = i;
// List<PersonnelPlanLog> list = planLogs.stream()
// .filter(item ->
// {
// calendar.setTime(item.getPlanDate());
// return month == finalI;
// })
// .toList();
//
// echartsData1.setMonth(finalI)
// .setNum1(list.stream().mapToInt(PersonnelPlanLog::getNumPeople).sum())
// .setNum2(list.size());
// map1.put(i, echartsData1);
}
vo.setList1(logStatPerMonth);
}
return R.ok(vo);
}
@Operation(summary = "分页")
@PostMapping("/page")
public R<Page<PersonnelPlanLog>> page(@RequestBody @Validated PersonnelPlanLogPage page) {
LambdaQueryWrapper<PersonnelPlanLog> query = Wrappers.lambdaQuery();
Date stm = page.getStm();
if (Objects.nonNull(stm)) {
query.ge(PersonnelPlanLog::getPlanDate, stm);
}
Date etm = page.getEtm();
if (Objects.nonNull(etm)) {
query.le(PersonnelPlanLog::getPlanDate, etm);
}
Integer type = page.getType();
if (Objects.nonNull(type)) {
query.eq(PersonnelPlanLog::getType, type);
}
Long planId = page.getPlanId();
if (Objects.nonNull(planId)) {
query.eq(PersonnelPlanLog::getPlanId, planId);
}
String trainees = page.getTrainees();
if (StringUtils.isNotBlank(trainees)) {
query.like(PersonnelPlanLog::getTrainees, trainees);
}
String unit = page.getUnit();
if (StringUtils.isNotBlank(unit)) {
query.like(PersonnelPlanLog::getUnit, unit);
}
query.orderByDesc(PersonnelPlanLog::getPlanDate);
Page<PersonnelPlanLog> data = service.page(page.getPageSo().toPage(), query);
data.getRecords().forEach(o -> {
o.setFiles1(fileService.getFiles(getGroupId(),String.valueOf(o.getId()),"1"));
o.setFiles2(fileService.getFiles(getGroupId(),String.valueOf(o.getId()),"2"));
});
return R.ok(data);
}
@Override
public String getGroupId() {
return "personnelPlanLog";
}
}