利用工廠模式+策略模式去掉if-else

需求：

假設有這么一個需求：

一個電商系統，當用戶消費滿1000 金額，可以根據用戶VIP等級，享受打折優惠。根據用戶VIP等級，計算出用戶最終的費用。

• 普通會員 不打折
• 白銀會員 優惠50元
• 黃金會員 8折
• 白金會員 優惠50元，再打7折

private static double getResult(long money, int type) {

    double result = money;

    if (money >= 1000) {
        if (type == UserType.SILVER_VIP.getCode()) {

            System.out.println("白銀會員 優惠50元");
            result = money - 50;
        } else if (type == UserType.GOLD_VIP.getCode()) {

            System.out.println("黃金會員 8折");
            result = money * 0.8;
        } else if (type == UserType.PLATINUM_VIP.getCode()) {

            System.out.println("白金會員 優惠50元，再打7折");
            result = (money - 50) * 0.7;
        } else {
            System.out.println("普通會員 不打折");
            result = money;
        }
    }

    return result;
}

工廠+策略

public interface Strategy {

    double compute(long money);

    // 返回 type
    int getType();
}


public class OrdinaryStrategy implements Strategy {

    @Override
    public double compute(long money) {
        System.out.println("普通會員 不打折");
        return money;
    }

    // 添加 type 返回
    @Override
    public int getType() {
        return UserType.SILVER_VIP.getCode();
    }
}

public class SilverStrategy implements Strategy {

    @Override
    public double compute(long money) {

        System.out.println("白銀會員 優惠50元");
        return money - 50;
    }

    // type 返回，表示該策略的type值
    @Override
    public int getType() {
        return UserType.SILVER_VIP.getCode();
    }
}

....省略剩下 Strategy

工廠類：

public class StrategyFactory {

    private Map<Integer, Strategy> map;

    public StrategyFactory() {

        List<Strategy> strategies = new ArrayList<>();

        strategies.add(new OrdinaryStrategy());
        strategies.add(new SilverStrategy());
        strategies.add(new GoldStrategy());
        strategies.add(new PlatinumStrategy());
        strategies.add(new PlatinumStrategy());

        // 看這里 看這里 看這里！  collect收集，把strategies的流的值都收集起來，轉換成Map<key,value>
        map = strategies.stream().collect(Collectors.toMap(Strategy::getType, strategy -> strategy));

        /* 等同上面
        map = new HashMap<>();
        for (Strategy strategy : strategies) {
            map.put(strategy.getType(), strategy);
        }*/
    }
　　　　
    public static class Holder {
        public static StrategyFactory instance = new StrategyFactory();
    }

    public static StrategyFactory getInstance() {
        return Holder.instance;
    }

    public Strategy get(Integer type) {
        return map.get(type);
    }
}

這里用到了靜態內部類去實現單例模式。

寫法：

package com.iot.api.design.singleton;

/**
 * @ProjectName: smartdata
 * @Package: com.iot.api.design.singleton
 * @ClassName: Singleton
 * @Author: heluwei
 * @Description: 靜態內部類完成 單例模式
 * @Date: 2020/1/16 17:49
 * @Version: 1.0
 */
 public class Singleton {
    //1:構造器
    private Singleton(){}
    //2:利用靜態內部類去創建類。
    private static class SingletonInstance{
        private static final Singleton INSTANCE = new Singleton();
    }
    //3：別人調用
    public static Singleton getInstance(){
        return SingletonInstance.INSTANCE;
    }
}

 

效果：

private static double getResult(long money, int type) {

    if (money < 1000) {
        return money;
    }

    Strategy strategy = StrategyFactory.getInstance().get(type);

    if (strategy == null){
        throw new IllegalArgumentException("please input right type");
    }

    return strategy.compute(money);
}