|
|
|
@ -1,5 +1,9 @@ |
|
|
|
|
package com.ruoyi.framework.security.context; |
|
|
|
|
|
|
|
|
|
import org.springframework.web.context.request.RequestAttributes; |
|
|
|
|
import org.springframework.web.context.request.RequestContextHolder; |
|
|
|
|
import com.ruoyi.common.core.text.Convert; |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* 权限信息 |
|
|
|
|
* |
|
|
|
@ -7,15 +11,17 @@ package com.ruoyi.framework.security.context; |
|
|
|
|
*/ |
|
|
|
|
public class PermissionContextHolder |
|
|
|
|
{ |
|
|
|
|
private static final ThreadLocal<String> contextHolder = new ThreadLocal<>(); |
|
|
|
|
private static final String PERMISSION_CONTEXT_ATTRIBUTES = "PERMISSION_CONTEXT"; |
|
|
|
|
|
|
|
|
|
public static void setContext(String permission) |
|
|
|
|
{ |
|
|
|
|
contextHolder.set(permission); |
|
|
|
|
RequestContextHolder.currentRequestAttributes().setAttribute(PERMISSION_CONTEXT_ATTRIBUTES, permission, |
|
|
|
|
RequestAttributes.SCOPE_REQUEST); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
public static String getContext() |
|
|
|
|
{ |
|
|
|
|
return contextHolder.get(); |
|
|
|
|
return Convert.toStr(RequestContextHolder.currentRequestAttributes().getAttribute(PERMISSION_CONTEXT_ATTRIBUTES, |
|
|
|
|
RequestAttributes.SCOPE_REQUEST)); |
|
|
|
|
} |
|
|
|
|
} |
|
|
|
|