package com.estate.expert.admin.initdata;
|
|
import com.estate.expert.admin.domain.business.device.usage.DeviceUsageDto;
|
import com.estate.expert.admin.domain.common.EntityRef;
|
import com.estate.expert.admin.domain.plan.PlanDto;
|
import com.estate.expert.admin.domain.plan.PlanType;
|
import com.estate.expert.admin.domain.project.ProjectDto;
|
import com.estate.expert.admin.domain.project.detail.AddressInfo;
|
import com.estate.expert.admin.domain.project.detail.CompanyInfo;
|
import com.estate.expert.admin.domain.project.enums.ProjectType;
|
import com.estate.expert.admin.domain.project.enums.ProjectUsageType;
|
import com.estate.expert.admin.domain.project.info.ProjectDangerousType;
|
import com.estate.expert.admin.domain.project.info.ProjectInfoDto;
|
import com.estate.expert.admin.domain.project.info.ProjectSuperDangerousType;
|
import com.estate.expert.admin.initdata.common.ImportUtil;
|
import com.estate.expert.admin.service.*;
|
import com.estate.expert.admin.service.expert.DeviceExpertService;
|
import com.estate.expert.admin.service.expert.SysExpertService;
|
import com.estate.expert.admin.service.model.DeviceModelService;
|
import com.estate.expert.admin.service.model.SystemModelService;
|
import lombok.Builder;
|
import lombok.extern.slf4j.Slf4j;
|
|
import java.text.SimpleDateFormat;
|
import java.time.LocalDate;
|
import java.util.Date;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@Builder
|
@Slf4j
|
public class ImportPlanAndProject {
|
|
private UserService userService;
|
|
private PlanService planService;
|
|
private ProjectService projectService;
|
|
private ProjectInfoService projectInfoService;
|
|
private DeviceModelService deviceModelService;
|
|
private SystemModelService systemModelService;
|
|
private SysExpertService sysExpertService;
|
|
private DeviceExpertService deviceExpertService;
|
|
private DeviceUsageService deviceUsageService;
|
|
private DeviceService deviceService;
|
|
public void importProjects() {
|
log.info("Importing projects...");
|
|
var dateFormat = new SimpleDateFormat("yyyyMMddHHmm");
|
var itemCodePrefix = dateFormat.format(new Date());
|
|
ImportUtil.doImport(i -> {
|
var project = ProjectDto.builder()
|
.name("测试项目_" + i)
|
.type(ProjectType.HOUSE_CONSTRUCTION)
|
.usageType(ProjectUsageType.RESIDENTIAL)
|
.client(EntityRef.builder().id(1).build())
|
.itemCode(itemCodePrefix + String.format("%02d", i))
|
.departmentPath("/0")
|
.addressInfo(AddressInfo.builder()
|
.province("浙江省")
|
.city("杭州市")
|
.district("西湖区")
|
.detail("古荡街道" + i + "号")
|
.build())
|
.constructionCompanyInfo(CompanyInfo.builder()
|
.name("测试建设公司")
|
.leader("张三")
|
.phone("12345678901")
|
.build())
|
.supervisorCompanyInfo(CompanyInfo.builder()
|
.name("测试监理公司")
|
.leader("李四")
|
.phone("12345678902")
|
.build())
|
.developmentCompanyInfo(CompanyInfo.builder()
|
.name("测试开发公司")
|
.leader("王五")
|
.phone("12345678903")
|
.build())
|
.projectCost(1000000)
|
.constructionLicense("TEST_LICENSE_" + i)
|
.constructionLicenseFiles(List.of(EntityRef.of(1)))
|
.branchCompany("测试分公司")
|
.build();
|
var prj = projectService.add(userService.getSystemUser(), project);
|
|
var info = ProjectInfoDto.builder()
|
.project(EntityRef.builder().id(prj.getId()).build())
|
.detail("测试项目详情_" + i)
|
.progress("测试项目进展_" + i)
|
.progressThisSeason("测试本季度进展_" + i)
|
.updateDate(LocalDate.now())
|
.dangerousOptions(new ProjectDangerousType[] {ProjectDangerousType.DEMOLITION})
|
.superDangerousOptions(new ProjectSuperDangerousType[] {ProjectSuperDangerousType.DEMOLITION})
|
.build();
|
projectInfoService.add(userService.getSystemUser(), info);
|
});
|
}
|
|
public void importPlans() {
|
ImportUtil.doImport( index -> {
|
var plan = PlanDto.builder()
|
.name("测试计划_" + index)
|
.client(EntityRef.builder().id(1).build())
|
.type(PlanType.values() [index % PlanType.values().length])
|
.systemModel(EntityRef.builder().id(systemModelService.getDefaultModel().getId()).build())
|
.systemScoreExpert(EntityRef.builder().id(sysExpertService.getDefaultExpert().getId()).build())
|
.systemFuzzyExpert(EntityRef.builder().id(sysExpertService.getDefaultExpert().getId()).build())
|
.deviceModel(EntityRef.builder().id(deviceModelService.getDefaultModel().getId()).build())
|
.deviceFuzzyExpert(EntityRef.builder().id(deviceExpertService.getDefaultExpert().getId()).build())
|
.deviceScoreExpert(EntityRef.builder().id(deviceExpertService.getDefaultExpert().getId()).build())
|
.build();
|
planService.add(userService.getSystemUser(), plan);
|
});
|
}
|
|
public void importDeviceUsage() {
|
ImportUtil.doImport( index -> {
|
var usage = DeviceUsageDto.builder()
|
.device(EntityRef.builder().id(index).build())
|
.project(EntityRef.builder().id(1).build())
|
.properties(buildDeviceUsageMap(index))
|
.build();
|
deviceUsageService.add(userService.getSystemUser(), usage);
|
});
|
}
|
|
private Map<String, String> buildDeviceUsageMap(int index) {
|
var device = deviceService.get(index);
|
var deviceUsageForm = deviceService.deviceUsageForm(device.getType());
|
var properties = new HashMap<String, String>();
|
deviceUsageForm.forEach(group -> {
|
group.getItems().forEach(item -> {
|
switch (item.getType())
|
{
|
case STRING:
|
properties.put(item.getName(), item.getName() + "xxx");
|
break;
|
case INT:
|
properties.put(item.getName(), "1");
|
break;
|
case ENUM:
|
properties.put(item.getName(), item.getOptions()[0]);
|
break;
|
case DATE:
|
properties.put(item.getName(), "2024-01-01");
|
break;
|
}
|
});
|
});
|
return properties;
|
}
|
}
|
