楼栋 单元 批量导入开发完成

Api/src/main/java/com/java110/api/ApiApplicationStart.java

@@ -87,6 +87,7 @@ public class ApiApplicationStart {
 
     /**
      * swagger 插件
+     *
      * @return Docket 对象
      */
     @Bean

WebService/src/main/java/com/java110/web/smo/assetImport/impl/AssetImportSMOImpl.java

@@ -1,19 +1,31 @@
 package com.java110.web.smo.assetImport.impl;
 
+import com.alibaba.fastjson.JSONObject;
+import com.java110.common.constant.ServiceConstant;
+import com.java110.common.util.Assert;
 import com.java110.common.util.ImportExcelUtils;
 import com.java110.core.context.IPageData;
+import com.java110.entity.assetImport.ImportFloor;
+import com.java110.entity.assetImport.ImportOwner;
+import com.java110.entity.assetImport.ImportParkingSpace;
+import com.java110.entity.assetImport.ImportRoom;
+import com.java110.entity.component.ComponentValidateResult;
 import com.java110.web.core.BaseComponentSMO;
 import com.java110.web.smo.assetImport.IAssetImportSMO;
 import org.apache.poi.ss.usermodel.Sheet;
 import org.apache.poi.ss.usermodel.Workbook;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpMethod;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Service;
+import org.springframework.web.client.RestTemplate;
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.InputStream;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
@@ -28,20 +40,291 @@ import java.util.List;
 public class AssetImportSMOImpl extends BaseComponentSMO implements IAssetImportSMO {
     private final static Logger logger = LoggerFactory.getLogger(AssetImportSMOImpl.class);
 
+    @Autowired
+    private RestTemplate restTemplate;
+
     @Override
-    public ResponseEntity<String> importExcelData(IPageData pd, MultipartFile uploadFile) throws Exception{
+    public ResponseEntity<String> importExcelData(IPageData pd, MultipartFile uploadFile) throws Exception {
+
+        ComponentValidateResult result = this.validateStoreStaffCommunityRelationship(pd, restTemplate);
+
         InputStream is = uploadFile.getInputStream();
 
         Workbook workbook = null;  //工作簿
-        Sheet sheet = null;         //工作表
+        //工作表
         String[] headers = null;   //表头信息
-
+        List<ImportFloor> floors = new ArrayList<ImportFloor>();
+        List<ImportOwner> owners = new ArrayList<ImportOwner>();
+        List<ImportRoom> rooms = new ArrayList<ImportRoom>();
+        List<ImportParkingSpace> parkingSpaces = new ArrayList<ImportParkingSpace>();
         workbook = ImportExcelUtils.createWorkbook(uploadFile);
+        //获取楼信息
+        getFloors(workbook, floors);
+        //获取业主信息
+        getOwners(workbook, owners);
+
+        //获取房屋信息
+        getRooms(workbook, rooms, floors, owners);
+
+        //获取车位信息
+        getParkingSpaces(workbook, parkingSpaces, owners);
+
+        //数据校验
+        importExcelDataValidate(floors, owners, rooms, parkingSpaces);
+
+        // 保存数据
+        return dealExcelData(pd, floors, owners, rooms, parkingSpaces, result);
+    }
+
+    /**
+     * 处理ExcelData数据
+     *
+     * @param floors        楼栋单元信息
+     * @param owners        业主信息
+     * @param rooms         房屋信息
+     * @param parkingSpaces 车位信息
+     */
+    private ResponseEntity<String> dealExcelData(IPageData pd, List<ImportFloor> floors, List<ImportOwner> owners, List<ImportRoom> rooms, List<ImportParkingSpace> parkingSpaces, ComponentValidateResult result) {
+        ResponseEntity<String> responseEntity = null;
+        //保存单元信息 和 楼栋信息
+        responseEntity = savedFloorAndUnitInfo(pd, floors, result, responseEntity);
+        return responseEntity;
+    }
+
+    /**
+     * 保存 楼栋和 单元信息
+     *
+     * @param pd
+     * @param floors
+     * @param result
+     * @param responseEntity
+     * @return
+     */
+    private ResponseEntity<String> savedFloorAndUnitInfo(IPageData pd, List<ImportFloor> floors, ComponentValidateResult result, ResponseEntity<String> responseEntity) {
+        String apiUrl = "";
+        JSONObject paramIn = null;
+        for (ImportFloor importFloor : floors) {
+            //先保存 楼栋信息
+            JSONObject savedFloorInfo = getExistsFloor(pd,result,importFloor);
+            // 如果不存在，才插入
+            if(savedFloorInfo == null) {
+                apiUrl = ServiceConstant.SERVICE_API_URL + "/api/floor.saveFloor";
+                Assert.jsonObjectHaveKey(paramIn, "name", "请求报文中未包含name");
+                Assert.jsonObjectHaveKey(paramIn, "userId", "请求报文中未包含userId");
+                Assert.jsonObjectHaveKey(paramIn, "floorNum", "请求报文中未包含floorNum");
+                Assert.jsonObjectHaveKey(paramIn, "communityId", "请求报文中未包含communityId");
+                paramIn.put("communityId", result.getCommunityId());
+                paramIn.put("floorNum", importFloor.getFloorNum());
+                paramIn.put("userId", result.getUserId());
+                paramIn.put("name", importFloor.getFloorNum() + "号楼");
+                responseEntity = this.callCenterService(restTemplate, pd, paramIn.toJSONString(), apiUrl, HttpMethod.POST);
+            }
+            if (responseEntity.getStatusCode() != HttpStatus.OK) { //跳过 保存单元信息
+                continue;
+            }
+
+            savedFloorInfo = getExistsFloor(pd,result,importFloor);
+
+            if(savedFloorInfo == null){
+                continue;
+            }
+
+            paramIn.clear();
+            apiUrl = ServiceConstant.SERVICE_API_URL + "/api/unit.saveUnit";
+
+            paramIn.put("communityId", result.getCommunityId());
+            paramIn.put("floorId", savedFloorInfo.getString("floorId"));
+            paramIn.put("unitNum", importFloor.getUnitNum());
+            paramIn.put("layerCount", importFloor.getLayerCount());
+            paramIn.put("lift", importFloor.getLift());
+            responseEntity = this.callCenterService(restTemplate, pd, paramIn.toJSONString(), apiUrl, HttpMethod.POST);
+        }
+        return responseEntity;
+    }
+
+    private JSONObject getExistsFloor(IPageData pd, ComponentValidateResult result, ImportFloor importFloor) {
+        String apiUrl = "";
+        ResponseEntity<String> responseEntity = null;
+        apiUrl = ServiceConstant.SERVICE_API_URL + "/api/floor.queryFloors?page=1&row=1&communityId=" + result.getCommunityId() + "&floorNum=" + importFloor.getFloorNum();
+        responseEntity = this.callCenterService(restTemplate, pd, "", apiUrl, HttpMethod.GET);
+
+        if (responseEntity.getStatusCode() != HttpStatus.OK) { //跳过 保存单元信息
+            return null;
+        }
+
+        JSONObject savedFloorInfoResult = JSONObject.parseObject(responseEntity.getBody());
+
+        if (!savedFloorInfoResult.containsKey("apiFloorDataVoList") || savedFloorInfoResult.getJSONArray("apiFloorDataVoList").size() != 1) {
+            return null;
+        }
+
+        JSONObject savedFloorInfo = savedFloorInfoResult.getJSONArray("apiFloorDataVoList").getJSONObject(0);
+
+        return savedFloorInfo;
+    }
+
+    /**
+     * 数据校验处理
+     *
+     * @param floors
+     * @param owners
+     * @param rooms
+     * @param parkingSpaces
+     */
+    private void importExcelDataValidate(List<ImportFloor> floors, List<ImportOwner> owners, List<ImportRoom> rooms, List<ImportParkingSpace> parkingSpaces) {
+    }
+
+    /**
+     * 获取车位信息
+     *
+     * @param workbook
+     * @param parkingSpaces
+     */
+    private void getParkingSpaces(Workbook workbook, List<ImportParkingSpace> parkingSpaces, List<ImportOwner> owners) {
+        Sheet sheet = null;
+        sheet = ImportExcelUtils.getSheet(workbook, "车位信息");
+        List<Object[]> oList = ImportExcelUtils.listFromSheet(sheet);
+        ImportParkingSpace importParkingSpace = null;
+        for (int osIndex = 0; osIndex < oList.size(); osIndex++) {
+            Object[] os = oList.get(osIndex);
+            if (osIndex == 0) { // 第一行是 头部信息 直接跳过
+                continue;
+            }
+            importParkingSpace = new ImportParkingSpace();
+            importParkingSpace.setPsNum(os[0].toString());
+            importParkingSpace.setTypeCd(os[1].toString());
+            importParkingSpace.setArea(Double.parseDouble(os[2].toString()));
+            ImportOwner importOwner = getImportOwner(owners, os[3].toString());
+            importParkingSpace.setImportOwner(importOwner);
+            if (importOwner != null) {
+                importParkingSpace.setCarNum(os[4].toString());
+                importParkingSpace.setCarBrand(os[5].toString());
+                importParkingSpace.setCarType(os[6].toString());
+                importParkingSpace.setCarColor(os[7].toString());
+            }
+
+            parkingSpaces.add(importParkingSpace);
+        }
+    }
+
+    /**
+     * 获取 房屋信息
+     *
+     * @param workbook
+     * @param rooms
+     */
+    private void getRooms(Workbook workbook, List<ImportRoom> rooms,
+                          List<ImportFloor> floors, List<ImportOwner> owners) {
+        Sheet sheet = null;
+        sheet = ImportExcelUtils.getSheet(workbook, "房屋信息");
+        List<Object[]> oList = ImportExcelUtils.listFromSheet(sheet);
+        ImportRoom importRoom = null;
+        for (int osIndex = 0; osIndex < oList.size(); osIndex++) {
+            Object[] os = oList.get(osIndex);
+            if (osIndex == 0) { // 第一行是 头部信息 直接跳过
+                continue;
+            }
+            importRoom = new ImportRoom();
+            importRoom.setRoomNum(os[0].toString());
+            importRoom.setFloor(getImportFloor(floors, os[1].toString(), os[2].toString()));
+            importRoom.setLayer(Integer.parseInt(os[3].toString()));
+            importRoom.setSection(os[4].toString());
+            importRoom.setBuiltUpArea(Double.parseDouble(os[5].toString()));
+            importRoom.setImportOwner(getImportOwner(owners, os[6].toString()));
+            rooms.add(importRoom);
+        }
+    }
+
+    /**
+     * 从导入的业主信息中获取业主，如果没有返回 null
+     *
+     * @param owners
+     * @param ownerNum
+     * @return
+     */
+    private ImportOwner getImportOwner(List<ImportOwner> owners, String ownerNum) {
+        for (ImportOwner importOwner : owners) {
+            if (ownerNum.equals(importOwner.getOwnerNum())) {
+                return importOwner;
+            }
+        }
+
+        return null;
 
+    }
+
+    /**
+     * get 导入楼栋信息
+     *
+     * @param floors
+     */
+    private ImportFloor getImportFloor(List<ImportFloor> floors, String floorNum, String unitNum) {
+        for (ImportFloor importFloor : floors) {
+            if (floorNum.equals(importFloor.getFloorNum())
+                    && unitNum.equals(importFloor.getUnitNum())) {
+                return importFloor;
+            }
+        }
+
+        throw new IllegalArgumentException("在楼栋单元sheet中未找到楼栋编号[" + floorNum + "],单元编号[" + unitNum + "]数据");
+    }
+
+    /**
+     * 获取业主信息
+     *
+     * @param workbook
+     * @param owners
+     */
+    private void getOwners(Workbook workbook, List<ImportOwner> owners) {
+        Sheet sheet = null;
+        sheet = ImportExcelUtils.getSheet(workbook, "业主信息");
+        List<Object[]> oList = ImportExcelUtils.listFromSheet(sheet);
+        ImportOwner importOwner = null;
+        for (int osIndex = 0; osIndex < oList.size(); osIndex++) {
+            Object[] os = oList.get(osIndex);
+            if (osIndex == 0) { // 第一行是 头部信息 直接跳过
+                continue;
+            }
+            importOwner = new ImportOwner();
+            importOwner.setOwnerNum(os[0].toString());
+            importOwner.setOwnerName(os[1].toString());
+            importOwner.setSex("男".equals(os[2].toString()) ? "0" : "1");
+            importOwner.setAge(Integer.parseInt(os[3].toString()));
+            importOwner.setTel(os[4].toString());
+            owners.add(importOwner);
+        }
+    }
+
+    /**
+     * 获取小区
+     *
+     * @param workbook
+     * @param floors
+     */
+    private void getFloors(Workbook workbook, List<ImportFloor> floors) {
+        Sheet sheet = null;
         sheet = ImportExcelUtils.getSheet(workbook, "楼栋单元");
         List<Object[]> oList = ImportExcelUtils.listFromSheet(sheet);
+        ImportFloor importFloor = null;
+        for (int osIndex = 0; osIndex < oList.size(); osIndex++) {
+            Object[] os = oList.get(osIndex);
+            if (osIndex == 0) { // 第一行是 头部信息 直接跳过
+                continue;
+            }
+            importFloor = new ImportFloor();
+            importFloor.setFloorNum(os[0].toString());
+            importFloor.setUnitNum(os[1].toString());
+            importFloor.setLayerCount(os[2].toString());
+            importFloor.setLift("有".equals(os[3].toString()) ? "Y" : "N");
+            floors.add(importFloor);
+        }
+    }
+
+    public RestTemplate getRestTemplate() {
+        return restTemplate;
+    }
 
-        logger.debug("请求参数为：%s", pd.getReqData());
-        return new ResponseEntity<String>("成功", HttpStatus.OK);
+    public void setRestTemplate(RestTemplate restTemplate) {
+        this.restTemplate = restTemplate;
     }
 }

java110-bean/src/main/java/com/java110/entity/assetImport/ImportFloor.java

@@ -0,0 +1,48 @@
+package com.java110.entity.assetImport;
+
+/**
+ * 小区楼对象
+ */
+public class ImportFloor {
+
+    private String floorNum;
+
+    private String unitNum;
+
+    private String layerCount;
+
+    private String lift;
+
+
+    public String getFloorNum() {
+        return floorNum;
+    }
+
+    public void setFloorNum(String floorNum) {
+        this.floorNum = floorNum;
+    }
+
+    public String getUnitNum() {
+        return unitNum;
+    }
+
+    public void setUnitNum(String unitNum) {
+        this.unitNum = unitNum;
+    }
+
+    public String getLayerCount() {
+        return layerCount;
+    }
+
+    public void setLayerCount(String layerCount) {
+        this.layerCount = layerCount;
+    }
+
+    public String getLift() {
+        return lift;
+    }
+
+    public void setLift(String lift) {
+        this.lift = lift;
+    }
+}

java110-bean/src/main/java/com/java110/entity/assetImport/ImportOwner.java

@@ -0,0 +1,60 @@
+package com.java110.entity.assetImport;
+
+
+/**
+ * 业主导入 实体
+ *
+ * add by wuxw 2019-09-24
+ */
+public class ImportOwner {
+
+    private String ownerNum;
+
+    private String ownerName;
+
+    private String sex;
+
+    private int age;
+
+    private String tel;
+
+    public String getOwnerNum() {
+        return ownerNum;
+    }
+
+    public void setOwnerNum(String ownerNum) {
+        this.ownerNum = ownerNum;
+    }
+
+    public String getOwnerName() {
+        return ownerName;
+    }
+
+    public void setOwnerName(String ownerName) {
+        this.ownerName = ownerName;
+    }
+
+    public String getSex() {
+        return sex;
+    }
+
+    public void setSex(String sex) {
+        this.sex = sex;
+    }
+
+    public int getAge() {
+        return age;
+    }
+
+    public void setAge(int age) {
+        this.age = age;
+    }
+
+    public String getTel() {
+        return tel;
+    }
+
+    public void setTel(String tel) {
+        this.tel = tel;
+    }
+}

java110-bean/src/main/java/com/java110/entity/assetImport/ImportParkingSpace.java

@@ -0,0 +1,90 @@
+package com.java110.entity.assetImport;
+
+/**
+ * 车位导入
+ *
+ * add by wuxw 2019-09-24
+ */
+public class ImportParkingSpace {
+
+    private String psNum;
+
+    private String typeCd;
+
+    private double area;
+
+    private ImportOwner importOwner;
+
+    private String carNum;
+
+    private String carBrand;
+
+    private String carType;
+
+    private String carColor;
+
+
+    public String getPsNum() {
+        return psNum;
+    }
+
+    public void setPsNum(String psNum) {
+        this.psNum = psNum;
+    }
+
+    public String getTypeCd() {
+        return typeCd;
+    }
+
+    public void setTypeCd(String typeCd) {
+        this.typeCd = typeCd;
+    }
+
+    public double getArea() {
+        return area;
+    }
+
+    public void setArea(double area) {
+        this.area = area;
+    }
+
+    public ImportOwner getImportOwner() {
+        return importOwner;
+    }
+
+    public void setImportOwner(ImportOwner importOwner) {
+        this.importOwner = importOwner;
+    }
+
+    public String getCarNum() {
+        return carNum;
+    }
+
+    public void setCarNum(String carNum) {
+        this.carNum = carNum;
+    }
+
+    public String getCarBrand() {
+        return carBrand;
+    }
+
+    public void setCarBrand(String carBrand) {
+        this.carBrand = carBrand;
+    }
+
+    public String getCarType() {
+        return carType;
+    }
+
+    public void setCarType(String carType) {
+        this.carType = carType;
+    }
+
+    public String getCarColor() {
+        return carColor;
+    }
+
+    public void setCarColor(String carColor) {
+        this.carColor = carColor;
+    }
+}

java110-bean/src/main/java/com/java110/entity/assetImport/ImportRoom.java

@@ -0,0 +1,67 @@
+package com.java110.entity.assetImport;
+
+/**
+ * 房屋导入对象
+ */
+public class ImportRoom {
+
+    private ImportFloor floor;
+
+    private String roomNum;
+
+    private int layer;
+
+    private String section;
+
+    private double builtUpArea;
+
+    private ImportOwner importOwner;
+
+    public ImportFloor getFloor() {
+        return floor;
+    }
+
+    public void setFloor(ImportFloor floor) {
+        this.floor = floor;
+    }
+
+    public String getRoomNum() {
+        return roomNum;
+    }
+
+    public void setRoomNum(String roomNum) {
+        this.roomNum = roomNum;
+    }
+
+    public int getLayer() {
+        return layer;
+    }
+
+    public void setLayer(int layer) {
+        this.layer = layer;
+    }
+
+    public String getSection() {
+        return section;
+    }
+
+    public void setSection(String section) {
+        this.section = section;
+    }
+
+    public double getBuiltUpArea() {
+        return builtUpArea;
+    }
+
+    public void setBuiltUpArea(double builtUpArea) {
+        this.builtUpArea = builtUpArea;
+    }
+
+    public ImportOwner getImportOwner() {
+        return importOwner;
+    }
+
+    public void setImportOwner(ImportOwner importOwner) {
+        this.importOwner = importOwner;
+    }
+}