luodangjia
2024-12-25 488b9158fc873d228c2c0106050ece8272a49653
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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
package com.sinata.system.service.impl;
 
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.sinata.common.core.domain.entity.SysDictData;
import com.sinata.common.entity.PageDTO;
import com.sinata.common.exception.ServiceException;
import com.sinata.common.utils.BeanUtils;
import com.sinata.common.utils.CollUtils;
import com.sinata.common.utils.StringUtils;
import com.sinata.system.domain.MwAttachment;
import com.sinata.system.domain.MwProtectionTask;
import com.sinata.system.domain.MwProtectionTaskEquipment;
import com.sinata.system.domain.dto.MwProtectionTaskDTO;
import com.sinata.system.domain.query.MwProtectionTaskQuery;
import com.sinata.system.domain.vo.MwAttachmentVO;
import com.sinata.system.domain.vo.MwProtectionTaskEquipmentVO;
import com.sinata.system.domain.vo.MwProtectionTaskVO;
import com.sinata.system.enums.AttachmentTypeEnum;
import com.sinata.system.mapper.MwProtectionTaskMapper;
import com.sinata.system.service.ISysDictDataService;
import com.sinata.system.service.MwAttachmentService;
import com.sinata.system.service.MwProtectionTaskEquipmentService;
import com.sinata.system.service.MwProtectionTaskService;
import com.sinata.system.service.SysDepartmentService;
import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.util.List;
import java.util.Objects;
 
/**
 * <p>
 * 防护作业 服务实现类
 * </p>
 *
 * @author mitao
 * @since 2024-12-02
 */
@Service
@RequiredArgsConstructor
public class MwProtectionTaskServiceImpl extends ServiceImpl<MwProtectionTaskMapper, MwProtectionTask> implements MwProtectionTaskService {
    private final SysDepartmentService sysDepartmentService;
    private final MwProtectionTaskEquipmentService mwProtectionTaskEquipmentService;
    private final MwAttachmentService mwAttachmentService;
    private final ISysDictDataService sysDictDataService;
 
    /**
     * 防护作业分页列表
     *
     * @param query
     * @return
     */
    @Override
    public PageDTO<MwProtectionTaskVO> pageList(MwProtectionTaskQuery query) {
        String treeCode = sysDepartmentService.getTreeCodeByDepartmentId(query.getDepartmentId());
        if (StringUtils.isBlank(treeCode)) {
            return PageDTO.empty(0L, 0L);
        }
        Page<MwProtectionTaskVO> page = baseMapper.pageList(new Page<>(query.getPageCurr(), query.getPageSize()), query, treeCode);
        return PageDTO.of(page);
    }
 
    /**
     * 防护作业详情
     *
     * @param id
     * @return
     */
    @Override
    public MwProtectionTaskVO detail(Long id) {
        MwProtectionTask task = getById(id);
        MwProtectionTaskVO mwProtectionTaskVO = null;
        if (Objects.nonNull(task)) {
            mwProtectionTaskVO = BeanUtils.copyBean(task, MwProtectionTaskVO.class);
            //查询器具列表
            List<MwProtectionTaskEquipmentVO> protectionTaskEquipmentVOList = mwProtectionTaskEquipmentService.getUsageEquipmentListByTaskId(id);
            mwProtectionTaskVO.setEquipmentList(protectionTaskEquipmentVOList);
            //查询附件列表
            List<MwAttachment> list = mwAttachmentService.lambdaQuery().eq(MwAttachment::getType, AttachmentTypeEnum.PROTECTION_TASK.getCode()).eq(MwAttachment::getTargetId, id).list();
            mwProtectionTaskVO.setAttachmentList(BeanUtils.copyToList(list, MwAttachmentVO.class));
            return mwProtectionTaskVO;
        }
        return mwProtectionTaskVO;
    }
 
