liujie
2026-01-21 2c72fce83175c05c9bdcb43814bc5e08d7fed047
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
170
package com.estate.expert.admin.service.impl;
 
import com.estate.expert.admin.domain.business.device.DeviceType;
import com.estate.expert.admin.domain.common.request.PageReq;
import com.estate.expert.admin.domain.common.response.PageResp;
import com.estate.expert.admin.domain.standard.hierachy.HierarchyData;
import com.estate.expert.admin.domain.standard.hierachy.HierarchyIds;
import com.estate.expert.admin.domain.standard.model.Model;
import com.estate.expert.admin.domain.standard.model.device.DeviceModel;
import com.estate.expert.admin.domain.standard.model.device.DeviceModelDto;
import com.estate.expert.admin.domain.standard.model.device.DeviceModelRepository;
import com.estate.expert.admin.domain.standard.problem.Problem;
import com.estate.expert.admin.domain.user.User;
import com.estate.expert.admin.enums.BizResultCode;
import com.estate.expert.admin.service.HierarchyService;
import com.estate.expert.admin.service.ProblemService;
import com.estate.expert.admin.service.UserService;
import com.estate.expert.admin.service.common.CrudServiceBaseImpl;
import com.estate.expert.admin.service.model.DeviceModelService;
import com.estate.expert.admin.utils.tree.TreeRoot;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
 
import javax.annotation.Resource;
import java.util.List;
import java.util.UUID;
 
/**
 * 设备模型服务实现
 */
@Service
public class DeviceModelServiceImpl extends CrudServiceBaseImpl<DeviceModel, DeviceModelDto> implements DeviceModelService {
 
    @Resource
    private DeviceModelRepository repository;
 
    @Resource
    private HierarchyService hierarchyService;
 
    @Resource
    private ProblemService problemService;
 
    @Resource
    private UserService userService;
 
    @Override
    public PageResp<DeviceModelDto> list(String keyword, PageReq pageReq) {
        var page = repository.findAllByKeyword(keyword, pageReq.toPageRequest());
        return PageResp.of(page, toDto(page.getContent()));
    }
 
    @Transactional
    @Override
    public DeviceModel add(User creator, DeviceModel entity) {
        var hierarchy = hierarchyService.clone(HierarchyIds.DEVICE);
 
        // 对于设备模型而言,需要克隆所有的问题
        problemService.cloneProblems(HierarchyIds.DEVICE, hierarchy.getId());
        // 初始化模型
        entity.setHierarchy(hierarchy);
        entity.setEnable(true);
        entity.setPublished(false);
        if(entity.getVersion() == null)
            entity.setVersion(0);
        return super.add(creator, entity);
    }
 
    @Override
    public DeviceModel copy(User user, Integer id) {
        var model = get(id);
        BizResultCode.ENTITY_NOT_FOUND.check(model == null);
 
        // 复制模型
        var clone = new DeviceModel();
        BeanUtils.copyProperties(model, clone);
 
        clone.setVersion(0);
 
        var newName = String.format("%s##%s##", model.getName(), UUID.randomUUID().toString().substring(0, 8));
        // 克隆的模型名称为随机字符串,不需要存在名字
        clone.setName(newName);
        return add(user, clone);
    }
 
    @Override
    public DeviceModel edit(DeviceModel entity) {
        BizResultCode.MODEL_PUBLISHED_CANNOT_MODIFY.check(entity.isPublished());
        return super.edit(entity);
    }
 
    @Override
    public void publish(User user, Integer id) {
        var model = get(id);
        BizResultCode.ENTITY_NOT_FOUND.check(model == null);
 
        // 发布模型
        model.setPublished(true);
        save(model);
    }
 
    @Override
    public void setEnable(User user, Integer modelId, boolean enable) {
        var model = get(modelId);
        BizResultCode.ENTITY_NOT_FOUND.check(model == null);
        model.setEnable(enable);
        save(model);
    }
 
    @Override
    public List<DeviceModel> findActiveModels() {
        return repository.findActiveModels();
    }
 
    @Override
    public DeviceModel getDefaultModel() {
        return repository.findByName("默认设备模型");
    }
 
    @Override
    public void createDefaultModel() {
        var model = getDefaultModel();
        if(model != null){
            return;
        }
 
        // 初始化默认模型
        var modelDto = DeviceModelDto.builder()
                .name("默认设备模型")
                .description("默认设备模型")
                .enable(true)
                .published(true)
                .build();
        model = add(userService.getSystemUser(), modelDto);
        publish(userService.getSystemUser(), model.getId());
    }
 
    @Override
    public TreeRoot<HierarchyData> getHierarchyTree(Model model) {
        return hierarchyService.get(model.getHierarchy().getId()).getContent();
    }
 
    @Override
    public List<Integer> getProblemIds(Model model, DeviceType deviceType) {
        return problemService.listIds(model.getHierarchy().getId(), List.of(deviceType.hierarchyPath()));
    }
 
    @Override
    public List<Problem> listProblems(Model model, String hierarchyPath) {
        BizResultCode.HIERARCHY_NODE_NOT_FOUND.check(hierarchyPath == null);
        var allPaths = hierarchyService.findAllPaths(model.getHierarchy().getId(), List.of(hierarchyPath));
        if(allPaths.isEmpty()){
            return List.of();
        }
 
        return problemService.findProblems(model.getHierarchy().getId(), allPaths, null);
    }
 
    @Override
    public void notifyHierarchyChange(Integer hierarchyId) {
        var model = repository.findByHierarchyId(hierarchyId);
        if(model == null){
            return;
        }
 
        // 更新模型版本
        model.setVersion(model.getVersion() + 1);
        save(model);
    }
}