feat 功能修正

xiaoyi

4 天以前 c9d4ba37c1f681b12e24014013fa9a28734bad42

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
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<Long> 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<MesProMaintenanceCalendarDO> 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;
    }
 
}