package org.springblade.test;
|
|
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
import org.junit.jupiter.api.Test;
|
import org.junit.jupiter.api.extension.ExtendWith;
|
import org.springblade.common.utils.SpringUtils;
|
import org.springblade.core.test.BladeBootTest;
|
import org.springblade.core.test.BladeSpringExtension;
|
import org.springblade.core.tool.utils.StringUtil;
|
import org.springblade.flow.engine.entity.FlowModel;
|
import org.springblade.flow.engine.service.FlowEngineService;
|
import org.springblade.modules.doorplateAddress.entity.DoorplateAddressEntity;
|
import org.springblade.modules.doorplateAddress.service.IDoorplateAddressService;
|
import org.springblade.modules.house.entity.HouseEntity;
|
import org.springblade.modules.house.service.IHouseService;
|
import org.springblade.modules.place.entity.PlaceEntity;
|
import org.springblade.modules.place.service.IPlaceService;
|
import org.springblade.modules.task.entity.TaskReportForRepairsEntity;
|
import org.springblade.modules.task.service.ITaskReportForRepairsService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import java.util.List;
|
|
/**
|
* Blade单元测试
|
*
|
* @author Chill
|
*/
|
@ExtendWith(BladeSpringExtension.class)
|
@BladeBootTest(appName = "blade-runner", enableLoader = true)
|
public class BladeTest {
|
|
@Autowired
|
private FlowEngineService service;
|
|
@Autowired
|
private IPlaceService iPlaceService;
|
|
@Autowired
|
private IDoorplateAddressService iDoorplateAddressService;
|
|
@Autowired
|
private ITaskReportForRepairsService iTaskReportForRepairsService;
|
|
// @Test
|
public void test() {
|
|
List<PlaceEntity> list = iPlaceService.list();
|
for (PlaceEntity placeEntity : list) {
|
DoorplateAddressEntity one = iDoorplateAddressService.getOne(Wrappers.<DoorplateAddressEntity>lambdaQuery()
|
.eq(DoorplateAddressEntity::getAddressCode, placeEntity.getHouseCode()).last("limit 1"));
|
if (one != null) {
|
placeEntity.setAoiCode(one.getAoiCode());
|
iPlaceService.updateById(placeEntity);
|
}
|
}
|
|
}
|
|
// @Test
|
public void test2() {
|
List<TaskReportForRepairsEntity> list = iTaskReportForRepairsService.list();
|
IHouseService bean = SpringUtils.getBean(IHouseService.class);
|
IPlaceService bean2 = SpringUtils.getBean(IPlaceService.class);
|
|
for (TaskReportForRepairsEntity taskReportForRepairsEntity : list) {
|
HouseEntity houseEntity = bean.getOne(Wrappers.<HouseEntity>lambdaQuery()
|
.eq(HouseEntity::getHouseCode, taskReportForRepairsEntity.getAddressCode()));
|
if (houseEntity != null) {
|
// taskReportForRepairsEntity.setGridCode(houseEntity.getGridCode());
|
iTaskReportForRepairsService.updateById(taskReportForRepairsEntity);
|
}else {
|
PlaceEntity placeEntity = bean2.getOne(Wrappers.<PlaceEntity>lambdaQuery()
|
.eq(PlaceEntity::getHouseCode, taskReportForRepairsEntity.getAddressCode()).last("limit 1"));
|
// taskReportForRepairsEntity.setGridCode(placeEntity.getGridCode());
|
iTaskReportForRepairsService.updateById(taskReportForRepairsEntity);
|
}
|
}
|
|
|
}
|
|
@Test
|
public void contextLoads() {
|
System.out.println("=====数据迁移启动=====");
|
|
// 获取 ACT_DE_MODEL 表需要转换的数据
|
List<FlowModel> list = service.list();
|
// 循环转换
|
list.forEach(flowModel -> {
|
if (StringUtil.isBlank(flowModel.getModelEditorXml())) {
|
service.update(Wrappers.<FlowModel>lambdaUpdate()
|
.set(FlowModel::getModelEditorXml, new String(service.getModelEditorXML(flowModel)))
|
.ge(FlowModel::getId, flowModel.getId())
|
);
|
}
|
});
|
|
System.out.println("=====数据迁移完毕=====");
|
}
|
|
}
|
