加入自定义报表导出

service-api/src/main/java/com/java110/api/components/assetImport/ImportAndExportFeeComponent.java

@@ -53,4 +53,17 @@ public class ImportAndExportFeeComponent {
     }
 
 
+    /**
+     * 添加应用数据
+     *
+     * @param pd 页面数据封装
+     * @return ResponseEntity 对象
+     */
+    public ResponseEntity<Object> exportCustomReportTableData(IPageData pd) throws Exception {
+        IPageData newPd = PageData.newInstance().builder(pd.getUserId(), pd.getUserName(), pd.getToken(), pd.getReqData(), pd.getComponentCode(), pd.getComponentMethod(), "",
+                pd.getSessionId(), AppDto.WEB_APP_ID, pd.getHeaders());
+        return exportRoomSMOImpl.exportCustomReportTableData(newPd);
+    }
+
+
 }

service-api/src/main/java/com/java110/api/smo/assetExport/IExportRoomSMO.java

@@ -28,4 +28,5 @@ public interface IExportRoomSMO {
      * @return ResponseEntity
      */
      ResponseEntity<Object> exportRoomExcelData(IPageData pd) throws Exception;
+     ResponseEntity<Object> exportCustomReportTableData(IPageData pd) throws Exception;
 }

service-api/src/main/java/com/java110/api/smo/assetExport/impl/ExportRoomSMOImpl.java

@@ -167,6 +167,7 @@ public class ExportRoomSMOImpl extends DefaultAbstractComponentSMO implements IE
         return new ResponseEntity<Object>(context, headers, HttpStatus.OK);
     }
 
+
     private void getParkspaceAndConfigs(JSONObject paramIn, Workbook workbook) {
         Sheet sheet = workbook.createSheet("创建费用");
         Row row = sheet.createRow(0);
@@ -305,6 +306,7 @@ public class ExportRoomSMOImpl extends DefaultAbstractComponentSMO implements IE
         sheet.addMergedRegion(region);
     }
 
+
     /**
      * 查询车辆
      *
@@ -477,6 +479,91 @@ public class ExportRoomSMOImpl extends DefaultAbstractComponentSMO implements IE
         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;
     }