在前面的文章中搂赋，松哥已經(jīng)和小伙伴們聊了 Spring Security 中的權(quán)限表達(dá)式了，還沒看過的小伙伴們可以先看下，本文將在前文的基礎(chǔ)上繼續(xù)完善：
1. SpEL 回顧
經(jīng)過上篇文章的學(xué)習(xí)污呼，小伙伴們已經(jīng)知道了歧胁，在 Spring Security 中，@PreAuthorize桶现、@PostAuthorize 等注解都是支持 SpEL 表達(dá)式的躲雅。
在 SpEL 表達(dá)式中，如果上來就直接寫要執(zhí)行的方法名骡和，那么就說明這個方法是 RootObject 對象中的方法相赁，如果要執(zhí)行其他對象的方法相寇，那么還需要寫上對象的名字，例如如下兩個例子：
@PreAuthorize("hasAuthority('system:user:add')")
public String add() {
return "add";
}
上面這個例子中钮科，表達(dá)式中的方法是 hasAuthority唤衫，沒有寫對象名，那么就說明這個方法是 SpEL 中 RootObject 對象中的方法绵脯。
@PreAuthorize("@ss.hasPermi('monitor:operlog:list')")
@GetMapping("/list")
public TableDataInfo list(SysOperLog operLog) {
startPage();
List<SysOperLog> list = operLogService.selectOperLogList(operLog);
return getDataTable(list);
}
上面這個例子中佳励，權(quán)限注解中的表達(dá)式方法是 @ss.hasPermi('monitor:operlog:list')，其中 ss 是指 Spring 容器中的一個對象名蛆挫，hasPermi 則是這個對象中的方法赃承。
好啦，經(jīng)過前面文章的學(xué)習(xí)悴侵，這些基本知識大家都已經(jīng)掌握了楣导。
2. 如何自定義
其實(shí)上面給出來的第二個例子就是一個自定義的例子。
不過畜挨，這種自定義方式太自由了，自由到?jīng)]有在 Spring Security 架構(gòu)內(nèi)完成這件事噩凹。所以巴元，今天我想和小伙伴們聊一聊，如何在不使用第三方對象的情況下驮宴，來自定義一個權(quán)限判斷的表達(dá)式逮刨。
首先小伙伴們知道，我們在 @PreAuthorize 注解中使用的不用加對象名就能調(diào)用的權(quán)限方法堵泽，如 hasAuthority修己、hasPermission、hasRole迎罗、hasAnyRole 等睬愤，基本上都是由 SecurityExpressionRoot 及其子類提供的，準(zhǔn)確來說是由 MethodSecurityExpressionRoot 類提供的纹安。
MethodSecurityExpressionRoot 類實(shí)際上繼承自 SecurityExpressionRoot尤辱，只不過增加了過濾對象以及返回值對象。我們來看下 MethodSecurityExpressionRoot 的方法摘要：
[圖片上傳失敗...(image-c1f6ef-1657541408706)]
再來看看 SecurityExpressionRoot 中的方法：
[圖片上傳失敗...(image-4214e6-1657541408706)]
這些就是 RootObject 對象中的所有方法了厢岂，也是我們能夠在 @PreAuthorize 注解中使用的所有方法了光督。
那么現(xiàn)在想在已有方法上繼續(xù)擴(kuò)展新方法，那么我們可以通過自定義類繼承自 SecurityExpressionRoot 對象塔粒，擴(kuò)展這個 RootObject 對象结借，在該對象中繼續(xù)添加新的方法，進(jìn)而實(shí)現(xiàn)自定義權(quán)限表達(dá)式卒茬。
好啦船老，說干就干咖熟，開搞！
本文的案例在前文的基礎(chǔ)上繼續(xù)完成努隙，所以這里我就不從頭開始寫了球恤。
3. 自定義 ExpressionRoot
首先我們自定義一個類繼承自 SecurityExpressionRoot 并實(shí)現(xiàn) MethodSecurityExpressionOperations 接口（本來直接繼承自 MethodSecurityExpressionRoot 即可，但是因?yàn)檫@個類不是 public 的荸镊，沒法繼承咽斧，所以我們就實(shí)現(xiàn) MethodSecurityExpressionOperations 接口即可）：
public class CustomSecurityExpressionRoot extends SecurityExpressionRoot implements MethodSecurityExpressionOperations {
private Object filterObject;
private Object returnObject;
private AntPathMatcher antPathMatcher = new AntPathMatcher();
/**
* Creates a new instance
*
* @param authentication the {@link Authentication} to use. Cannot be null.
*/
public CustomSecurityExpressionRoot(Authentication authentication) {
super(authentication);
}
/**
* 判斷當(dāng)前對象是否具備某一個權(quán)限
* @param permission
* @return
*/
public boolean hasPermission(String permission) {
//獲取當(dāng)前登錄用戶所具有的權(quán)限
Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities();
for (GrantedAuthority authority : authorities) {
if (antPathMatcher.match(authority.getAuthority(), permission)) {
return true;
}
}
return false;
}
/**
* 是否具備多個權(quán)限中的任意一個權(quán)限
* @param permissions
* @return
*/
public boolean hasAnyPermissions(String... permissions) {
if (permissions == null || permissions.length == 0) {
return false;
}
Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities();
for (GrantedAuthority authority : authorities) {
for (String permission : permissions) {
if (antPathMatcher.match(authority.getAuthority(), permission)) {
return true;
}
}
}
return false;
}
public boolean hasAllPermissions：(String... permissions) {
Collection<? extends GrantedAuthority> authorities = authentication.getAuthorities();
if (permissions == null || permissions.length == 0) {
return false;
}
for (String permission : permissions) {
boolean flag = false;
for (GrantedAuthority authority : authorities) {
if (antPathMatcher.match(authority.getAuthority(), permission)) {
flag = true;
}
}
if (!flag) {
return false;
}
}
return true;
}
@Override
public void setFilterObject(Object filterObject) {
this.filterObject = filterObject;
}
@Override
public Object getFilterObject() {
return filterObject;
}
@Override
public void setReturnObject(Object returnObject) {
this.returnObject = returnObject;
}
@Override
public Object getReturnObject() {
return returnObject;
}
@Override
public Object getThis() {
return this;
}
}
加了 @Override 注解的方法，都是普普通通的常規(guī)方法躬存，沒啥好說的张惹。我們自己主要實(shí)現(xiàn)了三個方法，分別是：
- hasPermission：判斷當(dāng)前用戶是否具備某一個給定的權(quán)限岭洲。
- hasAnyPermissions：判斷當(dāng)前用戶是否具備給定的多個權(quán)限中的某一個宛逗。
- hasAllPermissions：判斷當(dāng)前用戶是否具備所有的給定的權(quán)限。
這里邊的邏輯我就不啰嗦了盾剩，都是基本的 Java 語法而已雷激。
另外，用 AntPathMatcher 做比對是為了支持通配符告私，這個在上篇文章中已經(jīng)說過了屎暇，這里不再贅述。
Spring Security 中驻粟，MethodSecurityExpressionRoot 的配置是通過 DefaultMethodSecurityExpressionHandler 來完成的根悼，現(xiàn)在我們自定義了 CustomSecurityExpressionRoot，那也得有一個 Handler 來配置 CustomSecurityExpressionRoot蜀撑，所以挤巡，再來一個類繼承自 DefaultMethodSecurityExpressionHandler，如下：
public class CustomMethodSecurityExpressionHandler extends DefaultMethodSecurityExpressionHandler {
@Override
protected MethodSecurityExpressionOperations createSecurityExpressionRoot(Authentication authentication, MethodInvocation invocation) {
CustomSecurityExpressionRoot root = new CustomSecurityExpressionRoot(authentication);
root.setTrustResolver(getTrustResolver());
root.setPermissionEvaluator(getPermissionEvaluator());
root.setRoleHierarchy(getRoleHierarchy());
return root;
}
}
在 createSecurityExpressionRoot 方法中創(chuàng)建一個 CustomSecurityExpressionRoot 對象酷麦，對象的 TrustResolver矿卑、權(quán)限評估器以及角色層級等，統(tǒng)統(tǒng)都用默認(rèn)的方案即可沃饶。
配置完成后粪摘，再配置一下 CustomMethodSecurityExpressionHandler 這個 Bean 即可，如下：
@Bean
CustomMethodSecurityExpressionHandler customMethodSecurityExpressionHandler() {
return new CustomMethodSecurityExpressionHandler();
}
好啦绍坝，這就注入成功了徘意。
接下來，我們就可以在權(quán)限注解中使用這個自定義的方法了：
@PreAuthorize("hasPermission('system:user:add')")
public String add() {
return "add";
}
這個自定義權(quán)限表達(dá)式的思路轩褐，說到底還是在 Spring Security 體系中玩椎咧，個人感覺這種方式更合理一些。
在 TienChin 項(xiàng)目中，松哥也將按照這種思路去改造 RuoYi-Vue 腳手架勤讽。屆時在 TienChin 項(xiàng)目視頻中蟋座，我再和大伙細(xì)聊，對視頻感興趣的小伙伴脚牍，戳這里：TienChin 項(xiàng)目配套視頻來啦向臀。
