From 9bce48a9f30f7d80c43f43f46d40df20fcb00e15 Mon Sep 17 00:00:00 2001
From: 13693261870 <252740454@qq.com>
Date: 星期五, 03 二月 2023 09:39:59 +0800
Subject: [PATCH] 1

---
 src/main/java/com/lf/server/helper/WebHelper.java |  219 +++++++++++++++++++++++++++++++++++++++++++++++-------
 1 files changed, 190 insertions(+), 29 deletions(-)

diff --git a/src/main/java/com/lf/server/helper/WebHelper.java b/src/main/java/com/lf/server/helper/WebHelper.java
index 49b18b7..23937bb 100644
--- a/src/main/java/com/lf/server/helper/WebHelper.java
+++ b/src/main/java/com/lf/server/helper/WebHelper.java
@@ -1,12 +1,28 @@
 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.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+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;
 
 /**
@@ -18,6 +34,8 @@
 
     private final static String COMMA = ",";
 
+    private final static Log log = LogFactory.getLog(WebHelper.class);
+
     /**
      * 鑾峰彇GUID
      */
@@ -27,9 +45,6 @@
 
     /**
      * 鑾峰彇鐢ㄦ埛ip
-     *
-     * @param request
-     * @return
      */
     public static String getIpAddress(HttpServletRequest request) {
         String ip = request.getHeader("X-Forwarded-For");
@@ -82,9 +97,6 @@
 
     /**
      * 鑾峰彇褰撳墠鏃堕棿鎸囧畾鍒嗛挓鏁板悗鐨凾imestamp
-     *
-     * @param min 鍒嗛挓鏁�
-     * @return
      */
     public static Timestamp getTimestamp(int min) {
         Calendar now = Calendar.getInstance();
@@ -95,9 +107,6 @@
 
     /**
      * 浠嶤ookie涓幏鍙杢oken
-     *
-     * @param request
-     * @return
      */
     public static String getTokenFromCookie(HttpServletRequest request) {
         Cookie[] cookies = request.getCookies();
@@ -119,14 +128,10 @@
 
     /**
      * 鍚慍ookie涓坊鍔爐oken
-     *
-     * @param token
-     * @param request
-     * @param response
      */
     public static void saveToken2Cookie(String token, HttpServletRequest request, HttpServletResponse response) {
         // 鍏堝垹闄�
-        deleteCookie(StaticData.TOKEN_COOKIE_KEY, request);
+        deleteCookies(request, response);
 
         // 鍐嶄繚瀛�
         saveCookie(StaticData.TOKEN_COOKIE_KEY, token, response);
@@ -134,18 +139,14 @@
 
     /**
      * 淇濆瓨Cookie
-     *
-     * @param cookieKey
-     * @param value
-     * @param response
      */
-    public static void saveCookie(String cookieKey, String value, HttpServletResponse response) {
-        Cookie cookie = new Cookie(cookieKey, value);
+    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("")
+        //cookie.setDomain("*")
 
         response.setHeader("P3P", "CP=CAO PSA OUR");
         response.addCookie(cookie);
@@ -153,26 +154,54 @@
 
     /**
      * 鍒犻櫎cookie涓殑鍊�
-     *
-     * @param cookieKey
-     * @param request
      */
-    public static void deleteCookie(String cookieKey, HttpServletRequest request) {
+    public static void deleteCookie(String cookieKey, HttpServletRequest request, HttpServletResponse response) {
         Cookie[] cookies = request.getCookies();
         if (cookies != null && cookies.length > 0) {
             for (Cookie c : cookies) {
                 if (cookieKey.equalsIgnoreCase(c.getName())) {
                     c.setMaxAge(0);
+                    c.setPath("/");
+                    response.addCookie(c);
                 }
             }
         }
     }
 
     /**
+     * 鍒犻櫎鎵�鏈塁ookie
+     */
+    public static void deleteCookies(HttpServletRequest request, HttpServletResponse response) {
+        Cookie[] cookies = request.getCookies();
+        if (cookies != null && cookies.length > 0) {
+            for (Cookie c : cookies) {
+                c.setMaxAge(0);
+                c.setPath("/");
+                response.addCookie(c);
+            }
+        }
+    }
+
+    /**
+     * 鏍规嵁閿幏鍙朇ookie鍊�
+     */
+    public static String getCookieByKey(String key, HttpServletRequest request) {
+        Cookie[] cookies = request.getCookies();
+        if (cookies == null || cookies.length == 0) {
+            return null;
+        }
+
+        for (Cookie c : cookies) {
+            if (key.equals(c.getName())) {
+                return c.getValue();
+            }
+        }
+
+        return null;
+    }
+
+    /**
      * 鑾峰彇Token
-     *
-     * @param request
-     * @return
      */
     public static String getToken(HttpServletRequest request) {
         // 1.浠巙rl鍙傛暟涓紝鑾峰彇token
@@ -190,4 +219,136 @@
 
         return token;
     }
+
+    /**
+     * 鑾峰彇Request
+     */
+    public static HttpServletRequest getRequest() {
+        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+
+        return servletRequestAttributes.getRequest();
+    }
+
+    /**
+     * 鑾峰彇Response
+     */
+    public static HttpServletResponse getResponse() {
+        ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
+
+        return servletRequestAttributes.getResponse();
+    }
+
+    /**
+     * 鑾峰彇Session
+     */
+    public static HttpSession getSession() {
+        return getRequest().getSession();
+    }
+
+    /**
+     * 鑾峰彇鐪熷疄璺緞
+     */
+    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");
+        response.setDateHeader("Expires", 0);
+
+        PrintWriter out = response.getWriter();
+        out.print(jsonPack);
+
+        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();
+    }
+
+    /**
+     * 鎵ц鍛戒护
+     *
+     * @param cmd 鍛戒护
+     */
+    public static void exec(String cmd) {
+        try {
+            Runtime.getRuntime().exec(cmd);
+        } catch (Exception ex) {
+            log.error(ex.getMessage(), ex);
+        }
+    }
 }

--
Gitblit v1.9.3