From 4203edeed5f4e30e53409df445040c78490a82cb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=98=BF=E6=B2=90?= Date: Fri, 16 Jun 2023 11:21:15 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=9F=9F=E5=90=8D=E8=8E=B7?= =?UTF-8?q?=E5=8F=96=E6=96=B9=E5=BC=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../net/maku/framework/common/utils/HttpContextUtils.java | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/maku-framework/src/main/java/net/maku/framework/common/utils/HttpContextUtils.java b/maku-framework/src/main/java/net/maku/framework/common/utils/HttpContextUtils.java index fc35484..354f4a4 100644 --- a/maku-framework/src/main/java/net/maku/framework/common/utils/HttpContextUtils.java +++ b/maku-framework/src/main/java/net/maku/framework/common/utils/HttpContextUtils.java @@ -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() {