springBoot中怎么减少if---else，怎么动态手动注册类进入Spring容器

由于业务中经常有需要判断的if--eles操作，层层嵌套，看起来程序的可读性太差，结合策略模式进行改造

方法一、一般有策略模式  +  工厂模式进行代码的优化，减少 if---else；

方法二、还有就是利用策略模式  +  SpringBoot提供的某些类  进行包装

本次介绍采用方法二的方式，大概的思路是：

1、策略模式：将所有同类型的操作抽象出来一个接口(这个接口包含一个动作方法) 和 一个实现了接口的抽象类(不实现方法)；
2、根据需求，将同类型的操作抽象成一个一个产品，继承第一步的抽象类，并实现抽象方法，编写自己的业务逻辑，注意此类需要注入spring容器；
3、自定义一个类级别注解，用来区分不同操作类型的标识，此自定义标识要有返回一个类型的属性；
4、抽象出来一个处理所有产品的公共HandlerContext对象，此对象提供一个获取具体产品类的方法，该方法有个入参用于表明是具体那个产品，同时该HandlerContext对象还具有Map类型的属性变量，
存储key为具体的类型，value为具体的产品类对象，该Map对象通过构造函数的方式注入初始化进来；
5、编写一个加载所有产品类的全局process类，用于扫描加了注解@HandlerType的所有实现产品,给存储key 和 value产品对象Map赋值，初始化HandlerContext 将其注册到spring容器中；

需求

这里虚拟一个业务需求，让大家容易理解。假设有一个订单系统，里面的一个功能是根据订单的不同类型作出不同的处理。

订单实体：

service接口：

传统实现

根据订单类型写一堆的if else：

下面采用方法二来进行优化：

1、策略模式：

将所有同类型的操作抽象出来一个接口(这个接口包含一个动作方法) 和 一个实现了接口的抽象类(不实现方法)；

先定义一个数据传输的实体类DTO     OrderDTO

 @Data
 public class OrderDTO {

     private String code;

     private BigDecimal price;

     /**
      * 订单类型
      * 1：普通订单
      * 2：团购订单
      * 3：促销订单
      */
     private String orderType;
 }

定义一个抽象类的接口：

IHandlerService

 /**
  * <p>Title: com.aier.cloud.biz.simplify</p>
  * <p>Company:爱尔集团信息中心</p>
  * <p>Copyright:Copyright(c)</p>
  * User: duanm
  * Date: 2019/10/31 15:33
  * Description: No Description
  */
 public interface IHandlerService {

     String handler(OrderDTO orderDTO);
 }

AbstractHandlerService：抽象类

 public abstract class AbstractHandlerService  implements IHandlerService {

     abstract public String handler(OrderDTO orderDTO);

 }

2、根据需求，将同类型的操作抽象成一个一个产品，继承第一步的抽象类，并实现抽象方法，编写自己的业务逻辑，注意此类需要注入spring容器；

团购订单处理类：

 @Component
 @HandlerType(value = "2")
 public class GroupHandler extends AbstractHandlerService {

     @Override
     public String handler(OrderDTO orderDTO) {
         return "处理团购订单";
     }
 }

普通订单处理类：

 @Component
 @HandlerType(value = "1")
 public class NormalHandler extends AbstractHandlerService {

     @Override
     public String handler(OrderDTO orderDTO) {
         return "处理普通订单";
     }
 }

促销订单处理类：

 @Component
 @HandlerType(value = "3")
 public class PromotionHandler extends AbstractHandlerService {
     @Override
     public String handler(OrderDTO orderDTO) {
         return "处理促销订单";
     }
 }

注意事项：必须添加  @Component  注解，注入Spring容器      下面编写自定义的注解实现  HandlerType

3、自定义一个类级别注解，用来区分不同操作类型的标识，此自定义标识要有返回一个类型的属性；

 @Target({ElementType.TYPE})
 @Retention(RetentionPolicy.RUNTIME)
 @Documented
 @Inherited
 public @interface HandlerType {
     String value();
 }

4、 抽象出来一个公共HandlerContext对象

抽象出来一个处理所有产品的公共HandlerContext对象，此对象提供一个获取具体产品类的方法，该方法有个入参用于表明是具体那个产品，同时该HandlerContext对象还具有Map类型的属性变量，
存储key为具体的类型，value为具体的产品类对象，该Map类型变量初始化通过构造函数的方式注入；

 public class HandlerContext {

     private Map<String, Class> handlerMap;

     public HandlerContext(Map<String, Class> handlerMap) {
         this.handlerMap = handlerMap;
     }

     public AbstractHandlerService getInstance(String type) {
         Class clazz = handlerMap.get(type);
         if (clazz == null) {
             throw new IllegalArgumentException("not found handler for type : " + type);
         }
         return (AbstractHandlerService) BeanTool.getBean(clazz);
     }
 }

5、编写一个加载所有产品类的全局process类，用于扫描加了注解@HandlerType的所有实现产品,给存储key 和 value产品对象Map赋值，初始化HandlerContext 将其注册到spring容器中；

 @Component
 public class HandlerProcessor implements BeanFactoryPostProcessor {

     private static final String HANDLER_PACKAGE = "com.aier.cloud.biz.simplify";

     /**
      * 扫描@HandlerType,初始化HandlerContext 将其注册到spring容器中
      *
      * @param configurableListableBeanFactory
      * @throws BeansException
      */
     @Override
     public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
         Map<String, Class> handlerMap = Maps.newHashMap();

         ClassPathScanningCandidateComponentProvider provider = new ClassPathScanningCandidateComponentProvider(false);
         provider.addIncludeFilter(new AnnotationTypeFilter(HandlerType.class));
         Set<BeanDefinition> candidateComponents = provider.findCandidateComponents(HANDLER_PACKAGE);
         candidateComponents.forEach(Beanclass -> {
             try {
                 Class<?> clazz = Class.forName(Beanclass.getBeanClassName());
                 //获取注解中的类型值
                 String type = clazz.getAnnotation(HandlerType.class).value();
                 //将注解中的类型值做为key，对应的类作为value 保存在handlerMap中
                 handlerMap.put(type, clazz);
             } catch (ClassNotFoundException e) {
                 e.printStackTrace();
             }
         });

         //初始化HandlerContext类，将其注入到Spring容器中
         HandlerContext handlerContext = new HandlerContext(handlerMap);
         configurableListableBeanFactory.registerSingleton(HandlerContext.class.getName(), handlerContext);

     }
 }

