这才是责任链模式的优雅使用方式

2021-11-15 10:34:05 浏览数 (1)

本文节选自《设计模式就该这样学》作者 谭勇德(Tom)

1 使用责任链模式设计热插拔权限控制

首先创建一个实体类Member。

代码语言:txt复制
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   '''  
                '}';
    }
}

然后来看一段我们经常写的代码。

代码语言:txt复制
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类。

代码语言:txt复制
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类的代码如下。

代码语言:txt复制
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类的代码如下。

代码语言:txt复制
public class LoginHandler extends Handler {

    public void doHandle(Member member) {
        System.out.println("登录成功!");
        member.setRoleName("管理员");
        chain.doHandle(member);
    }
}

AuthHandler类的代码如下。

代码语言:txt复制
public class AuthHandler extends Handler {

    public void doHandle(Member member) {
        if(!"管理员".equals(member.getRoleName())){
            System.out.println("您不是管理员,没有操作权限");
            return;
        }
        System.out.println("您是管理员,允许操作");
    }
}

接着修改MemberService中的代码,其实只需要将前面定义好的几个Handler根据业务需求串联起来,形成一条链即可。

代码语言:txt复制
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));
    }
}

最后编写客户端调用代码。

代码语言:txt复制
public class Test {
    public static void main(String[] args) {
        MemberService service = new MemberService();
        service.login("tom","666");
    }
}

其运行结果如下图所示。

filefile

其实我们平时使用的很多权限校验框架都是运用这个原理的,将各个维度的权限处理解耦之后再串联起来,只处理各自相关的职责。如果职责与自己不相关,则抛给链上的下一个Handler,俗称“踢皮球”。

2 责任链模式和建造者模式结合使用

因为责任链模式具备链式结构,而在上面代码中,负责组装链式结构的角色是MemberService,当链式结构较长时,MemberService的工作会非常烦琐,并且MemberService的代码相对臃肿,且后续更改处理者或消息类型时,都必须在MemberService中进行修改,不符合开闭原则。产生这些问题的原因就是因为链式结构的组装过于复杂,而对于复杂结构的创建,我们很自然地就会想到建造者模式,使用建造者模式,完全可以对MemberService指定的处理节点对象进行自动链式组装,客户只需指定处理节点对象,其他任何事情都不用关心,并且客户指定的处理节点对象的顺序不同,构造出来的链式结构也随之不同。我们来改造一下,首先修改Handler的代码。

代码语言:txt复制
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的代码。

代码语言:txt复制
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更加高聚合,代码如下。

代码语言:txt复制
public abstract class Handler<T> {
    protected Handler chain;

    private void next(Handler handler){
        this.chain = handler;
    }
...
}

通过这个案例,小伙伴们应该已经感受到责任链和建造者结合的精髓了。

关注微信公众号『 Tom弹架构 』回复“设计模式”可获取完整源码。

【推荐】Tom弹架构:30个设计模式真实案例(附源码),挑战年薪60W不是梦

本文为“Tom弹架构”原创,转载请注明出处。技术在于分享,我分享我快乐!如果本文对您有帮助,欢迎关注和点赞;如果您有任何建议也可留言评论或私信,您的支持是我坚持创作的动力。关注微信公众号『 Tom弹架构 』可获取更多技术干货!

0 人点赞