From ceecebe477faf5a45b19f3b258f7add8f93b4583 Mon Sep 17 00:00:00 2001 From: futian.liu <liufutianyoo@163.com> Date: 星期二, 05 十二月 2023 10:00:24 +0800 Subject: [PATCH] 代码优化 结构调整 --- consum-base/src/main/java/com/consum/base/BaseController.java | 54 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 54 insertions(+), 0 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 5dc03e8..f41965a 100644 --- a/consum-base/src/main/java/com/consum/base/BaseController.java +++ b/consum-base/src/main/java/com/consum/base/BaseController.java @@ -1,11 +1,65 @@ package com.consum.base; +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.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; public abstract class BaseController extends PlatformAdapterController { protected TokenCacheProvider getTokenCache() { return BeanContextAware.getBeanByType(TokenCacheProvider.class); } + + protected FinOrgCacheProvider getFinOrgCache() { + return BeanContextAware.getBeanByType(FinOrgCacheProvider.class); + } + + protected FinSysTenantUserCacheProvider getFinSysTenantUserCache() { + return BeanContextAware.getBeanByType(FinSysTenantUserCacheProvider.class); + } + + protected FinSysTenantCacheProvider getFinSysTenantCache() { + return BeanContextAware.getBeanByType(FinSysTenantCacheProvider.class); + } + + /** + * @Description 鑾峰彇鍚庡彴鐧诲綍鐢ㄦ埛淇℃伅 + * @Author wh + * @Date 2023/7/25 9:59 + */ + protected FinSysTenantUser getSysInfo() { + S_user_core userInfo = UserUtils.getUserInfo(); + FinSysTenantUser finSysTenantUser = new FinSysTenantUser(); + if (userInfo.getId() == null) { + return finSysTenantUser; + } + // 浠庣紦瀛樹腑鍙栧嚭鐢ㄦ埛锛屽鏋滄病鏈夊垯鏂板 + 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")); + ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(); + workbook.write(byteArrayOutputStream); + response.setHeader("Content-Length", String.valueOf(byteArrayOutputStream.size())); + out.write(byteArrayOutputStream.toByteArray()); + workbook.close(); + out.close(); + } + } -- Gitblit v1.9.1