新需求

src/main/java/com/huiju/estateagents/common/CommConstant.java

@@ -592,4 +592,10 @@ public class CommConstant {
      */
     public static final String PRICE_TYPE_AVERAGE = "average";
 
+    /**
+     * excel后缀
+     */
+    public static final String SUFFIX_2003 = ".xls";
+    public static final String SUFFIX_2007 = ".xlsx";
+
 }

src/main/java/com/huiju/estateagents/controller/CommonController.java

@@ -28,7 +28,7 @@ public class CommonController {
     ITdMiniappTemplateTypeService iTdMiniappTemplateTypeService;
 
     /**
-     * 图片上传
+     * 图片
      * @param multipartFile
      * @return
      */

src/main/java/com/huiju/estateagents/controller/TaHousingResourcesController.java

@@ -7,10 +7,12 @@ import com.huiju.estateagents.entity.TaBuildingRoom;
 import com.huiju.estateagents.entity.TaHousingResources;
 import com.huiju.estateagents.service.ITaBuildingRoomService;
 import com.huiju.estateagents.service.ITaHousingResourcesService;
+import io.swagger.annotations.ApiOperation;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
 import java.util.List;
@@ -268,4 +270,11 @@ public class TaHousingResourcesController extends BaseController {
         }
         return responseBean;
     }
+
+    @PostMapping(value = "/admin/house/uploadExcel")
+    public ResponseBean uploadExcel(@RequestParam("file") MultipartFile file) {
+        ResponseBean responseBean = new ResponseBean();
+        responseBean = iTaHousingResourcesService.getExcelData(file);
+        return responseBean;
+    }
 }

src/main/java/com/huiju/estateagents/entity/TaHousingResources.java

@@ -112,7 +112,6 @@ public class TaHousingResources implements Serializable {
     /**
      * 实际热度
      */
-    @TableField(exist = false)
     private Integer realHeat;
 
     /**

src/main/java/com/huiju/estateagents/service/ITaHousingResourcesService.java

@@ -6,6 +6,7 @@ import com.huiju.estateagents.base.ResponseBean;
 import com.huiju.estateagents.entity.TaBuildingApartment;
 import com.huiju.estateagents.entity.TaHousingResources;
 import com.huiju.estateagents.po.TaHousingResourcesPO;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
@@ -87,4 +88,11 @@ public interface ITaHousingResourcesService extends IService<TaHousingResources>
      * @return
      */
     ResponseBean updateHouseResources(TaHousingResources taHousingResources);
+
+    /**
+     * 获取表格中的数据
+     * @param file
+     * @return
+     */
+    ResponseBean getExcelData(MultipartFile file);
 }

src/main/java/com/huiju/estateagents/service/impl/TaHousingResourcesServiceImpl.java

@@ -16,13 +16,22 @@ import com.huiju.estateagents.mapper.TaPreselectionRecordMapper;
 import com.huiju.estateagents.po.TaHousingResourcesPO;
 import com.huiju.estateagents.service.ITaHousingResourcesService;
 import org.apache.commons.collections.CollectionUtils;
+import org.apache.poi.hssf.usermodel.HSSFWorkbook;
+import org.apache.poi.ss.usermodel.Row;
+import org.apache.poi.ss.usermodel.Sheet;
+import org.apache.poi.ss.usermodel.Workbook;
+import org.apache.poi.xssf.usermodel.XSSFWorkbook;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
 import java.time.LocalDateTime;
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * <p>
@@ -324,6 +333,106 @@ public class TaHousingResourcesServiceImpl extends ServiceImpl<TaHousingResource
         return ResponseBean.success(taHousingResources);
     }
 
