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

喜欢ヅ旅行 2022-10-14 13:55 134阅读 0赞

![6f14dce1a8552e8b5d855a76e956b614.png][]

*来源：juejin.im/post/5db0e910518825648f2ef355*

## 前言 ##

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

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

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

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

## 业务场景 ##

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

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

## 快速实现 ##

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

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

### Step 1 实现策略类 ###

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

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

现在我们先定义个EntStrategy接口

/**  
     * @author Richard_yyf  
     */  
    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来当做工厂类。设计模式内容可以看：[设计模式内容聚合][Link 1]

@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值则为对应的策略实现类。

![a0f25fa40e88695ec3f80cfa67b61ec1.png][] 图片

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

/**  
     * @author Richard_yyf  
     */  
    @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 外部化配置的一些特性实现了这种方式。扩展：[最全面的SpringBoot配置文件详解][SpringBoot]

添加EntAlias类

/**  
     * @author Richard_yyf  
     */  
    @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  
     */  
    @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

![b93c06582d5e0f5d6a6d6ecc295df4c3.png][] 图片

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

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

## 结语 ##

到这里，整个实现过程已经介绍完了。

**过程中用了到Spring最常用的一些注解，通过Spring IoC依赖注入的特性，实现了策略实现类的注册过程，最终实现了整个功能。**

希望能对你有所启发。

另外，如果你对上述Spring IoC 是如何对Map和List变量进行赋值感兴趣的话，我会在下一篇文章中讲解相关的源码和调试技巧。

我们搞技术的，知其然更应知其所以然嘛。

推荐好文

[强大，10k+点赞的 SpringBoot 后台管理系统竟然出了详细教程！][10k_ SpringBoot]

[分享一套基于SpringBoot和Vue的企业级中后台开源项目，代码很规范！][SpringBoot_Vue]

[能挣钱的，开源 SpringBoot 商城系统，功能超全，超漂亮][SpringBoot 1]![339f5654f31d084a4b97772c7827f0f0.png][]

[6f14dce1a8552e8b5d855a76e956b614.png]: /images/20221014/f762c40522ff44bdb949f4d1e77d7221.png
[Link 1]: http://mp.weixin.qq.com/s?__biz=MzI4Njc5NjM1NQ%3D%3D&chksm=ebd62a07dca1a31188dafea0e6984b4883fa972a32ff31da2df6f52fcc73b56c6c9454751d73&idx=4&mid=2247488811&scene=21&sn=ec20f59a5b67a59d98d221bd20d78448#wechat_redirect
[a0f25fa40e88695ec3f80cfa67b61ec1.png]: /images/20221014/a4d76123ebff4168a60c3bce63fe51d8.png
[SpringBoot]: http://mp.weixin.qq.com/s?__biz=MzI4Njc5NjM1NQ%3D%3D&chksm=ebd631cadca1b8dc4baca2ead9d3e5296c484ab646bf979ff06dced06eaa77b29e786542dcc2&idx=2&mid=2247487206&scene=21&sn=c88028181d028f5207e3b466dcf5a0e1#wechat_redirect
[b93c06582d5e0f5d6a6d6ecc295df4c3.png]: /images/20221014/69195d8475134383ab0c357dda1cfc09.png
[10k_ SpringBoot]: http://mp.weixin.qq.com/s?__biz=MzI4Njc5NjM1NQ%3D%3D&chksm=ebd5c411dca24d07446a4f22772cd12d2cc1143b62eb922a14500ae981cd57fb850ef74ca257&idx=2&mid=2247498557&scene=21&sn=84902d7030e174e152615d8fae3755cc#wechat_redirect
[SpringBoot_Vue]: http://mp.weixin.qq.com/s?__biz=MzU2MTI4MjI0MQ%3D%3D&chksm=fc799be6cb0e12f0b58479f8b031bc5ae79efdca1880ff9b19f0a107c3a88c37b9596dc73203&idx=2&mid=2247497288&scene=21&sn=65d7111a19e920df1e9416959594b2f4#wechat_redirect
[SpringBoot 1]: http://mp.weixin.qq.com/s?__biz=MzU2MTI4MjI0MQ%3D%3D&chksm=fc799a9ccb0e138a64a2671f31b8d4be1b3c061a288f7b0d1d4af02a60c45a72231446b91e00&idx=2&mid=2247497522&scene=21&sn=c5b4e596be4ac7d7f293e7e2f0ea0769#wechat_redirect
[339f5654f31d084a4b97772c7827f0f0.png]: /images/20221014/58403c619e5b44dbb71291484571ecca.png