package cn.iocoder.yudao.module.mes.service.pro.workrecord;
|
|
import cn.hutool.core.util.ObjUtil;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.module.mes.controller.admin.pro.workrecord.vo.MesProWorkRecordLogPageReqVO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workrecord.MesProWorkRecordDO;
|
import cn.iocoder.yudao.module.mes.dal.dataobject.pro.workrecord.MesProWorkRecordLogDO;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.workrecord.MesProWorkRecordLogMapper;
|
import cn.iocoder.yudao.module.mes.dal.mysql.pro.workrecord.MesProWorkRecordMapper;
|
import cn.iocoder.yudao.module.mes.enums.pro.MesProWorkRecordTypeEnum;
|
import cn.iocoder.yudao.module.mes.service.md.workstation.MesMdWorkstationService;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.validation.annotation.Validated;
|
|
import java.time.LocalDateTime;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.WORK_RECORD_ALREADY_CLOCK_IN;
|
import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.WORK_RECORD_NOT_CLOCK_IN;
|
|
/**
|
* MES 工作记录 Service 实现类
|
*
|
* @author 芋道源码
|
*/
|
@Service
|
@Validated
|
public class MesProWorkRecordServiceImpl implements MesProWorkRecordService {
|
|
@Resource
|
private MesProWorkRecordLogMapper workRecordLogMapper;
|
@Resource
|
private MesProWorkRecordMapper workRecordMapper;
|
|
@Resource
|
private MesMdWorkstationService workstationService;
|
|
@Override
|
public PageResult<MesProWorkRecordLogDO> getWorkRecordLogPage(MesProWorkRecordLogPageReqVO pageReqVO) {
|
return workRecordLogMapper.selectPage(pageReqVO);
|
}
|
|
@Override
|
public MesProWorkRecordLogDO getWorkRecordLog(Long id) {
|
return workRecordLogMapper.selectById(id);
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Long clockInWorkRecord(Long userId, Long workstationId) {
|
// 1.1 校验工作站是否存在
|
workstationService.validateWorkstationExists(workstationId);
|
// 1.2 校验是否重复上线
|
MesProWorkRecordDO record = workRecordMapper.selectByUserId(userId);
|
if (record != null && ObjUtil.equal(MesProWorkRecordTypeEnum.CLOCK_IN.getType(), record.getType())) {
|
throw exception(WORK_RECORD_ALREADY_CLOCK_IN);
|
}
|
|
// 2. 写入上线流水
|
MesProWorkRecordLogDO log = MesProWorkRecordLogDO.builder().userId(userId)
|
.workstationId(workstationId).type(MesProWorkRecordTypeEnum.CLOCK_IN.getType()).build();
|
workRecordLogMapper.insert(log);
|
|
// 3. 更新用户工作站状态：如果没有记录，或先删除旧记录
|
if (record == null) {
|
record = MesProWorkRecordDO.builder().userId(userId).workstationId(workstationId)
|
.type(MesProWorkRecordTypeEnum.CLOCK_IN.getType()).clockInTime(LocalDateTime.now()).build();
|
workRecordMapper.insert(record);
|
} else {
|
workRecordMapper.updateById(new MesProWorkRecordDO().setId(record.getId())
|
.setWorkstationId(workstationId).setType(MesProWorkRecordTypeEnum.CLOCK_IN.getType())
|
.setClockInTime(LocalDateTime.now()));
|
}
|
return log.getId();
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public Long clockOutWorkRecord(Long userId) {
|
// 1. 校验是否有状态
|
MesProWorkRecordDO record = workRecordMapper.selectByUserId(userId);
|
if (record == null || ObjUtil.notEqual(MesProWorkRecordTypeEnum.CLOCK_IN.getType(), record.getType())) {
|
throw exception(WORK_RECORD_NOT_CLOCK_IN);
|
}
|
|
// 2. 写入下线流水
|
MesProWorkRecordLogDO log = MesProWorkRecordLogDO.builder().userId(userId)
|
.workstationId(record.getWorkstationId()).type(MesProWorkRecordTypeEnum.CLOCK_OUT.getType())
|
.build();
|
workRecordLogMapper.insert(log);
|
|
// 3. 更新用户状态
|
workRecordMapper.updateById(new MesProWorkRecordDO().setId(record.getId())
|
.setType(MesProWorkRecordTypeEnum.CLOCK_OUT.getType()).setClockOutTime(LocalDateTime.now()));
|
return log.getId();
|
}
|
|
@Override
|
public MesProWorkRecordDO getWorkRecord(Long userId) {
|
return workRecordMapper.selectByUserId(userId);
|
}
|
|
}
|