+    /**
+     * 获取表格中的数据
+     *
+     * @param file
+     * @return
+     */
+    @Override
+    public ResponseBean getExcelData(MultipartFile file) {
+        ResponseBean responseBean = new ResponseBean();
+        List<TaHousingResources> list = new ArrayList<>();
+        if (file == null) {
+            responseBean.addError("对象不能为空");
+            return responseBean;
+        }
+        //获取文件的名字
+        String originalFilename = file.getOriginalFilename();
+        Workbook workbook = null;
+        try {
+            if (originalFilename.endsWith(CommConstant.SUFFIX_2003)) {
+                workbook = new HSSFWorkbook(file.getInputStream());
+            } else if (originalFilename.endsWith(CommConstant.SUFFIX_2007)) {
+                workbook = new XSSFWorkbook(file.getInputStream());
+            }
+        } catch (Exception e) {
+            logger.info(originalFilename);
+            e.printStackTrace();
+            responseBean.addError("格式错误");
+            return responseBean;
+        }
+        if (workbook == null) {
+            logger.info(originalFilename);
+            responseBean.addError("格式错误");
+            return responseBean;
+        } else {
+            List<String> temHousingList = new ArrayList<String>();
+            //获取一个sheet也就是一个工作簿
+            Sheet sheet = workbook.getSheetAt(0);
+            int lastRowNum = sheet.getLastRowNum();
+            //从第一行开始第一行一般是标题
+            for (int j = 3; j <= lastRowNum; j++) {
+                Row row = sheet.getRow(j);
+                TaHousingResources taHousingResources = new TaHousingResources();
+                String termName = row.getCell(0).getStringCellValue().trim();
+                String blockName = row.getCell(1).getStringCellValue().trim();
+                String unitName = row.getCell(2).getStringCellValue().trim();
+                String floorName = row.getCell(3).getStringCellValue().trim();
+                String roomName = row.getCell(4).getStringCellValue().trim();
+                String price = row.getCell(5).getStringCellValue().trim();
+                String heat = row.getCell(6).getStringCellValue().trim();
+                Integer status = row.getCell(7).getStringCellValue().trim().equals("是") ? 1 : 0;
+
+                int currentRow = j+1;
+                if (com.baomidou.mybatisplus.core.toolkit.StringUtils.isEmpty(termName)){
+                    responseBean.addError("第" + currentRow + "行" + "期/区不能为空!");
+                    return responseBean;
+                }
+                if (com.baomidou.mybatisplus.core.toolkit.StringUtils.isEmpty(blockName)){
+                    responseBean.addError("第" + currentRow + "行" + "楼栋不能为空!");
+                    return responseBean;
+                }
+                if (com.baomidou.mybatisplus.core.toolkit.StringUtils.isEmpty(unitName)){
+                    responseBean.addError("第" + currentRow + "行" + "单元不能为空!");
+                    return responseBean;
+                }
+                if (com.baomidou.mybatisplus.core.toolkit.StringUtils.isEmpty(floorName)){
+                    responseBean.addError("第" + currentRow + "行" + "层不能为空!");
+                    return responseBean;
+                }
+                if (com.baomidou.mybatisplus.core.toolkit.StringUtils.isEmpty(roomName)){
+                    responseBean.addError("第" + currentRow + "行" + "房号不能为空!");
+                    return responseBean;
+                }
+
+                String houseInfo = termName + blockName + unitName + floorName + roomName;
+                if (temHousingList.contains(houseInfo)){
+                    logger.info("存在重复房源:{}",houseInfo);
+                    responseBean.addError("第"+ currentRow +"行存在重复房源");
+                    return responseBean;
+                }
+
+                //把数据塞入temList校验是否有重复的数据
+                temHousingList.add(houseInfo);
+                //构建数据
+                taHousingResources.setTermName(termName);
+                taHousingResources.setBlockName(blockName);
+                taHousingResources.setUnitName(unitName);
+                taHousingResources.setFloorName(floorName);
+                taHousingResources.setRoomName(roomName);
+                list.add(taHousingResources);
+            }
+        }
+        //构建分页
+        Map<String,Object> data = new HashMap<String, Object>();
+        data.put("list",list);
+        data.put("total",list.size());
+
+        responseBean.addSuccess(data);
+        return responseBean;
+    }
+
     //更新房号
     private void updateRoom(TaHousingResources taHousingResources) {
         TaBuildingRoom taBuildingRoom = new TaBuildingRoom();