From c63b59526a3684bd0b5868c430348c62b71cf504 Mon Sep 17 00:00:00 2001
From: java110 <928255095@qq.com>
Date: 星期一, 11 四月 2022 15:44:20 +0800
Subject: [PATCH] 加入自定义报表导出

---
 service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java |   87 +++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 87 insertions(+), 0 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 3b1e41f..1269735 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
@@ -167,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);
@@ -304,6 +305,7 @@
         CellRangeAddress region = new CellRangeAddress(0, 0, 0, 5);
         sheet.addMergedRegion(region);
     }
+
 
     /**
      * 鏌ヨ杞﹁締
@@ -477,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 < th.size(); tdIndex++) {
+            row = sheet.createRow(tdIndex + 1);
+            tdObj = th.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