枚举策略代替项目中大片的if-else

客官°小女子只卖身不卖艺 2023-05-30 03:52 58阅读 0赞

之前看过很多用`策略模式`代替if-else的文章，看过之后受到启发，于是我也在项目中应用了`策略模式`，但是我用的是`枚举策略`替换掉了项目中大片的if-else。并且在`Effective Java 第二版`的第30条中推荐使用`枚举策略模式`，所以用一个小demo记录一下`枚举策略模式`。

##### 不使用策略模式 #####

如果不使用策略模式，那么代码就会像如下一样，看起来很臃肿。

public String handle(OrderDTO dto) { 
            String type = dto.getType();
            if ("1".equals(type)) { 
                return "处理普通订单";
            } else if ("2".equals(type)) { 
                return "处理团购订单";
            } else if ("3".equals(type)) { 
                return "处理促销订单";
            }
            return null;
        }

##### 什么是策略模式 #####

策略模式是一种常见的设计模式，应用比较简单，官方定义是：定义一组算法，将每个算法都封装起来，并且使他们之间互相转换。策略模式的核心就是面向对象的继承和多态机制。之前写过一篇[策略模式][Link 1]的文章，不太了解[策略模式][Link 1]的小伙伴可以先去看看，然后再回来看下文。

##### 枚举策略模式实现处理订单 #####

*  策略模式通用类图

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhbmdjaGVuZ21pbmcx_size_16_color_FFFFFF_t_70]

*  定义一个策略枚举类

@AllArgsConstructor
    @Getter
    public enum OrderEnum { 
    
        /** * 处理普通订单 */
        NORMAL_TYPE { 
            @Override
            public String handle() { 
                return "处理普通订单";
            }
        },
    
        /** * 处理团购订单 */
        GROUP_TYPE { 
            @Override
            public String handle() { 
                return "处理团购订单";
            }
        },
    
        /** * 处理促销订单 */
        PROMOTION_TYPE { 
            @Override
            public String handle() { 
                return "处理促销订单";
            }
        };
    
        public abstract String handle();
    }

*  测试

@Test
        void strategyTest() { 
            String handle1 = OrderEnum.NORMAL_TYPE.handle();
            String handle2 = OrderEnum.GROUP_TYPE.handle();
            String handle3 = OrderEnum.PROMOTION_TYPE.handle();
    
            assert handle1.equals("处理普通订单");
            assert handle2.equals("处理团购订单");
            assert handle3.equals("处理促销订单");
        }

[Link 1]: https://blog.csdn.net/wangchengming1/article/details/85068565
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhbmdjaGVuZ21pbmcx_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191107110448750.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dhbmdjaGVuZ21pbmcx,size_16,color_FFFFFF,t_70