syy/SYYTe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化域名获取方式

This commit is contained in:
阿沐 2023-06-16 11:21:15 +08:00
parent e71f06c89d
commit 4203edeed5
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
maku-framework/src/main/java/net/maku/framework/common/utils/HttpContextUtils.java
View File

@ -3,6 +3,7 @@ package net.maku.framework.common.utils;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import jakarta.servlet.http.HttpServletRequest; import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse; import jakarta.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.HttpHeaders; import org.springframework.http.HttpHeaders;
import org.springframework.web.context.request.RequestAttributes; import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder; import org.springframework.web.context.request.RequestContextHolder;
@ -57,8 +58,16 @@ public class HttpContextUtils {
public static String getDomain() { public static String getDomain() {
HttpServletRequest request = getHttpServletRequest(); HttpServletRequest request = getHttpServletRequest();
StringBuffer url = request.getRequestURL();
return url.delete(url.length() - request.getRequestURI().length(), url.length()).toString(); return getDomain(request);
}
public static String getDomain(HttpServletRequest request) {
String domain = request.getHeader(HttpHeaders.ORIGIN);
if (StrUtil.isBlank(domain)) {
domain = request.getHeader(HttpHeaders.REFERER);
}
return StringUtils.removeEnd(domain, "/");
} }
public static String getOrigin() { public static String getOrigin() {