| | |
| | | |
| | | import cn.hutool.core.collection.CollUtil; |
| | | import cn.hutool.core.date.DateUtil; |
| | | import cn.hutool.core.util.StrUtil; |
| | | import cn.iocoder.yudao.framework.common.exception.ServiceException; |
| | | import cn.iocoder.yudao.framework.common.pojo.PageResult; |
| | | import cn.iocoder.yudao.framework.common.util.object.BeanUtils; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenancecalendar.vo.MesProMaintenanceCalendarImportExcelVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenancecalendar.vo.MesProMaintenanceCalendarImportRespVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenancecalendar.vo.MesProMaintenanceCalendarPageReqVO; |
| | | import cn.iocoder.yudao.module.mes.controller.admin.pro.maintenancecalendar.vo.MesProMaintenanceCalendarSaveReqVO; |
| | | import cn.iocoder.yudao.module.mes.dal.dataobject.pro.maintenancecalendar.MesProMaintenanceCalendarDO; |
| | |
| | | import org.springframework.validation.annotation.Validated; |
| | | |
| | | import java.time.LocalDateTime; |
| | | import java.util.ArrayList; |
| | | import java.util.Collection; |
| | | import java.util.LinkedHashMap; |
| | | import java.util.List; |
| | | import java.util.Objects; |
| | | import java.util.concurrent.atomic.AtomicInteger; |
| | | |
| | | import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; |
| | | import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_PRO_MAINTENANCE_CALENDAR_IMPORT_LIST_IS_EMPTY; |
| | | import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_PRO_MAINTENANCE_CALENDAR_NOT_EXISTS; |
| | | import static cn.iocoder.yudao.module.mes.enums.ErrorCodeConstants.MES_PRO_MAINTENANCE_CALENDAR_NOT_PREPARE; |
| | | |
| | |
| | | calendarMapper.updateById(new MesProMaintenanceCalendarDO().setId(id).setStatus(STATUS_PUBLISHED)); |
| | | } |
| | | |
| | | @Override |
| | | @Transactional(rollbackFor = Exception.class) |
| | | public MesProMaintenanceCalendarImportRespVO importCalendarList(List<MesProMaintenanceCalendarImportExcelVO> importList, |
| | | boolean updateSupport) { |
| | | if (CollUtil.isEmpty(importList)) { |
| | | throw exception(MES_PRO_MAINTENANCE_CALENDAR_IMPORT_LIST_IS_EMPTY); |
| | | } |
| | | MesProMaintenanceCalendarImportRespVO respVO = MesProMaintenanceCalendarImportRespVO.builder() |
| | | .createNames(new ArrayList<>()).updateNames(new ArrayList<>()) |
| | | .failureNames(new LinkedHashMap<>()).build(); |
| | | AtomicInteger index = new AtomicInteger(1); |
| | | importList.forEach(importItem -> { |
| | | int currentIndex = index.getAndIncrement(); |
| | | String key = StrUtil.blankToDefault(importItem.getName(), "第 " + currentIndex + " 行"); |
| | | if (StrUtil.isBlank(importItem.getName())) { |
| | | respVO.getFailureNames().put(key, "日历名称不能为空"); |
| | | return; |
| | | } |
| | | if (importItem.getWorkDate() == null) { |
| | | respVO.getFailureNames().put(key, "作业日期不能为空"); |
| | | return; |
| | | } |
| | | MesProMaintenanceCalendarSaveReqVO reqVO = BeanUtils.toBean(importItem, MesProMaintenanceCalendarSaveReqVO.class); |
| | | MesProMaintenanceCalendarDO existCalendar = calendarMapper.selectByName(importItem.getName()); |
| | | try { |
| | | if (existCalendar == null) { |
| | | reqVO.setId(null); |
| | | createCalendar(reqVO); |
| | | respVO.getCreateNames().add(importItem.getName()); |
| | | } else if (updateSupport) { |
| | | reqVO.setId(existCalendar.getId()); |
| | | updateCalendar(reqVO); |
| | | respVO.getUpdateNames().add(importItem.getName()); |
| | | } else { |
| | | respVO.getFailureNames().put(key, "日历名称已存在"); |
| | | } |
| | | } catch (ServiceException e) { |
| | | respVO.getFailureNames().put(key, e.getMessage()); |
| | | } |
| | | }); |
| | | return respVO; |
| | | } |
| | | |
| | | private String generateCode() { |
| | | return "RL" + DateUtil.format(LocalDateTime.now(), "yyyyMMddHHmmss") |
| | | + String.format("%03d", (int) (Math.random() * 1000)); |