1

sws

2022-11-26 ab849f796bdc17236a95ea5fe5c166fb8f24a75c

 src/main/java/com/lf/server/helper/WebHelper.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,337 @@
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;

/**
 * Web帮助类
 * @author WWW
 */
public class WebHelper {
    private final static String UNKNOWN = "unknown";

    private final static String COMMA = ",";

    /**
     * èŽ·å–GUID
     */
    public static String getGuid() {
        return UUID.randomUUID().toString();
    }

    /**
     * èŽ·å–用户ip
     */
    public static String getIpAddress(HttpServletRequest request) {
        String ip = request.getHeader("X-Forwarded-For");
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("WL-Proxy-Client-IP");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_X_FORWARDED_FOR");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_X_FORWARDED");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_X_CLUSTER_CLIENT_IP");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_CLIENT_IP");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_FORWARDED_FOR");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_FORWARDED");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("HTTP_VIA");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getHeader("REMOTE_ADDR");
        }
        if (ip == null || ip.length() == 0 || UNKNOWN.equalsIgnoreCase(ip)) {
            ip = request.getRemoteAddr();
        }
        if (ip.contains(COMMA)) {
            return ip.split(",")[0];
        }

        return ip;
    }

    /**
     * èŽ·å–当前时间的Timestamp
     */
    public static Timestamp getCurrentTimestamp() {
        return new Timestamp(System.currentTimeMillis());
    }

    /**
     * èŽ·å–当前时间指定分钟数后的Timestamp
     */
    public static Timestamp getTimestamp(int min) {
        Calendar now = Calendar.getInstance();
        now.add(Calendar.MINUTE, min);

        return new Timestamp(now.getTimeInMillis());
    }

    /**
     * ä»ŽCookie中获取token
     */
    public static String getTokenFromCookie(HttpServletRequest request) {
        Cookie[] cookies = request.getCookies();
        if (cookies == null || cookies.length == 0) {
            return null;
        }

        for (Cookie cookie : cookies) {
            switch (cookie.getName()) {
                case StaticData.TOKEN_COOKIE_KEY:
                    return cookie.getValue();
                default:
                    break;
            }
        }

        return null;
    }

    /**
     * å‘Cookie中添加token
     */
    public static void saveToken2Cookie(String token, HttpServletRequest request, HttpServletResponse response) {
        // å…ˆåˆ é™¤
        deleteCookies(request, response);

        // å†ä¿å­˜
        saveCookie(StaticData.TOKEN_COOKIE_KEY, token, response);
    }

    /**
     * ä¿å­˜Cookie
     */
    public static void saveCookie(String key, String value, HttpServletResponse response) {
        Cookie cookie = new Cookie(key, value);
        // è®¾ç½®cookie失效时间，单位为秒
        cookie.setMaxAge(SettingData.COOKIE_MAX_AGE);
        cookie.setHttpOnly(false);
        cookie.setPath("/");
        //cookie.setDomain("*")

        response.setHeader("P3P", "CP=CAO PSA OUR");
        response.addCookie(cookie);
    }

    /**
     * åˆ é™¤cookie中的值
     */
    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);
                }
            }
        }
    }

    /**
     * åˆ é™¤æ‰€æœ‰Cookie
     */
    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);
            }
        }
    }

    /**
     * æ ¹æ®é”®èŽ·å–Cookie值
     */
    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
     */
    public static String getToken(HttpServletRequest request) {
        // 1.从url参数中，获取token
        String token = request.getParameter(StaticData.TOKEN_KEY);

        // 2.为空，则从header里获取
        if (token == null) {
            token = request.getHeader(StaticData.TOKEN_KEY);
        }

        // 3.还为空，则从cookie里获取
        if (token == null) {
            token = getTokenFromCookie(request);
        }

        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);
        // è®¾ç½®å“åº”头的编码格式为UTF-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();
    }
}