work/book-api
4
1
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 jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.HttpHeaders;
import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
@ -57,8 +58,16 @@ public class HttpContextUtils {
public static String getDomain() {
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() {