1 使用責任鏈模式設計熱插拔權限控制
本文節選自《設計模式就該這樣學》作者 譚勇德(Tom)
首先創建一個物體類Member,
public class Member {
private String loginName;
private String loginPass;
private String roleName;
public Member(String loginName, String loginPass) {
this.loginName = loginName;
this.loginPass = loginPass;
}
public String getLoginName() {
return loginName;
}
public String getLoginPass() {
return loginPass;
}
public String getRoleName() {
return roleName;
}
public void setRoleName(String roleName) {
this.roleName = roleName;
}
@Override
public String toString() {
return "Member{" +
"loginName='" + loginName + '\'' +
", loginPass='" + loginPass + '\'' +
'}';
}
}
然后來看一段我們經常寫的代碼,
public class MemberService {
public void login(String loginName,String loginPass){
if(StringUtils.isEmpty(loginName) ||
StringUtils.isEmpty(loginPass)){
System.out.println("用戶名和密碼校驗成功,可以往下執行");
return;
}
System.out.println("用戶名和密碼不為空,可以往下執行");
Member member = checkExists(loginName,loginPass);
if(null == member){
System.out.println("用戶不存在");
return;
}
System.out.println("登錄成功!");
if(!"管理員".equals(member.getRoleName())){
System.out.println("您不是管理員,沒有操作權限");
return;
}
System.out.println("允許操作");
}
private Member checkExists(String loginName,String loginPass){
Member member = new Member(loginName,loginPass);
member.setRoleName("管理員");
return member;
}
public static void main(String[] args) {
MemberService service = new MemberService();
service.login("tom","666");
}
}
在上面代碼中,主要做了登錄前的資料驗證,其判斷邏輯是有先后順序的,首先做非空判斷,然后檢查賬號是否有效,最侄訓得用戶角色,根據用戶角色所擁有的權限匹配是否有操作權限,那么這樣的檢驗性代碼一般都是必不可少的,但是寫在具體的業務代碼中又顯得非常臃腫,因此可以用責任鏈模式,將這些檢查步驟串聯起來,而且不影響代碼美觀,可以使我們在編碼時更加專注于某一個具體的業務邏輯處理,
下面用責任鏈模式來優化代碼,首先創建一個Handler類,
public abstract class Handler {
protected Handler chain;
public void next(Handler handler){
this.chain = handler;
}
public abstract void doHandle(Member member);
}
然后分別創建非空校驗ValidateHandler類、登錄校驗LoginHandler類和權限校驗AuthHandler類,ValidateHandler類的代碼如下,
public class ValidateHandler extends Handler {
public void doHandle(Member member) {
if(StringUtils.isEmpty(member.getLoginName()) ||
StringUtils.isEmpty(member.getLoginPass())){
System.out.println("用戶名或者密碼為空");
return;
}
System.out.println("用戶名和密碼校驗成功,可以往下執行");
chain.doHandle(member);
}
}
LoginHandler類的代碼如下,
public class LoginHandler extends Handler {
public void doHandle(Member member) {
System.out.println("登錄成功!");
member.setRoleName("管理員");
chain.doHandle(member);
}
}
AuthHandler類的代碼如下,
public class AuthHandler extends Handler {
public void doHandle(Member member) {
if(!"管理員".equals(member.getRoleName())){
System.out.println("您不是管理員,沒有操作權限");
return;
}
System.out.println("您是管理員,允許操作");
}
}
接著修改MemberService中的代碼,其實只需要將前面定義好的幾個Handler根據業務需求串聯起來,形成一條鏈即可,
public class MemberService {
public void login(String loginName,String loginPass){
Handler validateHandler = new ValidateHandler();
Handler loginHandler = new LoginHandler();
Handler authHandler = new AuthHandler();
validateHandler.next(loginHandler);
loginHandler.next(authHandler);
validateHandler.doHandle(new Member(loginName,loginPass));
}
}
最后撰寫客戶端呼叫代碼,
public class Test {
public static void main(String[] args) {
MemberService service = new MemberService();
service.login("tom","666");
}
}
其運行結果如下圖所示,

其實我們平時使用的很多權限校驗框架都是運用這個原理的,將各個維度的權限處理解耦之后再串聯起來,只處理各自相關的職責,如果職責與自己不相關,則拋給鏈上的下一個Handler,俗稱“踢皮球”,
2 責任鏈模式和建造者模式結合使用
因為責任鏈模式具備鏈式結構,而在上面代碼中,負責組裝鏈式結構的角色是MemberService,當鏈式結構較長時,MemberService的作業會非常煩瑣,并且MemberService的代碼相對臃腫,且后續更改處理者或訊息型別時,都必須在MemberService中進行修改,不符合開閉原則,產生這些問題的原因就是因為鏈式結構的組裝過于復雜,而對于復雜結構的創建,我們很自然地就會想到建造者模式,使用建造者模式,完全可以對MemberService指定的處理節點物件進行自動鏈式組裝,客戶只需指定處理節點物件,其他任何事情都不用關心,并且客戶指定的處理節點物件的順序不同,構造出來的鏈式結構也隨之不同,我們來改造一下,首先修改Handler的代碼,
public abstract class Handler<T> {
protected Handler chain;
public void next(Handler handler){
this.chain = handler;
}
public abstract void doHandle(Member member);
public static class Builder<T> {
private Handler<T> head;
private Handler<T> tail;
public Builder<T> addHandler(Handler<T> handler) {
if (this.head == null) {
this.head = this.tail = handler;
return this;
}
this.tail.next(handler);
this.tail = handler;
return this;
}
public Handler<T> build() {
return this.head;
}
}
}
然后修改MemberService的代碼,
public class MemberService {
public void login(String loginName,String loginPass){
Handler.Builder builder = new Handler.Builder();
builder.addHandler(new ValidateHandler())
.addHandler(new LoginHandler())
.addHandler(new AuthHandler());
builder.build().doHandle(new Member(loginName,loginPass));
}
}
因為建造者模式要構建的是節點處理者,所以我們把Builder作為Handler的靜態內部類,并且因為客戶端不需要進行鏈式組裝,所以還可以把鏈式組裝方法next()方法設定為private,使Handler更加高聚合,代碼如下,
public abstract class Handler<T> {
protected Handler chain;
private void next(Handler handler){
this.chain = handler;
}
...
}
通過這個案例,小伙伴們應該已經感受到責任鏈和建造者結合的精髓了,
本文為“Tom彈架構”原創,轉載請注明出處,技術在于分享,我分享我快樂!
如果本文對您有幫助,歡迎關注和點贊;如果您有任何建議也可留言評論或私信,您的支持是我堅持創作的動力,關注微信公眾號『 Tom彈架構 』可獲取更多技術干貨!
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/338905.html
標籤:Java
