From 1d53dd8f501a98ddcce8146443b51b357ef5f9b1 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期四, 29 十二月 2022 16:55:46 +0800 Subject: [PATCH] 1 --- src/main/java/com/lf/server/helper/WebHelper.java | 135 +++++++++++++++++++++++++++++++-------------- 1 files changed, 93 insertions(+), 42 deletions(-) diff --git a/src/main/java/com/lf/server/helper/WebHelper.java b/src/main/java/com/lf/server/helper/WebHelper.java index d2b7f9f..8fabfc5 100644 --- a/src/main/java/com/lf/server/helper/WebHelper.java +++ b/src/main/java/com/lf/server/helper/WebHelper.java @@ -1,17 +1,26 @@ package com.lf.server.helper; +import com.alibaba.fastjson.JSON; +import com.lf.server.entity.all.HttpStatus; +import com.lf.server.entity.all.ResponseMsg; +import com.lf.server.entity.all.SettingData; import com.lf.server.entity.all.StaticData; import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.ServletRequestAttributes; +import javax.servlet.ServletContext; +import javax.servlet.ServletOutputStream; import javax.servlet.http.Cookie; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import javax.servlet.http.HttpSession; +import java.io.FileInputStream; import java.io.IOException; import java.io.PrintWriter; +import java.net.URLEncoder; import java.sql.Timestamp; import java.util.Calendar; +import java.util.Random; import java.util.UUID; /** @@ -32,9 +41,6 @@ /** * 鑾峰彇鐢ㄦ埛ip - * - * @param request - * @return */ public static String getIpAddress(HttpServletRequest request) { String ip = request.getHeader("X-Forwarded-For"); @@ -87,9 +93,6 @@ /** * 鑾峰彇褰撳墠鏃堕棿鎸囧畾鍒嗛挓鏁板悗鐨凾imestamp - * - * @param min 鍒嗛挓鏁� - * @return */ public static Timestamp getTimestamp(int min) { Calendar now = Calendar.getInstance(); @@ -100,9 +103,6 @@ /** * 浠嶤ookie涓幏鍙杢oken - * - * @param request - * @return */ public static String getTokenFromCookie(HttpServletRequest request) { Cookie[] cookies = request.getCookies(); @@ -124,14 +124,10 @@ /** * 鍚慍ookie涓坊鍔爐oken - * - * @param token - * @param request - * @param response */ public static void saveToken2Cookie(String token, HttpServletRequest request, HttpServletResponse response) { // 鍏堝垹闄� - deleteAll(request, response); + deleteCookies(request, response); // 鍐嶄繚瀛� saveCookie(StaticData.TOKEN_COOKIE_KEY, token, response); @@ -139,15 +135,11 @@ /** * 淇濆瓨Cookie - * - * @param key - * @param value - * @param response */ public static void saveCookie(String key, String value, HttpServletResponse response) { Cookie cookie = new Cookie(key, value); // 璁剧疆cookie澶辨晥鏃堕棿锛屽崟浣嶄负绉� - cookie.setMaxAge(4 * 60 * 60); + cookie.setMaxAge(SettingData.COOKIE_MAX_AGE); cookie.setHttpOnly(false); cookie.setPath("/"); //cookie.setDomain("*") @@ -158,9 +150,6 @@ /** * 鍒犻櫎cookie涓殑鍊� - * - * @param cookieKey - * @param request */ public static void deleteCookie(String cookieKey, HttpServletRequest request, HttpServletResponse response) { Cookie[] cookies = request.getCookies(); @@ -177,11 +166,8 @@ /** * 鍒犻櫎鎵�鏈塁ookie - * - * @param request - * @param response */ - public static void deleteAll(HttpServletRequest request, HttpServletResponse response) { + public static void deleteCookies(HttpServletRequest request, HttpServletResponse response) { Cookie[] cookies = request.getCookies(); if (cookies != null && cookies.length > 0) { for (Cookie c : cookies) { @@ -195,7 +181,7 @@ /** * 鏍规嵁閿幏鍙朇ookie鍊� */ - public static String getCookieByKey(String key,HttpServletRequest request) { + public static String getCookieByKey(String key, HttpServletRequest request) { Cookie[] cookies = request.getCookies(); if (cookies == null || cookies.length == 0) { return null; @@ -212,9 +198,6 @@ /** * 鑾峰彇Token - * - * @param request - * @return */ public static String getToken(HttpServletRequest request) { // 1.浠巙rl鍙傛暟涓紝鑾峰彇token @@ -235,8 +218,6 @@ /** * 鑾峰彇Request - * - * @return */ public static HttpServletRequest getRequest() { ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); @@ -246,8 +227,6 @@ /** * 鑾峰彇Response - * - * @return */ public static HttpServletResponse getResponse() { ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes(); @@ -257,21 +236,25 @@ /** * 鑾峰彇Session - * - * @return */ public static HttpSession getSession() { return getRequest().getSession(); } /** - * 杈撳嚭json鏁版嵁鍒板墠绔� - * - * @param response - * @param jsonPack - * @throws Exception + * 鑾峰彇鐪熷疄璺緞 */ - public static void write2Page(HttpServletResponse response, String jsonPack) throws IOException { + public static String getRealPath(String path) { + HttpServletRequest req = getRequest(); + ServletContext ctx = req.getSession().getServletContext(); + + return ctx.getRealPath("/" + path); + } + + /** + * 杈撳嚭json鏁版嵁鍒板墠绔� + */ + public static boolean write2Page(HttpServletResponse response, String jsonPack) throws IOException { response.setContentType("application/json;charset=UTF-8"); response.setHeader("Cache-Control", "no-cache"); response.setHeader("Pragma", "No-cache"); @@ -282,5 +265,73 @@ out.flush(); out.close(); + + return false; + } + + /** + * 鑾峰彇閿欒JSON + */ + public static String getErrJson(HttpStatus status, String msg) { + return JSON.toJSONString(new ResponseMsg<String>(status, msg)); + } + + /** + * 鍐欏搷搴斾俊鎭� + */ + public static void writeInfo(HttpStatus status, String info, HttpServletResponse res) { + try { + WebHelper.write2Page(res, WebHelper.getErrJson(status, info)); + } catch (Exception e) { + // + } + } + + /** + * 鑾峰彇闅忔満鏁存暟 + */ + public static int getRandomInt(int min, int max) { + return new Random().nextInt(max) % (max - min + 1) + min; + } + + /** + * 涓嬭浇鏂囦欢 + * + * @param file 鏂囦欢 + * @param fileName 鏂囦欢鍚� + * @param res 鍝嶅簲 + * @throws Exception 寮傚父 + */ + public static void download(String file, String fileName, HttpServletResponse res) throws Exception { + if (StringHelper.isNull(fileName)) { + fileName = URLEncoder.encode(FileHelper.getFileName(file), "UTF-8"); + } + + // 璁剧疆鍝嶅簲澶翠腑鏂囦欢鐨勪笅杞芥柟寮忎负闄勪欢鏂瑰紡锛屼互鍙婅缃枃浠跺悕 + res.setHeader("Content-Disposition", "attachment; filename=" + fileName); + // 璁剧疆鍝嶅簲澶寸殑缂栫爜鏍煎紡涓篣TF-8 + res.setCharacterEncoding("UTF-8"); + + // 閫氳繃response瀵硅薄璁剧疆鍝嶅簲鏁版嵁鏍煎紡(濡傦細"text/plain; charset=utf-8") + String ext = FileHelper.getExtension(file); + String mime = FileHelper.getMime(ext); + res.setContentType(mime); + + // 閫氳繃response瀵硅薄锛岃幏鍙栧埌杈撳嚭娴� + ServletOutputStream outputStream = res.getOutputStream(); + // 瀹氫箟杈撳叆娴侊紝閫氳繃杈撳叆娴佽鍙栨枃浠跺唴瀹� + FileInputStream fileInputStream = new FileInputStream(file); + + int len = 0; + byte[] bytes = new byte[1024]; + while ((len = fileInputStream.read(bytes)) != -1) { + // 閫氳繃杈撳叆娴佽鍙栨枃浠舵暟鎹紝鐒跺悗閫氳繃涓婅堪鐨勮緭鍑烘祦鍐欏洖娴忚鍣� + outputStream.write(bytes, 0, len); + outputStream.flush(); + } + + // 鍏抽棴璧勬簮 + outputStream.close(); + fileInputStream.close(); } } -- Gitblit v1.9.3