    /**
     * 新增防护作业
     *
     * @param dto
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void add(MwProtectionTaskDTO dto) {
        MwProtectionTask mwProtectionTask = BeanUtils.copyBean(dto, MwProtectionTask.class);
        SysDictData dictData = sysDictDataService.lambdaQuery().eq(SysDictData::getDictCode, dto.getProtectiveWorkType()).one();
        if (Objects.nonNull(dictData)) {
            mwProtectionTask.setProtectiveWorkTypeStr(dictData.getDictLabel());
        }
        save(mwProtectionTask);
        //保存使用器具列表
        List<MwProtectionTaskEquipment> mwProtectionTaskEquipments = BeanUtils.copyToList(dto.getEquipmentList(), MwProtectionTaskEquipment.class);
        mwProtectionTaskEquipments.forEach(equipment -> equipment.setProtectionTaskId(mwProtectionTask.getId()));
        mwProtectionTaskEquipmentService.saveBatch(mwProtectionTaskEquipments);
        //保存附件
        if (CollUtils.isNotEmpty(dto.getAttachmentList())) {
            List<MwAttachment> mwAttachments = BeanUtils.copyToList(dto.getAttachmentList(), MwAttachment.class);
            mwAttachments.forEach(attachment -> {
                attachment.setTargetId(mwProtectionTask.getId());
                attachment.setType(AttachmentTypeEnum.PROTECTION_TASK.getCode());
            });
            mwAttachmentService.saveBatch(mwAttachments);
        }
    }
 
    /**
     * 编辑防护作业
     *
     * @param dto
     * @return
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void edit(MwProtectionTaskDTO dto) {
        if (Objects.isNull(dto.getId())) {
            throw new ServiceException("防护作业id不能为空");
        }
        MwProtectionTask mwProtectionTask = BeanUtils.copyBean(dto, MwProtectionTask.class);
        SysDictData dictData = sysDictDataService.lambdaQuery().eq(SysDictData::getDictCode, dto.getProtectiveWorkType()).one();
        if (Objects.nonNull(dictData)) {
            mwProtectionTask.setProtectiveWorkTypeStr(dictData.getDictLabel());
        }
        updateById(mwProtectionTask);
        //删除原有的使用器具
        mwProtectionTaskEquipmentService.lambdaUpdate().eq(MwProtectionTaskEquipment::getProtectionTaskId, dto.getId()).remove();
        //保存使用器具列表
        List<MwProtectionTaskEquipment> mwProtectionTaskEquipments = BeanUtils.copyToList(dto.getEquipmentList(), MwProtectionTaskEquipment.class);
        mwProtectionTaskEquipments.forEach(equipment -> equipment.setProtectionTaskId(mwProtectionTask.getId()));
        mwProtectionTaskEquipmentService.saveBatch(mwProtectionTaskEquipments);
        //保存附件
        if (CollUtils.isNotEmpty(dto.getAttachmentList())) {
            //删除原有的附件
            mwAttachmentService.lambdaUpdate().eq(MwAttachment::getType, AttachmentTypeEnum.PROTECTION_TASK.getCode()).eq(MwAttachment::getTargetId, dto.getId()).remove();
            List<MwAttachment> mwAttachments = BeanUtils.copyToList(dto.getAttachmentList(), MwAttachment.class);
            mwAttachments.forEach(attachment -> {
                attachment.setTargetId(mwProtectionTask.getId());
                attachment.setType(AttachmentTypeEnum.PROTECTION_TASK.getCode());
            });
            mwAttachmentService.saveBatch(mwAttachments);
        }
    }
 
    /**
     * 删除
     *
     * @param id
     */
    @Override
    @Transactional(rollbackFor = Exception.class)
    public void delete(Long id) {
        removeById(id);
        //删除使用器具
        mwProtectionTaskEquipmentService.lambdaUpdate().eq(MwProtectionTaskEquipment::getProtectionTaskId, id).remove();
        //删除附件
        mwAttachmentService.lambdaUpdate().eq(MwAttachment::getType, AttachmentTypeEnum.PROTECTION_TASK.getCode()).eq(MwAttachment::getTargetId, id).remove();
    }
}