feat: 添加抽检相关字段和注释

yuan

7 天以前 21e1f6344369fc0775d1d10642214ab0fd993359

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
package com.ruoyi.safe.service.impl;
 
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.project.system.domain.SysUser;
import com.ruoyi.project.system.mapper.SysUserMapper;
import com.ruoyi.safe.pojo.SafeFacilityLedger;
import com.ruoyi.safe.pojo.SafeFacilityRectification;
import com.ruoyi.safe.mapper.SafeFacilityRectificationMapper;
import com.ruoyi.safe.service.SafeFacilityLedgerService;
import com.ruoyi.safe.service.SafeFacilityRectificationService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
 
import java.io.Serializable;
import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;
 
/**
 * <p>
 * 安全生产--安全设施整改 服务实现类
 * </p>
 *
 * @author 芯导软件（江苏）有限公司
 * @since 2026-06-29
 */
@Service
public class SafeFacilityRectificationServiceImpl extends ServiceImpl<SafeFacilityRectificationMapper, SafeFacilityRectification> implements SafeFacilityRectificationService {
 
    @Autowired
    private SafeFacilityLedgerService safeFacilityLedgerService;
    @Autowired
    private SysUserMapper sysUserMapper;
 
    @Override
    public SafeFacilityRectification getById(Serializable id) {
        SafeFacilityRectification record = super.getById(id);
        if (record != null) {
            fillExtraInfo(Collections.singletonList(record));
        }
        return record;
    }
 
    @Override
    public IPage<SafeFacilityRectification> pageSafeFacilityRectification(Page page, SafeFacilityRectification safeFacilityRectification) {
        IPage<SafeFacilityRectification> result = this.lambdaQuery()
                .eq(safeFacilityRectification.getInspectionId() != null, SafeFacilityRectification::getInspectionId, safeFacilityRectification.getInspectionId())
                .eq(safeFacilityRectification.getFacilityId() != null, SafeFacilityRectification::getFacilityId, safeFacilityRectification.getFacilityId())
                .eq(safeFacilityRectification.getStatus() != null, SafeFacilityRectification::getStatus, safeFacilityRectification.getStatus())
                .orderByDesc(SafeFacilityRectification::getCreateTime)
                .page(page);
 
        fillExtraInfo(result.getRecords());
        return result;
    }
 
    private void fillExtraInfo(List<SafeFacilityRectification> records) {
        if (records == null || records.isEmpty()) {
            return;
        }
 
        // 收集设施ID
        List<Integer> facilityIds = records.stream()
                .map(SafeFacilityRectification::getFacilityId)
                .filter(Objects::nonNull)
                .distinct()
                .collect(Collectors.toList());
 
        // 收集所有用户ID（整改责任人 + 验收人）
        Set<Long> userIds = new HashSet<>();
        records.stream().map(SafeFacilityRectification::getRectifyUserId).filter(Objects::nonNull).forEach(id -> userIds.add(id.longValue()));
        records.stream().map(SafeFacilityRectification::getVerifyUserId).filter(Objects::nonNull).forEach(id -> userIds.add(id.longValue()));
 
        // 批量查询设施信息
        Map<Integer, SafeFacilityLedger> facilityMap = facilityIds.isEmpty() ? Collections.emptyMap()
                : safeFacilityLedgerService.listByIds(facilityIds).stream()
                .collect(Collectors.toMap(SafeFacilityLedger::getId, f -> f, (v1, v2) -> v1));
 
        // 批量查询用户信息
        Map<Long, String> userMap = userIds.isEmpty() ? Collections.emptyMap()
                : sysUserMapper.selectUserByIds(new ArrayList<>(userIds)).stream()
                .collect(Collectors.toMap(SysUser::getUserId, SysUser::getNickName, (v1, v2) -> v1));
 
        for (SafeFacilityRectification record : records) {
            if (record.getFacilityId() != null) {
                SafeFacilityLedger facility = facilityMap.get(record.getFacilityId());
                if (facility != null) {
                    record.setFacilityName(facility.getFacilityName());
                }
            }
            if (record.getRectifyUserId() != null) {
                record.setRectifyUserName(userMap.getOrDefault(record.getRectifyUserId().longValue(), "未知用户"));
            }
            if (record.getVerifyUserId() != null) {
                record.setVerifyUserName(userMap.getOrDefault(record.getVerifyUserId().longValue(), "未知用户"));
            }
        }
    }
 
    @Override
    public boolean save(SafeFacilityRectification entity) {
        Long currentUserId = SecurityUtils.getUserId();
        if (entity.getRectifyUserId() == null) {
            entity.setRectifyUserId(currentUserId.intValue());
        }
        if (entity.getStatus() == null) {
            entity.setStatus("待整改");
        }
        return super.save(entity);
    }
 
    @Override
    public boolean updateById(SafeFacilityRectification entity) {
        Long currentUserId = SecurityUtils.getUserId();
        if ("已整改".equals(entity.getStatus()) && entity.getActualTime() == null) {
            entity.setActualTime(LocalDateTime.now());
        }
        if ("已验收".equals(entity.getStatus())) {
            entity.setVerifyUserId(currentUserId.intValue());
            entity.setVerifyTime(LocalDateTime.now());
        }
        return super.updateById(entity);
    }
}