新建

李林

2023-10-07 658d4927d468c47208fd012d9128b09249c07eff

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
package com.chinaztt.mes.production.service.impl;
 
import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.production.dto.HandymanTableDTO;
import com.chinaztt.mes.production.entity.HandymanTable;
import com.chinaztt.mes.production.entity.JoinPersonHandyman;
import com.chinaztt.mes.production.mapper.HandymanTableMapper;
import com.chinaztt.mes.production.mapper.JoinPersonHandymanMapper;
import com.chinaztt.mes.production.service.HandymanTableService;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import java.util.stream.Collectors;
 
/**
 * 杂工登记
 *
 * @author cxf
 * @date 2020-11-04 15:37:05
 */
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class HandymanTableServiceImpl extends ServiceImpl<HandymanTableMapper, HandymanTable> implements HandymanTableService {
 
    private JoinPersonHandymanMapper joinPersonHandymanMapper;
 
    @Override
    public boolean saveDto(HandymanTableDTO handymanTableDTO) {
        try {
            // 1.保存杂工
            baseMapper.insert(handymanTableDTO);
            // 2.保存关系表
            saveJoinPersonHandyman(handymanTableDTO);
        } catch (Exception e) {
            throw new RuntimeException("杂工登记失败");
        }
        return true;
    }
 
    @Override
    public boolean updateDtoById(HandymanTableDTO handymanTableDTO) {
        try {
            // 1.更新杂工
            baseMapper.updateById(handymanTableDTO);
            // 2.删除关系表
            joinPersonHandymanMapper.delete(Wrappers.<JoinPersonHandyman>lambdaQuery().eq(JoinPersonHandyman::getHandymanTableId, handymanTableDTO.getId()));
            // 3.保存关系表
            saveJoinPersonHandyman(handymanTableDTO);
        } catch (Exception e) {
            throw new RuntimeException("修改杂工失败");
        }
        return true;
    }
 
    public void saveJoinPersonHandyman(HandymanTableDTO handymanTableDTO) {
        if (CollectionUtil.isNotEmpty(handymanTableDTO.getPersonBoardIds())) {
            handymanTableDTO.getPersonBoardIds().forEach(personBoardId -> {
                JoinPersonHandyman joinPersonHandyman = new JoinPersonHandyman();
                joinPersonHandyman.setHandymanTableId(handymanTableDTO.getId());
                joinPersonHandyman.setPersonBoardId(personBoardId);
                joinPersonHandymanMapper.insert(joinPersonHandyman);
            });
        }
    }
 
    @Override
    public HandymanTableDTO getDtoById(Long id) {
        HandymanTableDTO handymanTableDTO = baseMapper.getDtoById(id);
        handymanTableDTO.setPersonBoardIds(joinPersonHandymanMapper.selectList(Wrappers.<JoinPersonHandyman>lambdaQuery().eq(JoinPersonHandyman::getHandymanTableId, id))
                .stream().map(JoinPersonHandyman::getPersonBoardId).collect(Collectors.toList()));
        return handymanTableDTO;
    }
 
 
    @Override
    public boolean deleteById(Long id) {
        try {
            // 1.删除杂工
            baseMapper.deleteById(id);
            // 2.删除关系表
            joinPersonHandymanMapper.delete(Wrappers.<JoinPersonHandyman>lambdaQuery().eq(JoinPersonHandyman::getHandymanTableId, id));
        } catch (Exception e) {
            throw new RuntimeException("删除杂工失败");
        }
        return true;
    }
}