lin
2024-03-01 50205a27be45a773f4ad05997eb897b9509f41e3
src/test/java/org/springblade/test/BladeTest.java
@@ -3,11 +3,20 @@
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;
@@ -24,6 +33,53 @@
   @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("=====数据迁移启动=====");