1、修改app端新建楼宇接口

censoft-rongtong/src/main/java/com/censoft/censoftrongtong/controller/RiskPlanAppController.java

@@ -748,7 +748,28 @@ public class RiskPlanAppController extends BaseController {
     public AjaxResult addBuilding(@RequestBody LedgerBuilding ledgerBuilding)
     {
         ledgerBuilding.setCreateBy(getUsername());
-        return toAjax(ledgerBuildingService.insertLedgerBuilding(ledgerBuilding));
+        ledgerBuilding.setFloor(ledgerBuilding.getGroundFloor()+ledgerBuilding.getUndergroundFloor()+"");
+        int result = ledgerBuildingService.insertLedgerBuilding(ledgerBuilding);
+        //导入楼层信息
+        if (ledgerBuilding.getGroundFloor()!=0){
+            for (int i = 1; i <= ledgerBuilding.getGroundFloor(); i++) {
+                LedgerFloor ledgerFloor = new LedgerFloor();
+                ledgerFloor.setBuildingId(ledgerBuilding.getId());
+                ledgerFloor.setName(i+"层");
+                ledgerFloor.setType("地上");
+                ledgerFloorService.insertLedgerFloor(ledgerFloor);
+            }
+        }
+        if (ledgerBuilding.getUndergroundFloor()!=0){
+            for (int i = 1; i <= ledgerBuilding.getUndergroundFloor(); i++) {
+                LedgerFloor ledgerFloor = new LedgerFloor();
+                ledgerFloor.setBuildingId(ledgerBuilding.getId());
+                ledgerFloor.setName(-i+"层");
+                ledgerFloor.setType("地下");
+                ledgerFloorService.insertLedgerFloor(ledgerFloor);
+            }
+        }
+        return toAjax(result);
     }
 
 }