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) {