package com.chinaztt.mes.production.service.impl;
|
|
import cn.hutool.core.collection.CollectionUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
|
import com.chinaztt.mes.basic.entity.Shift;
|
import com.chinaztt.mes.basic.entity.Staff;
|
import com.chinaztt.mes.basic.entity.Workstation;
|
import com.chinaztt.mes.basic.mapper.JoinCrewStaffMapper;
|
import com.chinaztt.mes.basic.mapper.ShiftMapper;
|
import com.chinaztt.mes.basic.mapper.WorkstationMapper;
|
import com.chinaztt.mes.common.numgen.NumberGenerator;
|
import com.chinaztt.mes.common.util.WxCpUtils;
|
import com.chinaztt.mes.production.config.UnsubmitDutyRecordConfig;
|
import com.chinaztt.mes.production.dto.DutyRecordDTO;
|
import com.chinaztt.mes.production.dto.PersonBoardDTO;
|
import com.chinaztt.mes.production.dto.ProductOutputDTO;
|
import com.chinaztt.mes.production.entity.ArtificialInformation;
|
import com.chinaztt.mes.production.entity.Computation;
|
import com.chinaztt.mes.production.entity.DutyRecord;
|
import com.chinaztt.mes.production.entity.PersonBoard;
|
import com.chinaztt.mes.production.excel.*;
|
import com.chinaztt.mes.production.mapper.*;
|
import com.chinaztt.mes.production.service.DutyRecordService;
|
import com.chinaztt.mes.production.util.BackUtils;
|
import com.chinaztt.ztt.common.data.datascope.DataScope;
|
import com.chinaztt.ztt.common.security.service.ZttUser;
|
import com.chinaztt.ztt.common.security.util.SecurityUtils;
|
import lombok.AllArgsConstructor;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
|
import java.time.LocalDateTime;
|
import java.time.format.DateTimeFormatter;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 上班记录
|
*
|
* @author cxf
|
* @date 2021-01-20 09:11:25
|
*/
|
@Service
|
@AllArgsConstructor
|
@Transactional(rollbackFor = Exception.class)
|
public class DutyRecordServiceImpl extends ServiceImpl<DutyRecordMapper, DutyRecord> implements DutyRecordService {
|
private static final DateTimeFormatter FORMATTER_DATE = DateTimeFormatter.ofPattern("yyyy-MM-dd");
|
private static final DateTimeFormatter FORMATTER_MONTH = DateTimeFormatter.ofPattern("yyyy-MM");
|
private ShiftMapper shiftMapper;
|
private PersonBoardMapper personBoardMapper;
|
private ProductOutputMapper productOutputMapper;
|
private ArtificialInformationMapper artificialInformationMapper;
|
private JoinCrewStaffMapper joinCrewStaffMapper;
|
private ComputationMapper computationMapper;
|
private NumberGenerator<DutyRecord> dutyRecordNumberGenerator;
|
private WorkstationMapper workstationMapper;
|
private UnsubmitDutyRecordConfig unsubmitDutyRecordConfig;
|
private final BackUtils backUtils;
|
private SegmentationTaskRecordMapper segmentationTaskRecordMapper;
|
|
@Override
|
public IPage getDutyRecordPage(Page page, QueryWrapper<DutyRecordDTO> gen) {
|
return baseMapper.getDutyRecordPage(page, gen, new DataScope());
|
}
|
|
@Override
|
public DutyRecordDTO saveDto(DutyRecordDTO dutyRecordDTO) {
|
dutyRecordDTO.setId(null);
|
ZttUser user = SecurityUtils.getUser();
|
dutyRecordDTO.setDeptId(user.getDeptId());
|
dutyRecordDTO.setIsConfirm(false);
|
dutyRecordDTO.setIsSubmit(false);
|
checkDutyMonth(dutyRecordDTO);
|
// 1.保存主表
|
Workstation workstation = workstationMapper.selectById(dutyRecordDTO.getWorkstationId());
|
dutyRecordDTO.setWorkCenter(workstation.getWorkCenter());
|
baseMapper.insert(dutyRecordDTO);
|
// 2.保存默认班组人员
|
if (dutyRecordDTO.getCrewId() != null) {
|
List<Staff> staffs = joinCrewStaffMapper.selectStaffByCrewId(dutyRecordDTO.getCrewId());
|
for (Staff staff : staffs) {
|
PersonBoard personBoard = new PersonBoard();
|
personBoard.setWorkTime(LocalDateTime.now());
|
personBoard.setStaffId(staff.getId());
|
personBoard.setWorkstationId(dutyRecordDTO.getWorkstationId());
|
personBoard.setDutyRecordId(dutyRecordDTO.getId());
|
personBoardMapper.insert(personBoard);
|
}
|
}
|
return getDtoById(dutyRecordDTO.getId());
|
}
|
|
@Override
|
public boolean updateById(DutyRecord dutyRecord) {
|
checkDutyMonth(dutyRecord);
|
DutyRecord record = this.getById(dutyRecord.getId());
|
Workstation workstation = workstationMapper.selectById(record.getWorkstationId());
|
dutyRecord.setWorkCenter(workstation.getWorkCenter());
|
// 如果是退回，需要删除对应的工资计算结果；刷新待处理字段
|
if (dutyRecord.getIsConfirm() != null && dutyRecord.getIsConfirm() == false) {
|
computationMapper.delete(Wrappers.<Computation>lambdaQuery().eq(Computation::getDutyRecordId, dutyRecord.getId()));
|
baseMapper.refreshPendingById(dutyRecord.getId());
|
}
|
return SqlHelper.retBool(baseMapper.updateById(dutyRecord));
|
}
|
|
private void checkDutyMonth(DutyRecord dutyRecord) {
|
LocalDateTime dutyDate = dutyRecord.getDutyDate();
|
if (dutyDate == null) {
|
return;
|
}
|
LambdaQueryWrapper<DutyRecord> wrapper = Wrappers.<DutyRecord>lambdaQuery().eq(DutyRecord::getWorkstationId, dutyRecord.getWorkstationId())
|
.eq(DutyRecord::getShiftId, dutyRecord.getShiftId()).eq(DutyRecord::getNowDutyDate, FORMATTER_DATE.format(dutyDate));
|
if (dutyRecord.getId() != null) {
|
wrapper = wrapper.ne(DutyRecord::getId, dutyRecord.getId());
|
} else {
|
dutyRecord.setDutyNo(dutyRecordNumberGenerator.generateNumberWithPrefix(DutyRecord.DIGIT, DutyRecord.PREFIX, DutyRecord::getDutyNo));
|
}
|
DutyRecord dutyRecordSame = baseMapper.selectOne(wrapper);
|
if (dutyRecordSame != null) {
|
throw new RuntimeException("该班次与" + dutyRecordSame.getDutyNo() + "重复");
|
}
|
dutyRecord.setNowDutyDate(FORMATTER_DATE.format(dutyDate));
|
// 计算该班次属于那个统计月度
|
if (dutyDate.getDayOfMonth() >= DutyRecord.START_DATE) {
|
dutyDate = dutyDate.plusMonths(1);
|
}
|
dutyRecord.setDutyMonth(FORMATTER_MONTH.format(dutyDate));
|
// 是否夜班
|
Shift shift = shiftMapper.selectById(dutyRecord.getShiftId());
|
if (shift != null) {
|
dutyRecord.setIsNight(shift.getIsNight());
|
}
|
}
|
|
@Override
|
public List<PersonBoardDTO> getPersonByDutyRecordId(Long id) {
|
return personBoardMapper.getPersonByDutyRecordId(id);
|
}
|
|
@Override
|
public List<PersonBoardDTO> getPersonByDutyRecordIdList(List<Long> idList) {
|
return personBoardMapper.getPersonByDutyRecordIdList(idList);
|
}
|
|
@Override
|
public List<ProductOutputDTO> getOutputByDutyRecordId(Long id) {
|
return productOutputMapper.getOutputByDutyRecordId(id);
|
}
|
|
@Override
|
public List<ProductOutputDTO> getOutputByDutyRecordIdList(List<Long> idList) {
|
List<ProductOutputDTO> result = productOutputMapper.getOutputByDutyRecordIdList(idList);
|
List<ProductOutputDTO> segmentationOutputList = segmentationTaskRecordMapper.getOutputListByDutyRecordId(idList);
|
if (CollectionUtil.isNotEmpty(segmentationOutputList)) {
|
result.addAll(segmentationOutputList);
|
}
|
return result;
|
}
|
|
@Override
|
public DutyRecordDTO getDutyRecordByWorkstationId(Long id) {
|
return baseMapper.getDutyRecordByWorkstationId(id);
|
}
|
|
@Override
|
public List<DutyRecordDTO> getOperatingRecord(Long id) {
|
return baseMapper.getOperatingRecord(id);
|
}
|
|
@Override
|
public DutyRecordDTO getDtoById(Long id) {
|
DutyRecordDTO dutyRecordDTO = baseMapper.getDtoById(id);
|
// 暂时人员不要带出
|
/*dutyRecordDTO.setPersonList(personBoardMapper.getPersonByDutyRecordId(id));*/
|
return dutyRecordDTO;
|
}
|
|
@Override
|
public boolean deleteById(Long id) {
|
try {
|
baseMapper.deleteById(id);
|
personBoardMapper.delete(Wrappers.<PersonBoard>query().lambda().eq(PersonBoard::getDutyRecordId, id));
|
artificialInformationMapper.delete(Wrappers.<ArtificialInformation>query().lambda().eq(ArtificialInformation::getDutyRecordId, id));
|
} catch (Exception e) {
|
throw new RuntimeException("上班记录删除失败");
|
}
|
return true;
|
}
|
|
@Override
|
public List<DutyRecordData> exportList(QueryWrapper<DutyRecordDTO> gen) {
|
return baseMapper.exportList(gen);
|
}
|
|
@Override
|
public List<PersonBoardData> exportPersonBoard(QueryWrapper<DutyRecordDTO> gen) {
|
return baseMapper.exportPersonBoard(gen);
|
}
|
|
@Override
|
public IPage getArtificialRecord(Page page, QueryWrapper<DutyRecordDTO> gen) {
|
return baseMapper.getArtificialRecord(page, gen);
|
}
|
|
@Override
|
public List<AttendanceData> exportAttendance(QueryWrapper<DutyRecordDTO> gen) {
|
return baseMapper.exportAttendance(gen);
|
}
|
|
@Override
|
public void refreshPendingById(Long dutyRecordId) {
|
if (dutyRecordId != null) {
|
baseMapper.refreshPendingById(dutyRecordId);
|
}
|
}
|
|
@Override
|
public List<Workstation> getWorkstationList(String workCenter) {
|
List<Workstation> workstationList = workstationMapper.selectList(Wrappers.<Workstation>query().lambda()
|
.eq(Workstation::getWorkCenter, workCenter)
|
.orderByAsc(Workstation::getWorkstationNo));
|
return workstationList;
|
}
|
|
@Override
|
public List<DutyRecordDTO> getUnsubmitWorkstationList(String workCenter) {
|
List<DutyRecordDTO> recordDTOList = baseMapper.selectUnsubmitWorkstation(workCenter);
|
return recordDTOList;
|
}
|
|
@Override
|
public List<DutyRecordDTO> batchSave(DutyRecordDTO dutyRecordDTO) {
|
ZttUser user = SecurityUtils.getUser();
|
List<DutyRecordDTO> result = new ArrayList<>();
|
// 找出工作站
|
Workstation workstation = workstationMapper.selectById(dutyRecordDTO.getWorkstationId());
|
// 找出该工作站的工作重心关联的所有工作站
|
//List<Workstation> workstationList = workstationMapper.selectList(Wrappers.<Workstation>query().lambda().eq(Workstation::getWorkCenter, workstation.getWorkCenter()));
|
// 循环创建班次记录
|
for (Long workstationId : dutyRecordDTO.getWorkstationIdList()) {
|
DutyRecordDTO newRecord = new DutyRecordDTO();
|
BeanUtils.copyProperties(dutyRecordDTO, newRecord);
|
newRecord.setId(null);
|
newRecord.setDeptId(user.getDeptId());
|
newRecord.setIsConfirm(false);
|
newRecord.setIsSubmit(false);
|
newRecord.setWorkstationId(workstationId);
|
LocalDateTime dutyDate = newRecord.getDutyDate();
|
if (dutyDate != null) {
|
LambdaQueryWrapper<DutyRecord> wrapper = Wrappers.<DutyRecord>lambdaQuery().eq(DutyRecord::getWorkstationId, newRecord.getWorkstationId())
|
.eq(DutyRecord::getShiftId, newRecord.getShiftId()).eq(DutyRecord::getNowDutyDate, FORMATTER_DATE.format(dutyDate));
|
if (newRecord.getId() != null) {
|
wrapper = wrapper.ne(DutyRecord::getId, newRecord.getId());
|
} else {
|
newRecord.setDutyNo(dutyRecordNumberGenerator.generateNumberWithPrefix(DutyRecord.DIGIT, DutyRecord.PREFIX, DutyRecord::getDutyNo));
|
}
|
DutyRecord dutyRecordSame = baseMapper.selectOne(wrapper);
|
if (dutyRecordSame != null) {
|
continue;
|
}
|
newRecord.setNowDutyDate(FORMATTER_DATE.format(dutyDate));
|
// 计算该班次属于那个统计月度
|
if (dutyDate.getDayOfMonth() >= DutyRecord.START_DATE) {
|
dutyDate = dutyDate.plusMonths(1);
|
}
|
newRecord.setDutyMonth(FORMATTER_MONTH.format(dutyDate));
|
// 是否夜班
|
Shift shift = shiftMapper.selectById(newRecord.getShiftId());
|
if (shift != null) {
|
newRecord.setIsNight(shift.getIsNight());
|
}
|
}
|
// 1.保存主表
|
newRecord.setWorkCenter(workstation.getWorkCenter());
|
baseMapper.insert(newRecord);
|
// 2.保存默认班组人员
|
if (newRecord.getCrewId() != null) {
|
List<Staff> staffs = joinCrewStaffMapper.selectStaffByCrewId(newRecord.getCrewId());
|
for (Staff staff : staffs) {
|
PersonBoard personBoard = new PersonBoard();
|
personBoard.setWorkTime(LocalDateTime.now());
|
personBoard.setStaffId(staff.getId());
|
personBoard.setWorkstationId(newRecord.getWorkstationId());
|
personBoard.setDutyRecordId(newRecord.getId());
|
personBoardMapper.insert(personBoard);
|
}
|
}
|
Workstation update = new Workstation();
|
update.setId(newRecord.getWorkstationId());
|
update.setDutyRecordId(newRecord.getId());
|
workstationMapper.updateById(update);
|
result.add(newRecord);
|
}
|
|
return result;
|
}
|
|
@Override
|
public List<ShiftWageData> exportShiftWage(QueryWrapper<DutyRecordDTO> gen){
|
return baseMapper.exportShiftWage(gen);
|
}
|
|
@Override
|
public List<ProductOutputDTO> getStaffProductionWage(Long dutyRecordId){
|
return productOutputMapper.getStaffProductionWage(dutyRecordId);
|
}
|
|
@Override
|
public boolean sendUnsubmitMsg() {
|
//统计所有未提交班次记录
|
Integer unsubmitRecordCount = baseMapper
|
.selectCount(Wrappers.<DutyRecord>lambdaQuery().eq(DutyRecord::getIsSubmit, false));
|
//每条消息最多发送100条记录，计算发送多少条消息
|
int page = unsubmitRecordCount / 100 + (unsubmitRecordCount % 100 == 0 ? 0 : 1);
|
List<DutyRecordDTO> dutyRecordDTOList = null;
|
Long startIndex = 0L;
|
for(int i=0;i<page;i++){
|
dutyRecordDTOList = baseMapper.selectUnsubmitRecord(startIndex);
|
startIndex = startIndex + 100;
|
StringBuilder msgBuilder = new StringBuilder();
|
msgBuilder.append("MES消息提醒：").append("\n").append("您好，目前有如下机台未提交班次记录，请及时提交：").append("\n");
|
for (DutyRecordDTO dutyRecordDTO:dutyRecordDTOList) {
|
msgBuilder.append("班次：").append(dutyRecordDTO.getDutyNo()).append(" ")
|
.append("工作站：").append(dutyRecordDTO.getWorkstationName()).append("\n");
|
}
|
WxCpUtils.informWebHook(unsubmitDutyRecordConfig.getDutyRecordWebhook(), msgBuilder.toString(),null);
|
}
|
|
return true;
|
}
|
|
@Override
|
public boolean batchSubmit(List<Long> ids) {
|
List<DutyRecord> dutyRecordList = new ArrayList<>();
|
for (Long id:ids) {
|
DutyRecord dutyRecord = new DutyRecord();
|
dutyRecord.setId(id);
|
dutyRecord.setIsSubmit(true);
|
dutyRecordList.add(dutyRecord);
|
}
|
boolean result = updateBatchById(dutyRecordList);
|
dutyRecordList.stream().forEach(a-> backUtils.backDutyRecordById(a.getId(),"提交"));
|
return result;
|
}
|
|
@Override
|
public boolean updateBatch(List<DutyRecordDTO> dutyRecordDTOList) {
|
List<DutyRecord> dutyRecordList = new ArrayList<>();
|
for (DutyRecordDTO dutyRecordDTO:dutyRecordDTOList) {
|
DutyRecord dutyRecord = new DutyRecord();
|
BeanUtils.copyProperties(dutyRecordDTO,dutyRecord);
|
dutyRecordList.add(dutyRecord);
|
}
|
this.updateBatchById(dutyRecordList);
|
return true;
|
}
|
}
|
