From 4e134e61ede1f8574189af45c810ca365c7c7f9a Mon Sep 17 00:00:00 2001
From: wuxw <928255095@qq.com>
Date: 星期四, 12 五月 2022 18:33:14 +0800
Subject: [PATCH] 优化代码
---
service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java | 137 ++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 134 insertions(+), 3 deletions(-)
diff --git a/service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java b/service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java
index 36cffd5..7ebc08c 100644
--- a/service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java
+++ b/service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java
@@ -9,11 +9,13 @@
import com.java110.dto.RoomDto;
import com.java110.dto.fee.FeeConfigDto;
import com.java110.dto.fee.FeeDto;
+import com.java110.dto.owner.OwnerCarDto;
import com.java110.dto.parking.ParkingSpaceDto;
import com.java110.entity.component.ComponentValidateResult;
import com.java110.intf.community.IParkingSpaceInnerServiceSMO;
import com.java110.intf.community.IRoomV1InnerServiceSMO;
import com.java110.intf.fee.IPayFeeConfigV1InnerServiceSMO;
+import com.java110.intf.user.IOwnerCarInnerServiceSMO;
import com.java110.utils.util.Assert;
import com.java110.utils.util.DateUtil;
import org.apache.poi.ss.usermodel.*;
@@ -31,6 +33,7 @@
import java.io.ByteArrayOutputStream;
import java.io.IOException;
+import java.util.ArrayList;
import java.util.List;
/**
@@ -49,6 +52,8 @@
public static final String TYPE_PARKSPACE = "2002";
public static final String TYPE_CONTRACT = "3003"; //鍚堝悓
+ public static final int DEFAULT_ROW = 500;
+
@Autowired
private RestTemplate restTemplate;
@@ -60,6 +65,9 @@
@Autowired
private IParkingSpaceInnerServiceSMO parkingSpaceInnerServiceSMOImpl;
+
+ @Autowired
+ private IOwnerCarInnerServiceSMO ownerCarInnerServiceSMOImpl;
@Override
public ResponseEntity<Object> exportExcelData(IPageData pd) throws Exception {
@@ -159,6 +167,7 @@
return new ResponseEntity<Object>(context, headers, HttpStatus.OK);
}
+
private void getParkspaceAndConfigs(JSONObject paramIn, Workbook workbook) {
Sheet sheet = workbook.createSheet("鍒涘缓璐圭敤");
Row row = sheet.createRow(0);
@@ -180,10 +189,16 @@
ParkingSpaceDto parkingSpaceDto = new ParkingSpaceDto();
parkingSpaceDto.setCommunityId(paramIn.getString("communityId"));
- parkingSpaceDto.setPsIds(paramIn.getString("psIds").split(","));
+ parkingSpaceDto.setPaIds(paramIn.getString("paIds").split(","));
List<ParkingSpaceDto> parkingSpaceDtos = parkingSpaceInnerServiceSMOImpl.queryParkingSpaces(parkingSpaceDto);
if (parkingSpaceDtos == null || parkingSpaceDtos.size() < 1) {
+ return;
+ }
+
+ //鏌ヨ杞︾墝鍙�
+ List<OwnerCarDto> ownerCarDtos = getOwnerCars(parkingSpaceDtos);
+ if (ownerCarDtos == null || ownerCarDtos.size() < 1) {
return;
}
FeeConfigDto feeConfigDto = new FeeConfigDto();
@@ -196,10 +211,10 @@
}
int roomIndex = 2;
- for (ParkingSpaceDto tmpParkingSpaceDto : parkingSpaceDtos) {
+ for (OwnerCarDto tmpOwnerCarDto : ownerCarDtos) {
for (FeeConfigDto tmpFeeConfigDto : feeConfigDtos) {
row = sheet.createRow(roomIndex);
- row.createCell(0).setCellValue(tmpParkingSpaceDto.getAreaNum() + "-" + tmpParkingSpaceDto.getNum());
+ row.createCell(0).setCellValue(tmpOwnerCarDto.getCarNum());
row.createCell(1).setCellValue("2002");
row.createCell(2).setCellValue(tmpFeeConfigDto.getConfigId());
row.createCell(3).setCellValue(tmpFeeConfigDto.getFeeName());
@@ -208,6 +223,34 @@
roomIndex += 1;
}
}
+
+ CellRangeAddress region = new CellRangeAddress(0, 0, 0, 5);
+ sheet.addMergedRegion(region);
+ }
+
+ private List<OwnerCarDto> getOwnerCars(List<ParkingSpaceDto> parkingSpaceDtos) {
+ List<String> psIds = new ArrayList<>();
+ List<OwnerCarDto> tmpOwnerCarDtos = new ArrayList<>();
+ for (int roomIndex = 0; roomIndex < parkingSpaceDtos.size(); roomIndex++) {
+ psIds.add(parkingSpaceDtos.get(roomIndex).getPsId());
+ if (roomIndex % DEFAULT_ROW == 0 && roomIndex != 0) {
+ // 澶勭悊鎴垮眿璐圭敤
+ OwnerCarDto ownerCarDto = new OwnerCarDto();
+ ownerCarDto.setPsIds(psIds.toArray(new String[psIds.size()]));
+ ownerCarDto.setCommunityId(parkingSpaceDtos.get(roomIndex).getCommunityId());
+ tmpOwnerCarDtos.addAll(ownerCarInnerServiceSMOImpl.queryOwnerCars(ownerCarDto));
+
+ psIds = new ArrayList<>();
+ }
+ }
+ if (psIds != null && psIds.size() > 0) {
+ OwnerCarDto ownerCarDto = new OwnerCarDto();
+ ownerCarDto.setPsIds(psIds.toArray(new String[psIds.size()]));
+ ownerCarDto.setCommunityId(parkingSpaceDtos.get(0).getCommunityId());
+ tmpOwnerCarDtos.addAll(ownerCarInnerServiceSMOImpl.queryOwnerCars(ownerCarDto));
+ }
+ return tmpOwnerCarDtos;
+
}
private void getRoomAndConfigs(JSONObject paramIn, Workbook workbook) {
@@ -259,7 +302,10 @@
roomIndex += 1;
}
}
+ CellRangeAddress region = new CellRangeAddress(0, 0, 0, 5);
+ sheet.addMergedRegion(region);
}
+
/**
* 鏌ヨ杞﹁締
@@ -433,6 +479,91 @@
sheet.addMergedRegion(region);
}
+
+ /**
+ * 瀵煎嚭鎶ヨ〃
+ *
+ * @param pd 鍓嶅彴鏁版嵁灏佽
+ * @return
+ * @throws Exception
+ */
+ @Override
+ public ResponseEntity<Object> exportCustomReportTableData(IPageData pd) throws Exception {
+ ComponentValidateResult result = this.validateStoreStaffCommunityRelationship(pd, restTemplate);
+
+ JSONObject paramIn = JSONObject.parseObject(pd.getReqData());
+
+ Assert.hasKeyAndValue(paramIn, "communityId", "璇锋眰涓湭鍖呭惈灏忓尯");
+ //Assert.hasKeyAndValue(paramIn, "floorIds", "璇锋眰涓湭鍖呭惈妤兼爧");
+
+ Workbook workbook = null; //宸ヤ綔绨�
+ //宸ヤ綔琛�
+ workbook = new XSSFWorkbook();
+
+ //鏌ヨ璧勪骇鍜岃垂鐢ㄩ」
+ getCustomReportTableData(paramIn, workbook, pd);
+
+ ByteArrayOutputStream os = new ByteArrayOutputStream();
+ MultiValueMap headers = new HttpHeaders();
+ headers.add("content-type", "application/octet-stream;charset=UTF-8");
+ headers.add("Content-Disposition", "attachment;filename=customReportTableImport_" + DateUtil.getyyyyMMddhhmmssDateString() + ".xlsx");
+ headers.add("Pargam", "no-cache");
+ headers.add("Cache-Control", "no-cache");
+ //headers.add("Content-Disposition", "attachment; filename=" + outParam.getString("fileName"));
+ headers.add("Accept-Ranges", "bytes");
+ byte[] context = null;
+ try {
+ workbook.write(os);
+ context = os.toByteArray();
+ os.close();
+ workbook.close();
+ } catch (IOException e) {
+ e.printStackTrace();
+ // 淇濆瓨鏁版嵁
+ return new ResponseEntity<Object>("瀵煎嚭澶辫触", HttpStatus.INTERNAL_SERVER_ERROR);
+ }
+ // 淇濆瓨鏁版嵁
+ return new ResponseEntity<Object>(context, headers, HttpStatus.OK);
+ }
+
+
+ private void getCustomReportTableData(JSONObject paramIn, Workbook workbook, IPageData pd) {
+ Sheet sheet = workbook.createSheet("鎶ヨ〃鏁版嵁");
+ String apiUrl = "reportCustomComponent.listReportCustomComponentData" + super.mapToUrlParam(paramIn);
+ ResponseEntity<String> responseEntity = this.callCenterService(restTemplate, pd, "", apiUrl, HttpMethod.GET);
+ if (responseEntity.getStatusCode() != HttpStatus.OK) {
+ return;
+ }
+ JSONObject paramOut = JSONObject.parseObject(responseEntity.getBody());
+ if (paramOut.getIntValue("code") != 0) {
+ return;
+ }
+ JSONArray th = paramOut.getJSONObject("data").getJSONArray("th");
+
+ if (th == null || th.size() < 1) {
+ return;
+ }
+
+ Row row = sheet.createRow(0);
+ for (int thIndex = 0; thIndex < th.size(); thIndex++) {
+ row.createCell(thIndex).setCellValue(th.getString(thIndex));
+ }
+
+ JSONArray td = paramOut.getJSONObject("data").getJSONArray("td");
+
+ if (td == null || td.size() < 1) {
+ return;
+ }
+ JSONObject tdObj = null;
+ for (int tdIndex = 0; tdIndex < td.size(); tdIndex++) {
+ row = sheet.createRow(tdIndex + 1);
+ tdObj = td.getJSONObject(tdIndex);
+ for (int thIndex = 0; thIndex < th.size(); thIndex++) {
+ row.createCell(thIndex).setCellValue(tdObj.getString(th.getString(thIndex)));
+ }
+ }
+ }
+
public RestTemplate getRestTemplate() {
return restTemplate;
}
--
Gitblit v1.8.0