如何去掉代码中大量if-else之多态

深碍√TFBOYSˉ_ 2022-01-11 02:21 461阅读 0赞

**关于代码中如何去掉大量的if-else**

常规代码开发中if...else是不可避免的，虽然 if...else 是必须的，但滥用 if...else 会对代码的可读性、可维护性造成很大伤害，进而危害到整个软件系统。现在软件开发领域出现了很多新技术、新概念，但 if...else 这种基本的程序形式并没有发生太大变化。使用好 if...else 不仅对于现在，而且对于将来，都是十分有意义的。今天我们就来看看如何“干掉”代码中的 if...else，还代码以清爽。

**问题一：if...else 过多**

**问题表现**  
1、if...else 过多的代码可以抽象为下面这段代码。其中只列出5个逻辑分支，但实际工作中，能见到一个方法包含10个、20个甚至更多的逻辑分支的情况。另外，if...else 过多通常会伴随着另两个问题：逻辑表达式复杂和 if...else 嵌套过深。

if (condition1) {
    
    } else if (condition2) {
    
    } else if (condition3) {
    
    } else if (condition4) {
    
    } else {
    
    }

2、if...else 过多的方法，通常可读性和可扩展性都不好。从软件设计角度讲，代码中存在过多的 if...else 往往意味着这段代码违反了违反单一职责原则和开闭原则。因为在实际的项目中，需求往往是不断变化的，新需求也层出不穷。所以，软件系统的扩展性是非常重要的。而解决 if...else 过多问题的最大意义，往往就在于提高代码的可扩展性。

如何解决  
1. 表驱动  
2. 职责链模式  
3. 注解驱动  
4. 事件驱动  
5. 有限状态机  
6. Optional  
7. Assert  
8. 多态

**本文章提供的解决方案**

1 创建一个注解 用于标记子类

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

2创建枚举类

@Getter
    @AllArgsConstructor
    public enum HandlerTypeEnum {
    
        WECHAT_PAY(10001,"微信支付"),
    
        ALIPAY_PAY(10002,"支付宝支付"),
    
        WECHAT_ORDER(20001,"微信订单"),
    
        ALIPAY_ORDER(20002,"支付宝订单");
    
        private Integer code;
    
        private String desc;
    }

3 定义一个顶级接口

public interface  BaseHandler {
    
    }

4 创建一个类实现BeanFactoryPostProcessor扫描所有被HandlerType注解所标注的类 注册到ApplicaitionContext 并存放到HandlerContext

@Component
    public class HandlerProcessor implements BeanFactoryPostProcessor {
    
        private static final String SCAN_PACKAGE = "com.along.blog" ;
    
        @Override
        public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
    
            Map handlerMap = Maps.newHashMapWithExpectedSize(3);
    
            ClassScanner classScanner = new DefaultClassScanner();
    
            classScanner.scanAndCallbackByAnno(Arrays.asList(SCAN_PACKAGE),HandlerType.class, new ScannerCallback() {
                @Override
                public void callback(List<Class> list) {
                    list.forEach(clazz -> {
                        HandlerType annotation = (HandlerType)clazz.getAnnotation(HandlerType.class);
                        HandlerTypeEnum value = annotation.value();
                        if(Objects.isNull(value)){
                            throw new IllegalArgumentException("HandlerType enmu value expect not null but null...");
                        }
                        handlerMap.put(value,clazz);
                    });
                }
            });
    
            HandlerContext context = new HandlerContext(handlerMap);
            configurableListableBeanFactory.registerResolvableDependency(HandlerContext.class,context);
        }
    }

5、 创建HandlerContext

public class HandlerContext {
    
        private static Map<HandlerTypeEnum,Class> handlerMap;
    
        public HandlerContext(Map<HandlerTypeEnum, Class> handlerMap) {
            HandlerContext.handlerMap = handlerMap;
        }
    
        public static BaseHandler getInstance(HandlerTypeEnum type){
            Class clazz = handlerMap.get(type);
            if(Objects.isNull(clazz)){
                throw new IllegalArgumentException("not found one initialized by handlerType");
            }
            return (BaseHandler) AppUtil.getBean(clazz);
        }
    }

6、 然后根据自己的业务需求去实现BaseHandler 顶级接口  
   订单类接口：（参数和返回值根据业务自定义）

public interface OrderBaseHandler extends BaseHandler {
    
        Object execute(Object param);
    }

支付类接口：

public interface  PayBaseHandler extends BaseHandler {
    
        Object execute(Object param);
    
    }

\- 具体业务实现类

@Component
    @HandlerType(HandlerTypeEnum.ALIPAY_ORDER)
    public class AlipayOrderHandler implements OrderBaseHandler {
    
        @Override
        public Object execute(Object param) {
            System.out.println("支付宝订单");
            return null;
        }
    }

@Component
    @HandlerType(HandlerTypeEnum.WECHAT_ORDER)
    public class WeChatOrderHandler implements OrderBaseHandler {
    
        @Override
        public Object execute(Object param) {
            System.out.println("微信订单");
            return null;
        }
    }

@Component
    @HandlerType(HandlerTypeEnum.ALIPAY_PAY)
    public class AlipayPayHandler implements PayBaseHandler {
    
        @Override
        public Object execute(Object param) {
            System.out.println("支付宝支付");
            return null;
        }
    }

\- 具体测试类

PayBaseHandler instance = (PayBaseHandler) HandlerContext.getInstance(HandlerTypeEnum.ALIPAY_PAY);
            instance.execute("");
            PayBaseHandler instance1 = (PayBaseHandler) HandlerContext.getInstance(HandlerTypeEnum.WECHAT_PAY);
            instance1.execute("");
            OrderBaseHandler instance2 = (OrderBaseHandler) HandlerContext.getInstance(HandlerTypeEnum.ALIPAY_ORDER);
            instance2.execute("");
            OrderBaseHandler instance3 = (OrderBaseHandler) HandlerContext.getInstance(HandlerTypeEnum.WECHAT_ORDER);
            instance3.execute("");

\- 输出结果

支付宝支付  
微信支付  
支付宝订单  
微信订单  
\`\`\`   
sout  
\`\`\`