274 lines
7.6 KiB
Java
274 lines
7.6 KiB
Java
/**
|
|
* Copyright (c) 2018-2028, DreamLu 卢春梦 (qq596392912@gmail.com).
|
|
* <p>
|
|
* Licensed under the GNU LESSER GENERAL PUBLIC LICENSE 3.0;
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
* <p>
|
|
* http://www.gnu.org/licenses/lgpl.html
|
|
* <p>
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
package org.springblade.core.tool.utils;
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
import org.springblade.core.tool.jackson.JsonUtil;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.lang.Nullable;
|
|
import org.springframework.util.Assert;
|
|
import org.springframework.web.bind.annotation.ResponseBody;
|
|
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 javax.servlet.http.Cookie;
|
|
import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import java.io.IOException;
|
|
import java.io.PrintWriter;
|
|
import java.util.Enumeration;
|
|
|
|
|
|
/**
|
|
* Miscellaneous utilities for web applications.
|
|
*
|
|
* @author L.cm
|
|
*/
|
|
@Slf4j
|
|
public class WebUtil extends org.springframework.web.util.WebUtils {
|
|
|
|
public static final String USER_AGENT_HEADER = "user-agent";
|
|
|
|
public static final String UN_KNOWN = "unknown";
|
|
|
|
/**
|
|
* 判断是否ajax请求
|
|
* spring ajax 返回含有 ResponseBody 或者 RestController注解
|
|
*
|
|
* @param handlerMethod HandlerMethod
|
|
* @return 是否ajax请求
|
|
*/
|
|
public static boolean isBody(HandlerMethod handlerMethod) {
|
|
ResponseBody responseBody = ClassUtil.getAnnotation(handlerMethod, ResponseBody.class);
|
|
return responseBody != null;
|
|
}
|
|
|
|
/**
|
|
* 读取cookie
|
|
*
|
|
* @param name cookie name
|
|
* @return cookie value
|
|
*/
|
|
@Nullable
|
|
public static String getCookieVal(String name) {
|
|
HttpServletRequest request = WebUtil.getRequest();
|
|
Assert.notNull(request, "request from RequestContextHolder is null");
|
|
return getCookieVal(request, name);
|
|
}
|
|
|
|
/**
|
|
* 读取cookie
|
|
*
|
|
* @param request HttpServletRequest
|
|
* @param name cookie name
|
|
* @return cookie value
|
|
*/
|
|
@Nullable
|
|
public static String getCookieVal(HttpServletRequest request, String name) {
|
|
Cookie cookie = getCookie(request, name);
|
|
return cookie != null ? cookie.getValue() : null;
|
|
}
|
|
|
|
/**
|
|
* 清除 某个指定的cookie
|
|
*
|
|
* @param response HttpServletResponse
|
|
* @param key cookie key
|
|
*/
|
|
public static void removeCookie(HttpServletResponse response, String key) {
|
|
setCookie(response, key, null, 0);
|
|
}
|
|
|
|
/**
|
|
* 设置cookie
|
|
*
|
|
* @param response HttpServletResponse
|
|
* @param name cookie name
|
|
* @param value cookie value
|
|
* @param maxAgeInSeconds maxage
|
|
*/
|
|
public static void setCookie(HttpServletResponse response, String name, @Nullable String value, int maxAgeInSeconds) {
|
|
Cookie cookie = new Cookie(name, value);
|
|
cookie.setPath("/");
|
|
cookie.setMaxAge(maxAgeInSeconds);
|
|
cookie.setHttpOnly(true);
|
|
response.addCookie(cookie);
|
|
}
|
|
|
|
/**
|
|
* 获取 HttpServletRequest
|
|
*
|
|
* @return {HttpServletRequest}
|
|
*/
|
|
public static HttpServletRequest getRequest() {
|
|
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
|
|
return (requestAttributes == null) ? null : ((ServletRequestAttributes) requestAttributes).getRequest();
|
|
}
|
|
|
|
/**
|
|
* 返回json
|
|
*
|
|
* @param response HttpServletResponse
|
|
* @param result 结果对象
|
|
*/
|
|
public static void renderJson(HttpServletResponse response, Object result) {
|
|
renderJson(response, result, MediaType.APPLICATION_JSON_VALUE);
|
|
}
|
|
|
|
/**
|
|
* 返回json
|
|
*
|
|
* @param response HttpServletResponse
|
|
* @param result 结果对象
|
|
* @param contentType contentType
|
|
*/
|
|
public static void renderJson(HttpServletResponse response, Object result, String contentType) {
|
|
response.setCharacterEncoding("UTF-8");
|
|
response.setContentType(contentType);
|
|
try (PrintWriter out = response.getWriter()) {
|
|
out.append(JsonUtil.toJson(result));
|
|
} catch (IOException e) {
|
|
log.error(e.getMessage(), e);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取ip
|
|
*
|
|
* @return {String}
|
|
*/
|
|
public static String getIP() {
|
|
return getIP(WebUtil.getRequest());
|
|
}
|
|
|
|
/**
|
|
* 获取ip
|
|
*
|
|
* @param request HttpServletRequest
|
|
* @return {String}
|
|
*/
|
|
@Nullable
|
|
public static String getIP(HttpServletRequest request) {
|
|
Assert.notNull(request, "HttpServletRequest is null");
|
|
String ip = request.getHeader("X-Requested-For");
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getHeader("X-Forwarded-For");
|
|
}
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getHeader("Proxy-Client-IP");
|
|
}
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getHeader("WL-Proxy-Client-IP");
|
|
}
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getHeader("HTTP_CLIENT_IP");
|
|
}
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getHeader("HTTP_X_FORWARDED_FOR");
|
|
}
|
|
if (StringUtil.isBlank(ip) || UN_KNOWN.equalsIgnoreCase(ip)) {
|
|
ip = request.getRemoteAddr();
|
|
}
|
|
return StringUtil.isBlank(ip) ? null : ip.split(",")[0];
|
|
}
|
|
|
|
|
|
/***
|
|
* 获取 request 中 json 字符串的内容
|
|
*
|
|
* @param request request
|
|
* @return 字符串内容
|
|
*/
|
|
public static String getRequestParamString(HttpServletRequest request) {
|
|
try {
|
|
return getRequestStr(request);
|
|
} catch (Exception ex) {
|
|
return StringPool.EMPTY;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 获取 request 请求内容
|
|
*
|
|
* @param request request
|
|
* @return String
|
|
* @throws IOException IOException
|
|
*/
|
|
public static String getRequestStr(HttpServletRequest request) throws IOException {
|
|
String queryString = request.getQueryString();
|
|
if (StringUtil.isNotBlank(queryString)) {
|
|
return new String(queryString.getBytes(Charsets.ISO_8859_1), Charsets.UTF_8).replaceAll("&", "&").replaceAll("%22", "\"");
|
|
}
|
|
return getRequestStr(request, getRequestBytes(request));
|
|
}
|
|
|
|
/**
|
|
* 获取 request 请求的 byte[] 数组
|
|
*
|
|
* @param request request
|
|
* @return byte[]
|
|
* @throws IOException IOException
|
|
*/
|
|
public static byte[] getRequestBytes(HttpServletRequest request) throws IOException {
|
|
int contentLength = request.getContentLength();
|
|
if (contentLength < 0) {
|
|
return null;
|
|
}
|
|
byte[] buffer = new byte[contentLength];
|
|
for (int i = 0; i < contentLength; ) {
|
|
|
|
int readlen = request.getInputStream().read(buffer, i, contentLength - i);
|
|
if (readlen == -1) {
|
|
break;
|
|
}
|
|
i += readlen;
|
|
}
|
|
return buffer;
|
|
}
|
|
|
|
/**
|
|
* 获取 request 请求内容
|
|
*
|
|
* @param request request
|
|
* @param buffer buffer
|
|
* @return String
|
|
* @throws IOException IOException
|
|
*/
|
|
public static String getRequestStr(HttpServletRequest request, byte[] buffer) throws IOException {
|
|
String charEncoding = request.getCharacterEncoding();
|
|
if (charEncoding == null) {
|
|
charEncoding = StringPool.UTF_8;
|
|
}
|
|
String str = new String(buffer, charEncoding).trim();
|
|
if (StringUtil.isBlank(str)) {
|
|
StringBuilder sb = new StringBuilder();
|
|
Enumeration<String> parameterNames = request.getParameterNames();
|
|
while (parameterNames.hasMoreElements()) {
|
|
String key = parameterNames.nextElement();
|
|
String value = request.getParameter(key);
|
|
StringUtil.appendBuilder(sb, key, "=", value, "&");
|
|
}
|
|
str = StringUtil.removeSuffix(sb.toString(), "&");
|
|
}
|
|
return str.replaceAll("&", "&");
|
|
}
|
|
|
|
|
|
}
|
|
|