用设计模式消除大量if-else

快来打我* 2021-09-23 12:12 575阅读 0赞

## 前言 ##

先来看张经典的啊都给神图，感受下大量if-else的“魅力”

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2d3ZDExNTQ5NzgzNTI_size_16_color_FFFFFF_t_70][]

有时候业务上的拖拉可能要求我去写出如上的多层if-else嵌套代码，如果你碰上了一个对团队成员要求相对比较严格的主管的话，那么恭喜你，死定了……脾气比较爆的大佬可能直接就跟你说要么解决上面问题，要么收拾包袱滚蛋了。

那上述代码能否用设计模式相关解决呢，答案是：YES，可以用策略模式+工厂模式。

限于本文篇幅，上述的两种模式这边就不详细介绍了，不懂的可以直接百度，本篇博文简单粗暴，会以最暴力的代码来解决问题。

## 原始代码 ##

public class Coffee {
        public int getPrice(int type){
            if(type==1){
                System.out.println("卡布奇诺");
                return 22;
            }
            if(type==2){
                System.out.println("拿铁");
                return 25;
            }
            if(type==3){
                System.out.println("美咖");
                return 20;
            }
            return 21;
        }
    }

## 改进后的代码 ##

### 策略模式 ###

分别定义这几个咖啡

public interface Coffee {
        int getPrice();
    }

public class Cappuccino implements Coffee {
        @Override
        public int getPrice() {
            return 22;
        }
    }

public class AmericanoCafe implements Coffee {
        @Override
        public int getPrice() {
            return 20;
        }
    }

public class Latte implements Coffee {
        @Override
        public int getPrice() {
            return 25;
        }
    }

那就可以根据type来new相应对象了，但实质还没有解决上述多if问题。

### 工厂模式 ###

public class CoffeeFactory {
        private static Map<Integer,Coffee> coffeeMap=new HashMap<>();
    
        public static Coffee buildCoffee(int type){
            return coffeeMap.get(type);
        }
    
        public static void register(int type,Coffee coffee){
            coffeeMap.put(type,coffee);
        }
    }

### 改进后的策略 ###

此处会发现，每个具体的Coffee必须在初始化的时候将自身注册到这个对应的coffeeMap中，否则在buildCoffee的时候会返回null,那该如何修改呢？看下面代码

@Component
    public class Cappuccino implements Coffee, InitializingBean {
        @Override
        public int getPrice() {
            return 22;
        }
    
        @Override
        public void afterPropertiesSet() throws Exception {
            CoffeeFactory.register(1,this);
        }
    }

@Component
    public class AmericanoCafe implements Coffee, InitializingBean {
        @Override
        public int getPrice() {
            return 20;
        }
    
        @Override
        public void afterPropertiesSet() throws Exception {
            CoffeeFactory.register(2,this);
        }
    }

@Component
    public class Latte implements Coffee, InitializingBean {
        @Override
        public int getPrice() {
            return 25;
        }
    
        @Override
        public void afterPropertiesSet() throws Exception {
            CoffeeFactory.register(3,this);
        }
    }

**之所以要实现InitializingBean接口是为了Spring初始化的时候，创建对应的Coffee Bean的时候将其注册到工厂map中**

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2d3ZDExNTQ5NzgzNTI_size_16_color_FFFFFF_t_70]: /images/20210923/b6a3d64ad03d4034891cb7ece92d91e1.png