From b4c27cf4af545af1a240f56d34702d38a783062d Mon Sep 17 00:00:00 2001 From: futian.liu <liufutianyoo@163.com> Date: 星期二, 12 十二月 2023 17:22:45 +0800 Subject: [PATCH] 添加分类重复验证 --- consum-base/src/main/java/com/consum/base/BaseController.java | 33 ++++++++++++++++++--------------- 1 files changed, 18 insertions(+), 15 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..468f48c 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,19 @@ package com.consum.base; +import java.io.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.InputStream; + +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 +48,16 @@ return this.getFinSysTenantUserCache().get(String.valueOf(userInfo.getId())); } - protected void downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws IOException { - OutputStream out = response.getOutputStream(); - response.setCharacterEncoding("UTF-8"); - response.setHeader("content-Type", "application/vnd.ms-excel"); - response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName + ".xls", "UTF-8")); + protected String downLoadExcel(String fileName, HttpServletResponse response, Workbook workbook) throws Exception { ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); workbook.write(byteArrayOutputStream); - response.setHeader("Content-Length", String.valueOf(byteArrayOutputStream.size())); - out.write(byteArrayOutputStream.toByteArray()); - workbook.close(); - out.close(); + + InputStream byteArrayInputStream = new ByteArrayInputStream(byteArrayOutputStream.toByteArray()); + FileInfo fileInfo = this.uploadFileToLocal(byteArrayInputStream, fileName + ".xls", "-1", + byteArrayOutputStream.size(), 0, "-1"); + String fileUrl = fileInfo.getUrl(); + + return "/file/" + fileUrl; } + } -- Gitblit v1.9.1