Spring 原生注解来快速实现 策略模式 + 工厂模式

前言

这阵子在做项目组重构的工作，工作中的一部分就是就目前代码库中与企业交互的逻辑抽离出来，单独做一个微服务，实现企业交互逻辑的关注点分离。

在这里面我很自然而然的就用到了策略模式 + 工厂模式的方式，包装内部实现细节，向外提供统一的调用方式，有效的减少if/else的业务代码，使得代码更容易维护，扩展。

之前看过一些文章，是使用自定义注解+自动BeanProcessor的方式来实现，个人感觉有点麻烦。因为Spring原生就提供类似的特性。

本篇旨在介绍在实际的业务场景，如何借助Spring IoC 依赖注入的特性，使用Spring 原生注解 来快速实现 策略模式 + 工厂模式。希望能够对你有启发。

业务场景

从原项目抽离出来的企业服务，承担的是与外部企业交互的职责。不同企业，虽然会产生的交互行为是相同的，但是交互行为内部的实现逻辑各有不同，比如发送报文接口，不同企业可能报文格式会不同。

针对这种不同企业交互细节不同的场景，将与企业的交互行为抽象出来EntStrategy接口，根据服务消费者传入的企业号选择对应的实现类（策略类），逻辑简化之后如下图。

快速实现

现在让我们用快速用一个DEMO实现上述场景。

我们的期望目标是，根据不同的企业编号，我们能够快速找到对应的策略实现类去执行发送报文的操作。

Step 1 实现策略类

假设我们现在对外提供的服务Api是这样的，

/**
 * @param entNum 企业编号
 */
public void send(String entNum) {
    // 根据不同的企业编号，我们能够快速找到对应的策略实现类去执行发送报文的操作
}

现在我们先定义个EntStrategy接口

/**
 * @author Richard_yyf
 * @version 1.0 2019/10/23
 */
public interface EntStrategy {

    String getStuff();

    void send();
}

三个策略类

DefaultStrategy

@Component
public class DefaultStrategy  implements EntStrategy {
    @Override
    public String getStuff() {
        return "其他企业";
    }

    @Override
    public void send() {
        System.out.println("发送默认标准的报文给对应企业");
    }

    @Override
    public String toString() {
        return getStuff();
    }
}

EntAStrategy

@Component
public class EntAStrategy implements EntStrategy {
    @Override
    public String getStuff() {
        return "企业A";
    }

    @Override
    public void send() {
        System.out.println("发送A标准的报文给对应企业");
    }

    @Override
    public String toString() {
        return getStuff();
    }
}

EntBStrategy

@Component
public class EntBStrategy implements EntStrategy {
    @Override
    public String getStuff() {
        return "企业B";
    }

    @Override
    public void send() {
        System.out.println("发送B标准的报文给对应企业");
    }

    @Override
    public String toString() {
        return getStuff();
    }
}

Step 2 借助Spring 强大的依赖注入

下面的设计是消除if/else的关键代码，这里我定义了一个EntStrategyHolder来当做工厂类。

@Component
public class EntStrategyHolder {

    // 关键功能 Spring 会自动将 EntStrategy 接口的类注入到这个Map中
    @Autowired
    private Map<String, EntStrategy> entStrategyMap;

    public EntStrategy getBy(String entNum) {
        return entStrategyMap.get(entNum);
    }
}

这一步的关键就是， Spring 会自动将 EntStrategy 接口的实现类注入到这个Map中。前提是你这个实现类得是交给Spring 容器管理的。

这个Map的key值就是你的 bean id，你可以用@Component("value")的方式设置，像我上面直接用默认的方式的话，就是首字母小写。value值则为对应的策略实现类。

 到这一步，实际上我们的期望功能大部分已经实现了，先让用一个简单的启动类试一下。

/**
 * @author Richard_yyf
 * @version 1.0 2019/10/23
 */
@Configuration
@ComponentScan("ric.study.demo.ioc.autowire_all_implementation_demo_set")
public class Boostrap {

    public static void main(String[] args) {
        String entNum = "entBStrategy";
        send(entNum);
        entNum = "defaultStrategy";
        send(entNum);
    }

    // 用这个方法模拟 企业代理服务 提供的Api
    public static void send(String entNum) {
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Boostrap.class);
        context.getBean(EntStrategyHolder.class).getBy(entNum).send();
    }
}

输出结果

发送B标准的报文给对应企业
发送默认标准的报文给对应企业

Step 3 别名转换

大家眼睛如果稍微利索的点的话，会发现我上面启动类里面的企业编号entNum填的实际上是bean id的值。那在实际业务中肯定是不会这样的，怎么可能把一个企业编号定义的这么奇怪呢。

所以这里还需要一步操作，将传入的企业编号，转义成对应的策略类的bean id。

实际上这一步的逻辑和你的实际业务是有很强的相关性的，因为在我业务里面的entNum在实际上就是一种标识，程序怎么识别解析这个标识，找到对应的策略实现类，应该是根据你的业务需求定制的。

我这里把这一步也写出来，主要是想给大家提供一种思路。

因为我的微服务是用SpringBoot做基础框架的，所以我借助SpringBoot 外部化配置的一些特性实现了这种方式。

