package cn.iocoder.yudao.module.mes.service.pro.maintenancecalendar; import cn.hutool.core.collection.CollUtil; import cn.hutool.core.date.DateUtil; 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.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 cn.iocoder.yudao.module.mes.dal.mysql.pro.maintenancecalendar.MesProMaintenanceCalendarMapper; 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 java.util.Collection; import java.util.Objects; import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception; 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; /** * MES 电力运维日历 Service 实现类 * * @author xiaoyi */ @Service @Validated public class MesProMaintenanceCalendarServiceImpl implements MesProMaintenanceCalendarService { private static final Integer STATUS_DRAFT = 0; private static final Integer STATUS_PUBLISHED = 10; @Resource private MesProMaintenanceCalendarMapper calendarMapper; @Override @Transactional(rollbackFor = Exception.class) public Long createCalendar(MesProMaintenanceCalendarSaveReqVO createReqVO) { MesProMaintenanceCalendarDO calendar = BeanUtils.toBean(createReqVO, MesProMaintenanceCalendarDO.class, req -> req .setCode(generateCode()) .setStatus(STATUS_DRAFT)); calendarMapper.insert(calendar); return calendar.getId(); } @Override @Transactional(rollbackFor = Exception.class) public void updateCalendar(MesProMaintenanceCalendarSaveReqVO updateReqVO) { MesProMaintenanceCalendarDO oldCalendar = validateCalendarCanModify(updateReqVO.getId()); MesProMaintenanceCalendarDO updateObj = BeanUtils.toBean(updateReqVO, MesProMaintenanceCalendarDO.class, req -> req .setCode(oldCalendar.getCode()) .setStatus(oldCalendar.getStatus())); calendarMapper.updateById(updateObj); } @Override @Transactional(rollbackFor = Exception.class) public void deleteCalendar(Collection ids) { if (CollUtil.isEmpty(ids)) { return; } for (Long id : ids) { validateCalendarCanModify(id); calendarMapper.deleteById(id); } } @Override public MesProMaintenanceCalendarDO getCalendar(Long id) { return calendarMapper.selectById(id); } @Override public PageResult getCalendarPage(MesProMaintenanceCalendarPageReqVO pageReqVO) { return calendarMapper.selectPage(pageReqVO); } @Override @Transactional(rollbackFor = Exception.class) public void publishCalendar(Long id) { MesProMaintenanceCalendarDO calendar = validateCalendarExists(id); if (!Objects.equals(calendar.getStatus(), STATUS_DRAFT)) { throw exception(MES_PRO_MAINTENANCE_CALENDAR_NOT_PREPARE); } calendarMapper.updateById(new MesProMaintenanceCalendarDO().setId(id).setStatus(STATUS_PUBLISHED)); } private String generateCode() { return "RL" + DateUtil.format(LocalDateTime.now(), "yyyyMMddHHmmss") + String.format("%03d", (int) (Math.random() * 1000)); } private MesProMaintenanceCalendarDO validateCalendarExists(Long id) { MesProMaintenanceCalendarDO calendar = calendarMapper.selectById(id); if (calendar == null) { throw exception(MES_PRO_MAINTENANCE_CALENDAR_NOT_EXISTS); } return calendar; } private MesProMaintenanceCalendarDO validateCalendarCanModify(Long id) { MesProMaintenanceCalendarDO calendar = validateCalendarExists(id); if (!Objects.equals(calendar.getStatus(), STATUS_DRAFT)) { throw exception(MES_PRO_MAINTENANCE_CALENDAR_NOT_PREPARE); } return calendar; } }