注意：主要使用了spring的资料加载工具类，把所有的产品实现类都扫描 存储到map中，并利用继承  BeanFactoryPostProcessor   通过实现它的方法，动态的注入 HandlerContext 进入spring容器

自定义注解和抽象处理器都很简单，那么如何将处理器注册到spring容器中呢？

具体思路是：

1、扫描指定包中标有@HandlerType的类；

2、将注解中的类型值作为key，对应的类作为value，保存在Map中；

3、以上面的map作为构造函数参数，初始化HandlerContext，将其注册到spring容器中；

几个关键的工具类：

 @Component
 public class SpringContextUtils implements ApplicationContextAware {

     private static ApplicationContext applicationContext;

     @Override
     public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
         SpringContextUtils.applicationContext = applicationContext;
     }

     /**
      * 取得存储在静态变量中的ApplicationContext.
      */
     public static ApplicationContext getApplicationContext() {
         checkApplicationContext();
         return applicationContext;
     }

     /**
      * 清除applicationContext静态变量.
      */
     public static void cleanApplicationContext() {
         applicationContext = null;
     }

     private static void checkApplicationContext() {
         if (applicationContext == null) {
             throw new IllegalStateException("applicaitonContext未注入,请在applicationContext.xml中定义SpringContextHolder");
         }
     }

     /**
      * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
      */
     @SuppressWarnings("unchecked")
     public static <T> T getBean(String name) {
         checkApplicationContext();
         return (T) applicationContext.getBean(name);
     }

     /**
      * 从静态变量ApplicationContext中取得Bean, 自动转型为所赋值对象的类型.
      */
     @SuppressWarnings("unchecked")
     public static <T> T getBean(Class<T> clazz) {
         checkApplicationContext();
         return (T) applicationContext.getBeansOfType(clazz);
     }
 }

 public class BeanTool {

     public static  <T> T getBean(Class<T> clazz) {
         String clazzName = clazz.getName();
         clazzName = clazzName.replace(".",",");
         String[] split = clazzName.split(",");
         String name = split[split.length - 1];
         return SpringContextUtils.getBean(lowerFirst(name));
     }

     public static String lowerFirst(String oldStr) {
         char[] chars = oldStr.toCharArray();
         chars[0] += 32;
         return String.valueOf(chars);
     }
 }

注意此处可以优化下，可以简单来获取类的名字，

 public static  <T> T getBean(Class<T> clazz) {
         //获取类的名字
         String simpleName = clazz.getSimpleName();
         //根据类的名称获取类的实列对象
         return SpringContextUtils.getBean(lowerFirst(simpleName));
     }

思考：  还是JAVA反射 不太熟悉，导致走了弯路来处理，JAVA反射还是要吃透，多写写。

（1）反射机制极大的提高了程序的灵活性和扩展性，降低模块的耦合性，提高自身的适应能力。

（2）通过反射机制可以让程序创建和控制任何类的对象，无需提前硬编码目标类。

（3）使用反射机制能够在运行时构造一个类的对象、判断一个类所具有的成员变量和方法、调用一个对象的方法。

（4）反射机制是构建框架技术的基础所在，使用反射可以避免将代码写死在框架中。

6、测试运行代码，查看结果

测试运行：在service层进行调用，可以如下来编写：

 @Service
 public class OrderServiceImpl implements IOrderService {

     @Resource
     private HandlerContext handlerContext;

     @Override
     public String handle(OrderDTO orderDTO) {
         IHandlerService instance = handlerContext.getInstance(orderDTO.getOrderType());
         System.out.println(instance.handler(orderDTO));
         return instance.handler(orderDTO);
     }
 }

运行结果如下：

最后请注意一点，HandlerProcessor和BeanTool必须能被扫描到，或者通过@Bean的方式显式的注册，才能在项目启动时发挥作用。

总结

利用策略模式可以简化繁杂的if else代码，方便维护，而利用自定义注解和自注册的方式，可以方便应对需求的变更。本文只是提供一个大致的思路，还有很多细节可以灵活变化，例如使用枚举类型、或者静态常量，作为订单的类型，相信你能想到更多更好的方法。

2、后续追加更优雅处理

针对实现了策略模式的具体操作类，在业务处理类中，可以通过spring的已有功能进行处理，怎么通过不同的类型，获取到该类型的实现类。

通过业务类的构造方法，

hashMap  hp = new hashMap();// 最好采用线程安全的MAP对象，需要优化

@Autowired
public AemrMessageServiceImpl(List<策略模式的接口对象参数>){
   // 初始化map，存储起来

   map.put(业务类型key，业务类型实现)
}

思路：

1、通过spring的IOC快速实现通过类型type注入这个type的所有实现；

2、通过申明一个线程安全的Map初始化数据，在业务类的构造函数中初始化进来，map.put(业务类型，策略模式业务的具体实现类)；

3、在具体使用的业务类处理方法中，通过类型获从map中拿到具体的实现类型，完成调用；