添加EntAlias类

/**
 * @author Richard_yyf
 * @version 1.0 2019/10/15
 */
@Component
@EnableConfigurationProperties
@ConfigurationProperties(prefix = "ent")
public class EntAlias {

    private HashMap<String, String> aliasMap;

    public static final String DEFAULT_STATEGY_NAME = "defaultStrategy";

    public HashMap<String, String> getAliasMap() {
        return aliasMap;
    }

    public void setAliasMap(HashMap<String, String > aliasMap) {
        this.aliasMap = aliasMap;
    }

    String of(String entNum) {
        return aliasMap.get(entNum);
    }
}

在对应配置文件application.yml中配置：

ent:
  aliasMap:
    entA: entAStrategy
    entB: entBStrategy

....省略

这里注意哦，要实现对应getter和setter的，不然属性会注入不进去的。

改写一下EntStrategyHolder类

@Component
public class EntStrategyHolder {

    @Autowired
    private EntAlias entAlias;

    // 关键功能 Spring 会自动将 EntStrategy 接口的类注入到这个Map中
    @Autowired
    private Map<String, EntStrategy> entStrategyMap;

    // 找不到对应的策略类，使用默认的
    public EntStrategy getBy(String entNum) {
        String name = entAlias.of(entNum);
        if (name == null) {
            return entStrategyMap.get(EntAlias.DEFAULT_STATEGY_NAME);
        }
        EntStrategy entStrategy = entStrategyMap.get(name);
        if (entStrategy == null) {
            return entStrategyMap.get(EntAlias.DEFAULT_STATEGY_NAME);
        }
        return entStrategy;
    }
}

现在我们再启动一下看看：

/**
 * @author Richard_yyf
 * @version 1.0 2019/10/23
 */
@Configuration
@ComponentScan("ric.study.demo.ioc.autowire_all_implementation_demo_set")
public class Boostrap {
    public static void main(String[] args) {
        String entNum = "entA";
        send(entNum);
        entNum = "entB";
        send(entNum);
        entNum = "entC";
        send(entNum);
    }

    public static void send(String entNum) {
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Boostrap.class);
        context.getBean(EntStrategyHolder.class).getBy(entNum).send();
    }
}

输出结果

发送A标准的报文给对应企业
发送B标准的报文给对应企业
发送默认标准的报文给对应企业

非SpringBoot

上面的代码中我采用SpringBoot的特性，通过yml文件来管理别名转化，是为了让代码看起来更美观。如果是Spring框架的话。我会这样去实现。

/**
 * @author Richard_yyf
 * @version 1.0 2019/10/23
 */
public class EntAlias {

    private static Map<String, String> aliasMap;

    private static final String ENTA_STATEGY_NAME = "entAStrategy";
    private static final String ENTB_STATEGY_NAME = "entBStrategy";
    public static final String DEFAULT_STATEGY_NAME = "defaultStrategy";

    static {
        // 这个别名容器怎么注册别名、初始化，有很多种方式。
        aliasMap = new LinkedHashMap<>();
        aliasMap.put("entA", ENTA_STATEGY_NAME);
        aliasMap.put("entB", ENTB_STATEGY_NAME);
    }

    public static String of(String entNum) {
        return aliasMap.get(entNum);
    }
}

Spring IoC 的依赖注入

这里我想再谈一下上面的第二个步骤，第二个步骤的核心就是通过Spring IoC依赖注入的特性，实现了策略实现类的注册过程（这一步自己实现会需要很多工作，并且代码不会很好看）。

实际上除了Map这种变量类型，Spring 还能给List 变量进行自动装配。比如下面的代码。

@Component
public class EntStrategyHolder {

    @Autowired
    private Map<String, EntStrategy> entStrategyMap;

    @Autowired
    private List<EntStrategy> entStrategyList;

    public EntStrategy getBy(String entNum) {
        return entStrategyMap.get(entNum);
    }

    public void print() {
        System.out.println("===== implementation Map =====");
        System.out.println(entStrategyMap);
        entStrategyMap.forEach((name, impl)-> {
            System.out.println(name + ":" + impl.getStuff());
        });
        System.out.println("===== implementation List =====");
        System.out.println(entStrategyList);
        entStrategyList.forEach(impl-> System.out.println(impl.getStuff()));
    }
}

启动类

@Configuration
@ComponentScan("ric.study.demo.ioc.autowire_all_implementation_demo_set")
public class Boostrap {

    public static void main(String[] args) {
        AnnotationConfigApplicationContext context = new AnnotationConfigApplicationContext(Boostrap.class);
        context.getBean(EntStrategyHolder.class).print();
    }
}

输出结果

===== implementation Map =====
{defaultStrategy=其他企业, entAStrategy=企业A, entBStrategy=企业B}
defaultStrategy:其他企业
entAStrategy:企业A
entBStrategy:企业B
===== implementation List =====
[其他企业, 企业A, 企业B]
其他企业
企业A
企业B

 可以看到entStrategyList被成功赋值了。

只不过这个特性我暂时没有找到应用场景，所以单独拿出来说一下。