package com.estate.expert.admin.initdata;
|
|
import com.estate.expert.admin.domain.common.request.PageReq;
|
import com.estate.expert.admin.domain.standard.hierachy.HierarchyIds;
|
import com.estate.expert.admin.domain.standard.model.system.SysModel;
|
import com.estate.expert.admin.domain.standard.problem.ProblemDto;
|
import com.estate.expert.admin.initdata.common.ImportUtil;
|
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.model.DeviceModelService;
|
import com.estate.expert.admin.service.model.MeasureModelService;
|
import com.estate.expert.admin.service.model.SystemModelService;
|
import lombok.Builder;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.io.File;
|
import java.util.ArrayList;
|
import java.util.List;
|
|
/**
|
* 导入模型任务
|
*/
|
@Builder
|
@Slf4j
|
public class ImportModelJob {
|
|
private UserService userService;
|
|
private HierarchyService hierarchyService;
|
|
private SystemModelService systemModelService;
|
|
private DeviceModelService deviceModelService;
|
|
private MeasureModelService measureModelService;
|
|
private ProblemService problemService;
|
|
public void createDefaultModels() {
|
// 初始化默认模型
|
// systemModelService.createDefaultModel();
|
// deviceModelService.createDefaultModel();
|
measureModelService.createDefaultModel();
|
}
|
|
public void importSystemModel() {
|
|
// 导入衡泰专家模型
|
var model = loadModel("衡泰专家","/衡泰专家/层", HierarchyIds.SYSTEM);
|
systemModelService.add(userService.getSystemUser(), model);
|
}
|
|
/**
|
* 导入衡泰模型
|
*/
|
private SysModel loadModel(String name, String root, Integer hierarchyId){
|
var hierarchy = hierarchyService.get(hierarchyId);
|
var rootDir = new File(ImportUtil.docsRoot(root));
|
var dirPaths = new ArrayList<String>();
|
|
// 遍历目录下面的所有目录
|
listDir(rootDir, dirPaths);
|
|
var model = new SysModel();
|
var hierarchyPaths = new ArrayList<String>();
|
for (String dirPath : dirPaths) {
|
var hierarchyPath = dirPath.substring(dirPath.indexOf(root) + root.length());
|
var node = hierarchy.getContent().getByNamePath(hierarchyPath);
|
if(node == null){
|
log.error("node is null, filePath: " + hierarchyPath);
|
continue;
|
}
|
hierarchyPaths.add(node.getPath());
|
}
|
model.setName(name);
|
model.setHierarchyPaths(hierarchyPaths);
|
|
// add problems
|
var problems = problemService.list(hierarchyId, hierarchyPaths, null, PageReq.of(0, 10000000));
|
var problemIds = problems.getList().stream().map(ProblemDto::getId).toList();
|
model.setHierarchy(hierarchy);
|
model.setProblemIds(problemIds);
|
model.setVersion(0);
|
return model;
|
}
|
|
private void listDir(File rootDir, List<String> dirPaths){
|
File[] files = rootDir.listFiles();
|
if(files == null || files.length == 0){
|
dirPaths.add(rootDir.getAbsolutePath());
|
return;
|
}
|
for (File f : files) {
|
if (f.isDirectory()) {
|
listDir(f, dirPaths);
|
}
|
}
|
}
|
}
|
