From e20b36b1e7312137744b8ad1e27beae59500d6d2 Mon Sep 17 00:00:00 2001 From: haha Date: Sat, 6 Apr 2024 15:34:26 +0800 Subject: [PATCH] update code --- .../com/ruoyi/common/core/domain/BaseEntity.java | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java index e5e52a18..17e22af8 100644 --- a/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java +++ b/ruoyi-common/src/main/java/com/ruoyi/common/core/domain/BaseEntity.java @@ -8,6 +8,11 @@ import java.util.Map; import com.fasterxml.jackson.annotation.JsonFormat; import com.fasterxml.jackson.annotation.JsonIgnore; import com.fasterxml.jackson.annotation.JsonInclude; +import org.springframework.web.context.request.RequestContextHolder; +import org.springframework.web.context.request.ServletRequestAttributes; + +import javax.servlet.http.HttpServletRequest; + /** * Entity基类 @@ -200,7 +205,13 @@ public class BaseEntity implements Serializable } public String getProType() { - return proType; + ServletRequestAttributes servletRequestAttributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes(); + if(servletRequestAttributes==null){ + return ""; + } + RequestContextHolder.setRequestAttributes(servletRequestAttributes,true); + HttpServletRequest request = servletRequestAttributes.getRequest(); + return request.getHeader("proType"); } public void setProType(String proType) {