/*
|
* Copyright (c) 2018-2025, ztt All rights reserved.
|
*
|
* Redistribution and use in source and binary forms, with or without
|
* modification, are permitted provided that the following conditions are met:
|
*
|
* Redistributions of source code must retain the above copyright notice,
|
* this list of conditions and the following disclaimer.
|
* Redistributions in binary form must reproduce the above copyright
|
* notice, this list of conditions and the following disclaimer in the
|
* documentation and/or other materials provided with the distribution.
|
* Neither the name of the pig4cloud.com developer nor the names of its
|
* contributors may be used to endorse or promote products derived from
|
* this software without specific prior written permission.
|
* Author: ztt
|
*/
|
|
package com.chinaztt.mes.production.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.excel.ExcelWriter;
|
import com.alibaba.excel.write.metadata.WriteSheet;
|
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.chinaztt.mes.basic.entity.Workstation;
|
import com.chinaztt.mes.basic.service.WorkstationService;
|
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
|
import com.chinaztt.mes.production.dto.DutyRecordDTO;
|
import com.chinaztt.mes.production.entity.DutyRecord;
|
import com.chinaztt.mes.production.excel.*;
|
import com.chinaztt.mes.production.service.DutyRecordService;
|
import com.chinaztt.mes.production.util.BackUtils;
|
import com.chinaztt.ztt.common.core.util.R;
|
import com.chinaztt.ztt.common.log.annotation.SysLog;
|
import com.chinaztt.ztt.common.security.annotation.Inner;
|
import com.google.common.collect.Lists;
|
import com.xxl.job.core.handler.annotation.XxlJob;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import lombok.AllArgsConstructor;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.security.access.prepost.PreAuthorize;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.util.Arrays;
|
import java.util.Iterator;
|
import java.util.List;
|
import java.util.Optional;
|
import java.util.stream.Collectors;
|
|
|
/**
|
* 上班记录
|
*
|
* @author cxf
|
* @date 2021-01-20 09:11:25
|
*/
|
@RestController
|
@AllArgsConstructor
|
@RequestMapping("/dutyRecord")
|
@Api(value = "dutyRecord", tags = "上班记录管理")
|
public class DutyRecordController {
|
|
private final DutyRecordService dutyRecordService;
|
private final BackUtils backUtils;
|
|
/**
|
* 分页查询
|
*
|
* @param page 分页对象
|
* @param dutyRecordDTO 上班记录
|
* @return
|
*/
|
@ApiOperation(value = "分页查询", notes = "分页查询")
|
@GetMapping("/page")
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_view','product_workbench')")
|
public R getDutyRecordPage(Page page, DutyRecordDTO dutyRecordDTO) {
|
return R.ok(dutyRecordService.getDutyRecordPage(page, QueryWrapperUtil.gen(dutyRecordDTO)));
|
}
|
|
/**
|
* 导出
|
*
|
* @param dutyRecordDTO 上班记录-人工明细
|
* @return
|
*/
|
@ApiOperation(value = "导出", notes = "导出")
|
@Inner(false)
|
@GetMapping("/exportDutyRecord")
|
public void exportDutyRecord(HttpServletResponse response, DutyRecordDTO dutyRecordDTO) throws IOException {
|
response.setContentType("application/vnd.ms-excel");
|
response.setCharacterEncoding("UTF-8");
|
// 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
|
String fileName = URLEncoder.encode("人工记录", "UTF-8");
|
response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");
|
try {
|
//新建ExcelWriter
|
ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()).build();
|
//获取sheet0对象
|
WriteSheet mainSheet = EasyExcel.writerSheet(0, "人工记录").head(DutyRecordData.class).build();
|
//向sheet0写入数据 传入空list这样只导出表头
|
excelWriter.write(dutyRecordService.exportList(QueryWrapperUtil.gen(dutyRecordDTO)), mainSheet);
|
//获取sheet1对象
|
WriteSheet detailSheet = EasyExcel.writerSheet(1, "人员出勤").head(PersonBoardData.class).build();
|
//向sheet1写入数据 传入空list这样只导出表头
|
excelWriter.write(dutyRecordService.exportPersonBoard(QueryWrapperUtil.gen(dutyRecordDTO)), detailSheet);
|
//获取sheet2对象
|
WriteSheet attendanceSheet = EasyExcel.writerSheet(2, "考勤统计").head(AttendanceData.class).build();
|
//向sheet2写入数据 传入空list这样只导出表头
|
excelWriter.write(dutyRecordService.exportAttendance(QueryWrapperUtil.gen(dutyRecordDTO)), attendanceSheet);
|
//获取sheet3对象
|
WriteSheet shiftWageSheet = EasyExcel.writerSheet(3, "班次工资").head(ShiftWageData.class).build();
|
//向sheet3写入数据 传入空list这样只导出表头
|
excelWriter.write(dutyRecordService.exportShiftWage(QueryWrapperUtil.gen(dutyRecordDTO)), shiftWageSheet);
|
//关闭流
|
excelWriter.finish();
|
} catch (IOException e) {
|
throw new RuntimeException("导出失败");
|
}
|
}
|
|
/**
|
* 通过id查询上班记录
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id查询", notes = "通过id查询")
|
@GetMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_view','product_workbench')")
|
public R getById(@PathVariable("id") Long id) {
|
return R.ok(dutyRecordService.getDtoById(id));
|
}
|
|
/**
|
* 新增上班记录
|
*
|
* @param dutyRecordDTO 上班记录
|
* @return R
|
*/
|
@ApiOperation(value = "新增上班记录", notes = "新增上班记录")
|
@SysLog("新增上班记录")
|
@PostMapping
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_add','product_workbench')")
|
public R save(@RequestBody DutyRecordDTO dutyRecordDTO) {
|
DutyRecordDTO dutyRecord = dutyRecordService.saveDto(dutyRecordDTO);
|
backUtils.backDutyRecordById(dutyRecordDTO.getId(), "新增");
|
return R.ok(dutyRecord);
|
}
|
|
|
/**
|
* 展示该工作站对应的所有的机台
|
*
|
* @param workCenter
|
* @return R
|
*/
|
@ApiOperation(value = "展示该工作站对应的所有的机台", notes = "展示该工作站对应的所有的机台")
|
@SysLog("展示该工作站对应的所有的机台")
|
@GetMapping("/getWorkstation")
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_add','product_workbench')")
|
public R getWorkstation(@RequestParam("workCenter") String workCenter) {
|
return R.ok(dutyRecordService.getWorkstationList(workCenter));
|
}
|
|
|
/**
|
* 展示该工作中心对应的未提交班次信息
|
*
|
* @param workCenter
|
* @return R
|
*/
|
@ApiOperation(value = "展示该工作中心对应的未提交班次信息", notes = "展示该工作中心对应的未提交班次信息")
|
@SysLog("展示该工作中心对应的未提交班次信息")
|
@GetMapping("/getUnsubmitWorkstation")
|
public R getUnsubmitWorkstation(@RequestParam("workCenter") String workCenter) {
|
return R.ok(dutyRecordService.getUnsubmitWorkstationList(workCenter));
|
}
|
|
|
|
/**
|
* 批量新增上班记录
|
* @param dutyRecordDTO
|
* @return
|
*/
|
@ApiOperation(value = "批量新增上班记录", notes = "批量新增上班记录")
|
@SysLog("批量新增上班记录")
|
@PostMapping("/batch")
|
@PreAuthorize("@pms.hasPermission('product_add_shift_batch','product_workbench')")
|
public R batchSave(@RequestBody DutyRecordDTO dutyRecordDTO) {
|
List<DutyRecordDTO> records = dutyRecordService.batchSave(dutyRecordDTO);
|
backUtils.batchbackDutyRecordByIds(records.stream().map(DutyRecordDTO::getId).collect(Collectors.toList()), "批量新增");
|
Optional<DutyRecordDTO> first = records.stream().filter(p -> p.getWorkstationId().equals(dutyRecordDTO.getWorkstationId())).findFirst();
|
if (first.isPresent()) {
|
return R.ok(first.get());
|
}
|
return R.ok();
|
}
|
|
/**
|
* 修改上班记录
|
*
|
* @param dutyRecordDTO 上班记录
|
* @return R
|
*/
|
@ApiOperation(value = "修改上班记录", notes = "修改上班记录")
|
@SysLog("修改上班记录")
|
@PutMapping
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_edit','product_workbench')")
|
public R updateById(@RequestBody DutyRecordDTO dutyRecordDTO) {
|
//工作台模块限制直接更新
|
if(dutyRecordDTO.getIsLimitUpd()) {
|
Boolean isSubmit = dutyRecordService.getById(dutyRecordDTO.getId()).getIsSubmit();
|
if (isSubmit) {
|
throw new RuntimeException("班次已提交，不可编辑");
|
}
|
}
|
boolean result = dutyRecordService.updateById(dutyRecordDTO);
|
backUtils.backDutyRecordById(dutyRecordDTO.getId(), dutyRecordDTO.getOperationType());
|
return R.ok(dutyRecordService.getDtoById(dutyRecordDTO.getId()));
|
}
|
|
/**
|
* 通过id删除上班记录
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id删除上班记录", notes = "通过id删除上班记录")
|
@SysLog("通过id删除上班记录")
|
@DeleteMapping("/{id}")
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_del','product_workbench')")
|
public R removeById(@PathVariable Long id) {
|
backUtils.backDeleteDutyRecordById(Arrays.asList(id));
|
return R.ok(dutyRecordService.deleteById(id));
|
}
|
|
/**
|
* 通过上班记录id查询人员List
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过上班记录id查询人员List", notes = "通过上班记录id查询人员List")
|
@GetMapping("/getPersonByDutyRecordId/{id}")
|
public R getPersonByDutyRecordId(@PathVariable("id") Long id) {
|
return R.ok(dutyRecordService.getPersonByDutyRecordId(id));
|
}
|
|
/**
|
* 通过上班记录idList查询人员List
|
*
|
* @param idList
|
* @return R
|
*/
|
@ApiOperation(value = "通过上班记录idList查询人员List", notes = "通过上班记录idList查询人员List")
|
@GetMapping("/getPersonByDutyRecordIdList")
|
public R getPersonByDutyRecordIdList(@RequestParam("idList") List<Long> idList) {
|
return R.ok(dutyRecordService.getPersonByDutyRecordIdList(idList));
|
}
|
|
/**
|
* 通过上班记录id查询产量
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过上班记录id查询产量", notes = "通过上班记录id查询产量")
|
@GetMapping("/getOutputByDutyRecordId/{id}")
|
public R getOutputByDutyRecordId(@PathVariable("id") Long id) {
|
return R.ok(dutyRecordService.getOutputByDutyRecordId(id));
|
}
|
|
|
/**
|
* 通过上班记录idList查询产量
|
*
|
* @param idList
|
* @return R
|
*/
|
@ApiOperation(value = "通过上班记录idList查询产量", notes = "通过上班记录idList查询产量")
|
@GetMapping("/getOutputByDutyRecordIdList")
|
public R getOutputByDutyRecordIdList(@RequestParam("idList") List<Long> idList) {
|
return R.ok(dutyRecordService.getOutputByDutyRecordIdList(idList));
|
}
|
|
|
|
|
/**
|
* 通过工作站id查询班次
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过工作站id查询班次", notes = "通过工作站id查询班次")
|
@GetMapping("/getDutyRecordByWorkstationId/{id}")
|
public R getDutyRecordByWorkstationId(@PathVariable("id") Long id) {
|
return R.ok(dutyRecordService.getDutyRecordByWorkstationId(id));
|
}
|
|
/**
|
* 通过id查询操作记录
|
*
|
* @param id id
|
* @return R
|
*/
|
@ApiOperation(value = "通过id查询操作记录", notes = "通过id查询操作记录")
|
@GetMapping("/getOperatingRecord/{id}")
|
public R getOperatingRecord(@PathVariable("id") Long id) {
|
return R.ok(dutyRecordService.getOperatingRecord(id));
|
}
|
|
/**
|
* 分页查询人工记录
|
*
|
* @param page 分页对象
|
* @param dutyRecordDTO 上班记录
|
* @return
|
*/
|
@ApiOperation(value = "分页查询人工记录", notes = "分页查询人工记录")
|
@GetMapping("/getArtificialRecord")
|
public R getArtificialRecord(Page page, DutyRecordDTO dutyRecordDTO) {
|
return R.ok(dutyRecordService.getArtificialRecord(page, QueryWrapperUtil.gen(dutyRecordDTO)));
|
}
|
|
/**
|
* 工作台班次未提交提醒用于定时任务
|
*
|
* @return
|
*/
|
@ApiOperation(value = "工作台班次未提交提醒", notes = "工作台班次未提交提醒")
|
@GetMapping("/sendUnsubmitMsgAuto")
|
@XxlJob("sendUnsubmitMsgAutoHandler")
|
public R sendUnsubmitMsg(){
|
return R.ok(dutyRecordService.sendUnsubmitMsg());
|
}
|
|
|
/**
|
* 班次批量提交
|
*
|
* @param ids
|
* @return R
|
*/
|
@ApiOperation(value = "班次批量提交", notes = "班次批量提交")
|
@SysLog("班次批量提交")
|
@PostMapping("/batchSubmit")
|
public R batchSubmit(@RequestBody List<Long> ids) {
|
return R.ok(dutyRecordService.batchSubmit(ids));
|
}
|
|
|
|
/**
|
* 批量修改工时计算上班记录
|
*
|
* @param dutyRecordDTOList 上班记录
|
* @return R
|
*/
|
@ApiOperation(value = "批量修改工时计算上班记录", notes = "批量修改工时计算上班记录")
|
@SysLog("批量修改工时计算上班记录")
|
@PostMapping("updateBatchById")
|
@PreAuthorize("@pms.hasPermission('production_dutyrecord_edit','product_workbench')")
|
public R updateBatchById(@RequestBody List<DutyRecordDTO> dutyRecordDTOList) {
|
dutyRecordService.updateBatch(dutyRecordDTOList);
|
for (DutyRecordDTO dutyRecordDTO:dutyRecordDTOList) {
|
backUtils.backDutyRecordById(dutyRecordDTO.getId(), dutyRecordDTO.getOperationType());
|
}
|
return R.ok();
|
}
|
}
|
