package com.java110.front.smo.assetImport.impl; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; import com.java110.core.component.BaseComponentSMO; import com.java110.core.context.IPageData; import com.java110.core.factory.GenerateCodeFactory; import com.java110.dto.meterWater.ImportExportMeterWaterDto; import com.java110.entity.component.ComponentValidateResult; import com.java110.front.smo.assetImport.IImportRoomFeeSMO; import com.java110.utils.constant.ServiceConstant; import com.java110.utils.util.Assert; import com.java110.utils.util.DateUtil; import com.java110.utils.util.ImportExcelUtils; import com.java110.utils.util.StringUtil; import com.java110.vo.ResultVo; 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.text.SimpleDateFormat; import java.util.ArrayList; import java.util.Date; import java.util.List; /** * @ClassName AssetImportSmoImpl * @Description TODO * @Author wuxw * @Date 2019/9/23 23:14 * @Version 1.0 * add by wuxw 2019/9/23 **/ @Service("importRoomFeeSMOImpl") public class ImportRoomFeeSMOImpl extends BaseComponentSMO implements IImportRoomFeeSMO { private final static Logger logger = LoggerFactory.getLogger(ImportRoomFeeSMOImpl.class); private static final int DEFAULT_ADD_FEE_COUNT = 500; @Autowired private RestTemplate restTemplate; @Override public ResponseEntity importExcelData(IPageData pd, MultipartFile uploadFile) throws Exception { try { ComponentValidateResult result = this.validateStoreStaffCommunityRelationship(pd, restTemplate); //InputStream is = uploadFile.getInputStream(); Workbook workbook = ImportExcelUtils.createWorkbook(uploadFile); //工作簿 List rooms = new ArrayList(); //获取楼信息 getRooms(workbook, rooms); // 保存数据 return dealExcelData(pd, rooms, result); } catch (Exception e) { logger.error("导入失败 ", e); return new ResponseEntity("非常抱歉,您填写的模板数据有误:" + e.getMessage(), HttpStatus.BAD_REQUEST); } } /** * 处理ExcelData数据 */ private ResponseEntity dealExcelData(IPageData pd, List rooms, ComponentValidateResult result) { ResponseEntity responseEntity = null; //保存单元信息 和 楼栋信息 responseEntity = savedRoomFees(pd, rooms, result); if (responseEntity == null || responseEntity.getStatusCode() != HttpStatus.OK) { return responseEntity; } return responseEntity; } private ResponseEntity savedRoomFees(IPageData pd, List rooms, ComponentValidateResult result) { JSONObject reqJson = JSONObject.parseObject(pd.getReqData()); if (rooms.size() < 1) { throw new IllegalArgumentException("没有数据需要处理"); } JSONObject data = JSONObject.parseObject(pd.getReqData()); data.put("importFeeId", GenerateCodeFactory.getGeneratorId(GenerateCodeFactory.CODE_PREFIX_feeId)); data.put("storeId", result.getStoreId()); data.put("userId", result.getUserId()); data.put("configId", reqJson.getString("configId")); data.put("feeTypeCd", reqJson.getString("feeTypeCd")); data.put("communityId", result.getCommunityId()); List tmpImportRoomFees = new ArrayList<>(); for (int roomIndex = 0; roomIndex < rooms.size(); roomIndex++) { tmpImportRoomFees.add(rooms.get(roomIndex)); if (roomIndex % DEFAULT_ADD_FEE_COUNT == 0 && roomIndex != 0) { createRoomFee(pd, tmpImportRoomFees, data); tmpImportRoomFees = new ArrayList<>(); } } if (tmpImportRoomFees != null && tmpImportRoomFees.size() > 0) { createRoomFee(pd, tmpImportRoomFees, data); } return ResultVo.success(); } /** * 创建费用 * * @param pd * @param tmpImportRoomFees */ private void createRoomFee(IPageData pd, List tmpImportRoomFees, JSONObject data) { JSONArray importRoomFees = JSONArray.parseArray(JSONObject.toJSONString(tmpImportRoomFees)); data.put("importMeteWaterFees", importRoomFees); String apiUrl = ServiceConstant.SERVICE_API_URL + "/api/meterWater/importMeterWater"; this.callCenterService(restTemplate, pd, data.toJSONString(), apiUrl, HttpMethod.POST); } /** * 获取 房屋信息 * * @param workbook * @param rooms */ private void getRooms(Workbook workbook, List rooms) { Sheet sheet = null; sheet = ImportExcelUtils.getSheet(workbook, "房屋费用信息"); List oList = ImportExcelUtils.listFromSheet(sheet); ImportExportMeterWaterDto importRoomFee = null; for (int osIndex = 0; osIndex < oList.size(); osIndex++) { Object[] os = oList.get(osIndex); if (osIndex == 0 || osIndex == 1) { // 第一行是 头部信息 直接跳过 continue; } if (StringUtil.isNullOrNone(os[0])) { continue; } Assert.hasValue(os[1], (osIndex + 1) + "单元编号不能为空"); Assert.hasValue(os[2], (osIndex + 1) + "房屋编号不能为空"); Assert.hasValue(os[3], (osIndex + 1) + "上期度数不能为空"); Assert.hasValue(os[4], (osIndex + 1) + "上期度数时间不能为空"); Assert.hasValue(os[5], (osIndex + 1) + "本期度数不能为空"); Assert.hasValue(os[6], (osIndex + 1) + "本期度数时间不能为空"); // String startTime = excelDoubleToDate(os[5].toString()); String endTime = excelDoubleToDate(os[6].toString()); Assert.isDate(startTime, DateUtil.DATE_FORMATE_STRING_A, (osIndex + 1) + "行开始时间格式错误 请填写YYYY-MM-DD hh:mm:ss 文本格式"); Assert.isDate(endTime, DateUtil.DATE_FORMATE_STRING_A, (osIndex + 1) + "行结束时间格式错误 请填写YYYY-MM-DD hh:mm:ss 文本格式"); importRoomFee = new ImportExportMeterWaterDto(); importRoomFee.setFloorNum(os[0].toString()); importRoomFee.setUnitNum(os[1].toString()); importRoomFee.setRoomNum(os[2].toString()); importRoomFee.setPreDegrees(os[3].toString()); importRoomFee.setPreReadingTime(startTime); importRoomFee.setCurDegrees(endTime); importRoomFee.setCurReadingTime(os[6].toString()); rooms.add(importRoomFee); } } //解析Excel日期格式 public static String excelDoubleToDate(String strDate) { if (strDate.length() == 5) { try { SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd"); Date tDate = DoubleToDate(Double.parseDouble(strDate)); return sdf.format(tDate); } catch (Exception e) { e.printStackTrace(); return strDate; } } return strDate; } //解析Excel日期格式 public static Date DoubleToDate(Double dVal) { Date tDate = new Date(); long localOffset = tDate.getTimezoneOffset() * 60000; //系统时区偏移 1900/1/1 到 1970/1/1 的 25569 天 tDate.setTime((long) ((dVal - 25569) * 24 * 3600 * 1000 + localOffset)); return tDate; } public RestTemplate getRestTemplate() { return restTemplate; } public void setRestTemplate(RestTemplate restTemplate) { this.restTemplate = restTemplate; } }