package com.estate.expert.admin.service.impl;
|
|
import com.estate.expert.admin.domain.common.EntityRef;
|
import com.estate.expert.admin.domain.common.mapper.EntityRefMapper;
|
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.system.SysModel;
|
import com.estate.expert.admin.domain.standard.model.system.SysModelDto;
|
import com.estate.expert.admin.domain.standard.model.system.SysModelRepository;
|
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.SystemModelService;
|
import com.estate.expert.admin.utils.tree.SimpleTreeRoot;
|
import com.estate.expert.admin.utils.tree.TreeNode;
|
import com.estate.expert.admin.utils.tree.TreeRoot;
|
import com.estate.expert.admin.utils.tree.helper.TreeHelper;
|
import org.springframework.stereotype.Service;
|
import org.springframework.util.CollectionUtils;
|
|
import javax.annotation.Resource;
|
import javax.lang.model.type.NullType;
|
import java.util.HashSet;
|
import java.util.List;
|
|
/**
|
* 系统模型服务实现
|
*/
|
@Service
|
public class SystemModelServiceImpl extends CrudServiceBaseImpl<SysModel, SysModelDto> implements SystemModelService {
|
|
@Resource
|
private SysModelRepository repository;
|
|
@Resource
|
private UserService userService;
|
|
@Resource
|
private HierarchyService hierarchyService;
|
|
@Resource
|
private EntityRefMapper entityRefMapper;
|
|
@Resource
|
private ProblemService problemService;
|
|
|
@Override
|
public PageResp<EntityRef> list(String keyword, PageReq pageReq) {
|
// 要显示的信息比较简单，就不单独写mapper了
|
var page = repository.findAllByKeyword(keyword, pageReq.toPageRequest());
|
var list = page.getContent().stream().map(entityRefMapper::toReference).toList();
|
return PageResp.of(page, list);
|
}
|
|
@Override
|
public SysModel add(User creator, SysModel entity) {
|
// 初始化模型
|
entity.setHierarchy(hierarchyService.get(HierarchyIds.SYSTEM));
|
entity.setVersion(0);
|
return super.add(creator, entity);
|
}
|
|
@Override
|
public void sync(Integer id) {
|
var model = get(id);
|
BizResultCode.MODEL_NOT_FOUND.check(model == null);
|
model.setVersion(model.getVersion() + 1);
|
|
// 根据 hierarchyPaths 重新获取当前对应的问题ID
|
if (model.getHierarchyPaths() != null && !model.getHierarchyPaths().isEmpty()) {
|
var problemIds = problemService.listIds(model.getHierarchy().getId(), model.getHierarchyPaths());
|
model.setProblemIds(problemIds);
|
} else {
|
// 如果 hierarchyPaths 为空，清空 problemIds
|
model.setProblemIds(null);
|
}
|
|
save(model);
|
}
|
|
@Override
|
public SysModel getDefaultModel() {
|
return get(HierarchyIds.SYSTEM);
|
}
|
|
@Override
|
public void createDefaultModel() {
|
var model = getDefaultModel();
|
if(model != null){
|
return;
|
}
|
|
// 初始化默认模型
|
var modelDto = SysModelDto.builder()
|
.name("默认综合模型")
|
.version(0)
|
.build();
|
add(userService.getSystemUser(), modelDto);
|
}
|
|
@Override
|
public TreeRoot<HierarchyData> getHierarchyTree(Integer modelId) {
|
var model = get(modelId);
|
var root = hierarchyService.root(HierarchyIds.SYSTEM);
|
if(model.getHierarchyPaths() == null || model.getHierarchyPaths().isEmpty()){
|
return root;
|
}
|
var paths = root.findAllPaths(model.getHierarchyPaths());
|
var pathSet = new HashSet<>(paths);
|
|
var newRoot = TreeHelper.pruneTree(root.root(), (node) -> pathSet.contains(node.getPath()));
|
return new SimpleTreeRoot<>(newRoot);
|
}
|
|
@Override
|
public TreeNode<NullType> getNullHierarchyTree(Integer modelId) {
|
var tree = getHierarchyTree(modelId);
|
return TreeHelper.convertTree(tree.root(), (node, newNode) -> null);
|
}
|
|
@Override
|
public List<Integer> listModelProblemIds(Model model) {
|
var ids = ((SysModel)model).getProblemIds();
|
if(ids != null){
|
return ids;
|
}
|
|
// 如果模型中没有设置，就返回所有问题
|
return problemService.findAllProblemIds(model.getHierarchy().getId());
|
}
|
|
@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));
|
BizResultCode.HIERARCHY_NODE_NOT_FOUND.check(allPaths.isEmpty());
|
|
var ids = ((SysModel)model).getProblemIds();
|
ids = CollectionUtils.isEmpty(ids) ? null : ids;
|
return problemService.findProblems(model.getHierarchy().getId(), allPaths, ids);
|
}
|
}
|
