Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.dennisit.vplus.data.utils.WebUtils Maven / Gradle / Ivy
/*--------------------------------------------------------------------------
* Copyright (c) 2010-2020, Elon.su All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met:
*
* Redistributions of source code must retain the above copyright notice,
* this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* Neither the name of the elon developer nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
* Author: Elon.su, you can also mail [email protected]
*--------------------------------------------------------------------------
*/
package com.github.dennisit.vplus.data.utils;
import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.view.RedirectView;
import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.nio.charset.StandardCharsets;
import java.util.Map;
import java.util.Optional;
/**
* Created by Elon.su on 17/5/13.
*/
@Slf4j
public class WebUtils extends org.springframework.web.util.WebUtils {
/**
* 默认主题
*/
public static final String THEME = "theme";
public static RequestAttributes getRequestAttributes() {
return RequestContextHolder.getRequestAttributes();
}
public static HttpServletRequest getHttpServletRequest() {
return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
}
public static HttpServletResponse getHttpServletResponse() {
return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getResponse();
}
/**
* 获取session
*
* @return HttpSession
*/
public static HttpSession getSession() {
log.debug("getSession -- Thread id :{}, name : {}", Thread.currentThread().getId(), Thread.currentThread().getName());
return getHttpServletRequest().getSession();
}
/**
* 获取session的Attribute
*
* @param name session的key
* @return Object
*/
public static Object getSession(String name) {
log.debug("getSession -- Thread id :{}, name : {}", Thread.currentThread().getId(), Thread.currentThread().getName());
return getRequestAttributes().getAttribute(name, RequestAttributes.SCOPE_SESSION);
}
public static void setSession(String name, Object value) {
log.debug("setSession -- Thread id :{}, name : {}", Thread.currentThread().getId(), Thread.currentThread().getName());
getRequestAttributes().setAttribute(name, value, RequestAttributes.SCOPE_SESSION);
}
public static void removeSession(String name) {
log.debug("removeSession -- Thread id :{}, name : {}", Thread.currentThread().getId(), Thread.currentThread().getName());
getRequestAttributes().removeAttribute(name, RequestAttributes.SCOPE_SESSION);
}
/**
* 获取所有session key
*
* @return String[]
*/
public static String[] getSessionKeys() {
log.debug("getSessionKeys -- Thread id :{}, name : {}", Thread.currentThread().getId(), Thread.currentThread().getName());
return getRequestAttributes().getAttributeNames(RequestAttributes.SCOPE_SESSION);
}
public static ModelAndView toView(String theme, String viewPath) {
return toView(theme + viewPath);
}
public static ModelAndView toView(String viewPath) {
return toView(viewPath, Maps.newHashMap());
}
public static ModelAndView toView(String theme, String viewPath, Map context) {
return toView(theme + viewPath, context);
}
public static ModelAndView toView(String viewPath, Map context) {
return new ModelAndView(viewPath, context);
}
public static RedirectView redirect(String uri) {
return new RedirectView(uri);
}
public static String getRequestIP() {
return Optional.ofNullable(WebUtils.getHttpServletRequest()).map(x -> getIP(x)).orElse(StringUtils.EMPTY);
}
public static String getIP(HttpServletRequest request) {
String ip = request.getHeader("X-Requested-For");
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("X-Forwarded-For");
}
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("Proxy-Client-IP");
}
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("WL-Proxy-Client-IP");
}
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("HTTP_CLIENT_IP");
}
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getHeader("HTTP_X_FORWARDED_FOR");
}
if (StringUtils.isNotBlank(ip) || "unknown".equalsIgnoreCase(ip)) {
ip = request.getRemoteAddr();
}
return StringUtils.isBlank(ip) ? null : ip.split(",")[0];
}
public static String getCookieValue(String name) {
return getCookieValue(getHttpServletRequest(), name);
}
public static String getCookieValue(HttpServletRequest request, String name) {
Cookie cookie = getCookie(request, name);
if (null == cookie) return null;
try {
return URLDecoder.decode(cookie.getValue(), StandardCharsets.UTF_8.name());
} catch (UnsupportedEncodingException e) {
e.printStackTrace();
}
return StringUtils.EMPTY;
}
public static void clearCookie(String key) {
setCookie(getHttpServletResponse(), key, null, 0);
}
public static void clearCookie(HttpServletResponse response, String key) {
setCookie(response, key, null, 0);
}
public static void setCookie(String name, String value, int maxAgeInSeconds) {
setCookie(getHttpServletResponse(), null, name, value, maxAgeInSeconds);
}
public static void setCookie(HttpServletResponse response, String name, String value, int maxAgeInSeconds) {
setCookie(response, null, name, value, maxAgeInSeconds);
}
public static void setCookie(String domain, String name, String value, int maxAgeInSeconds) {
setCookie(getHttpServletResponse(), domain, name, value, maxAgeInSeconds);
}
public static void setCookie(HttpServletResponse response, String domain, String name, String value, int maxAgeInSeconds) {
Cookie cookie = new Cookie(name, value);
cookie.setPath("/");
cookie.setMaxAge(maxAgeInSeconds);
cookie.setHttpOnly(true);
if (StringUtils.isNotBlank(domain)) {
cookie.setDomain(domain);
}
response.addCookie(cookie);
}
public static boolean isAjax(HandlerMethod handlerMethod) {
ResponseBody responseBody = handlerMethod.getMethodAnnotation(ResponseBody.class);
if (null != responseBody) {
return true;
}
RestController restAnnotation = handlerMethod.getBean().getClass().getAnnotation(RestController.class);
if (null != restAnnotation) {
return true;
}
return false;
}
}
