From 9cf4b513cfc65d641d46be6ea424a9f45e4d48f7 Mon Sep 17 00:00:00 2001 From: 石广澎 <shiguangpeng@163.com> Date: 星期二, 12 十二月 2023 11:43:24 +0800 Subject: [PATCH] feat: bug修复 12564 --- consum-base/src/main/java/com/consum/base/BaseController.java | 31 ++++++++++++++++++++----------- 1 files changed, 20 insertions(+), 11 deletions(-) diff --git a/consum-base/src/main/java/com/consum/base/BaseController.java b/consum-base/src/main/java/com/consum/base/BaseController.java index 0ed10f4..be0d5ba 100644 --- a/consum-base/src/main/java/com/consum/base/BaseController.java +++ b/consum-base/src/main/java/com/consum/base/BaseController.java @@ -1,16 +1,20 @@ package com.consum.base; +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.InputStream; +import java.net.URLEncoder; + +import javax.servlet.http.HttpServletResponse; + +import org.apache.poi.ss.usermodel.Workbook; + import com.consum.model.po.FinSysTenantUser; import com.iplatform.base.PlatformAdapterController; import com.iplatform.base.util.UserUtils; import com.iplatform.core.BeanContextAware; import com.iplatform.model.po.S_user_core; -import java.io.ByteArrayOutputStream; -import java.io.IOException; -import java.io.OutputStream; -import java.net.URLEncoder; -import javax.servlet.http.HttpServletResponse; -import org.apache.poi.ss.usermodel.Workbook; +import com.walker.file.FileInfo; public abstract class BaseController extends PlatformAdapterController { @@ -45,16 +49,21 @@ return this.getFinSysTenantUserCache().get(String.valueOf(userInfo.getId())); } - protected void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws IOException { - OutputStream out = response.getOutputStream(); + protected String downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws Exception { response.setCharacterEncoding("UTF-8"); response.setHeader("content-Type", "application/vnd.ms-excel"); - response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName + ".xls", "UTF-8")); + response.setHeader("Content-Disposition", + "attachment;filename=" + URLEncoder.encode(fileName + ".xls", "UTF-8")); ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); workbook.write(byteArrayOutputStream); response.setHeader("Content-Length", String.valueOf(byteArrayOutputStream.size())); - out.write(byteArrayOutputStream.toByteArray()); + + InputStream byteArrayInputStream = new ByteArrayInputStream(byteArrayOutputStream.toByteArray()); + FileInfo fileInfo = this.uploadFileToLocal(byteArrayInputStream, fileName + ".xls", "-1", + byteArrayOutputStream.size(), 0, "-1"); + String fileUrl = fileInfo.getUrl(); workbook.close(); - out.close(); + return fileUrl; } + } -- Gitblit v1.9.1