转载

如何在Spring中使用责任链设计模式

笔者文笔功力尚浅，如有不妥，请慷慨指出，必定感激不尽

本片文章不是讲Spring源码中使用责任链的设计实例，而是会讲在Spring中如何设计自己的责任链并且如何调用。

责任链设计模式作为我们常用的设计模式之一，用途非常的广，例如在一些流程化的执行中、或者是一些动态拦截中我们都可以使用责任链设计模式进行设计需求，从而使我们的项目无论是可用性还是可扩展性都会非常的好。

大家对于责任链还有不了解的可以看我之前的博文设计模式——责任链模式

如何定义链条

在Java开发中我们如果想要自己定义一个链条，其实非常简单，也就定义一个抽象类，然后定义几个实现方法，然后设置其next属性即可，但是在Spring中如何将框架与我们的链条结合起来呢？其实这里用到了三个注解 @Component 、 @Order 、 @PostConstruct 。

@Component
@Order
@PostConstruct

接下来我们直接看代码，首先来看抽象类，其实责任链的抽象类基本上都一样的。

public abstract class PrintChainPattern {

    private PrintChainPattern next;

    public final void print() {
        String message = getMessage();

        log.info("{} : {}",message,message);
        if (getNext()!=null){
            getNext().print();
        }
    }
    public abstract String getMessage();
}

然后我们看实现类，后面有四个实现类，依次返回是 two 、 three 、 four 。 @Order 注解中数字依次递增。这里只演示第一个实现类的代码。

@Order(1)
@Component
public class OnePrintChainPattern extends PrintChainPattern{
    @Override
    public String getMessage() {
        return "one";
    }
}

接下来就到了如何利用Spring来组装我们的链条了

@Configuration
public class InitPrintChainPattern {

    @Autowired
    private List<PrintChainPattern> printChainPatterns;

    @PostConstruct
    private void initPrintChainPattern(){
        Collections.sort(printChainPatterns, AnnotationAwareOrderComparator.INSTANCE);

        int size = printChainPatterns.size();
        for (int i = 0; i < size; i++) {
            if (i == size-1){
                printChainPatterns.get(i).setNext(null);
            }else {
                printChainPatterns.get(i).setNext(printChainPatterns.get(i+1));
            }
        }
    }

    public void print(int index){
        printChainPatterns.get(index-1).print();
    }
}

这里我们可以看到在 @PostConstruct 方法中我们做了两件事

将 List<PrintChainPattern> 中按照 @Order 注解的数字进行排序
依次设置每个节点的next值

这样我们就已经将这个链条组合了起来。接下来我们就可以随意的对这个链条进行操作，例如我下面的 print() 方法中，就是根据传进来的值的不同会从不同的节点进行执行。

如何在抽象类中使用@Autowired

在上面我们已经将我们链条组合了起来，但是如果我们的所有子类都公有一些类的话，那么这个类就要放在抽象类中。那么如果这个类我们想要从Spring的容器中取得呢？

比如我们有如下的类交给了Spring管理，我们所有子类都要使用这个类。

@Bean
public User setUser(){
    return User.builder().name("张三").age(14).build();
}

只需要在抽象类中定义一次即可，只需要在 set 方法上加 @Autowired 注解就能够将Spring容器中的类给注入进来。

private User user;

@Autowired
public void setUser(User user){
    this.user = user;
}

然后在子类中直接调用 getUser() 方法就行

@Override
public String getMessage() {
    log.info("name:{},age:{}",getUser().getName(),getUser().getAge());
    return "one";
}

如何在枚举类中使用@Autowired

为什么要在枚举类中使用 @Autowired ，是因为我在做需求时将责任链设计模式和策略模式结合起来做了，关于策略模式不明白的话可以看我之前的文章设计模式——策略模式。我们可以使用枚举类使我们的代码更加清晰可见。

我们定义一个简单的枚举策略模式的枚举类。例如我们在这里面要使用Spring容器中得了类的话，我们该如何写呢？例如还是 User 类。

public enum  HumanEnum {

    MAN("man"){
        @Override
        public void invoke() {
            log.info("i am man");
        }
    },
    WOMAN("woman"){
        @Override
        public void invoke() {
            log.info("i am woman");
        }
    };
    
    String value;

    HumanEnum(String value){
        this.value = value;
    }

    public abstract void invoke();
}

只需要在枚举类中定义一个内部类，然后将注入进来的类赋值给枚举类即可。

User user;

public void setUse(User user){
    this.user = user;
}

@Component
public static class HumanEnumInjector{

    @Autowired
    private  User user;

    @PostConstruct
    public void setValue(){
        for (HumanEnum humanEnum : EnumSet.allOf(HumanEnum.class)){
            humanEnum.setUse(user);
        }
    }
}