Java利用策略模式實現條件判斷,告別if else

2022-02-24 16:00:18

定義

策略模式定義了一系列演演算法，並且將每個演演算法封裝起來，使得他們可以相互替換，而且演演算法的變化不會影響使用演演算法的使用者端。

使用場景

一個系統需要動態的在幾種演演算法中選擇一種，可以把每個演演算法封裝到具體的策略類中
一個類中定義了多種行為，可以去代替條件轉移語句，減少寫死
系統中各個演演算法或者說函數是彼此獨立的，而且要求對客戶隱藏演演算法具體實現細節的時候
多個類只區別在表現行為的不同，可以使用策略模式，在執行時動態的選擇要執行的行為

案例

需求

根據不同的水果口味，製作不同的蛋糕，比如現在有蘋果味和香蕉味的蛋糕，入參為蘋果味，則製作蘋果味蛋糕；入參為香蕉味，則製作香蕉味的蛋糕

實現方案

定義製作蛋糕的抽象類

/**
 * 蛋糕製作抽象類
 * @author:liyajie
 * @createTime:2022/2/24 10:53
 * @version:1.0
 */
public abstract class CakeHandler {
    /**
     * 製作蛋糕
     * @author: liyajie
     * @date: 2022/2/24 10:54
     * @param
     * @return void
     * @exception:
     * @update:
     * @updatePerson:
     **/
    public abstract void makeCake();
}

定義製作蘋果味蛋糕的策略類，繼承製作蛋糕抽象類，重寫makeCake方法

/**
 * 製作蘋果蛋糕策略
 * @author:liyajie
 * @createTime:2022/2/24 10:55
 * @version:1.0
 */
public class AppleCakeHandler extends CakeHandler{
    @Override
    public void makeCake() {
        System.out.println("正在製作蘋果蛋糕");
    }
}

定義製作香蕉味蛋糕的策略類，繼承製作蛋糕抽象類，重寫makeCake方法

/**
 * 製作香蕉蛋糕策略
 * @author:liyajie
 * @createTime:2022/2/24 10:55
 * @version:1.0
 */
public class BananaCakeHandler extends CakeHandler{
    @Override
    public void makeCake() {
        System.out.println("正在製作香蕉蛋糕");
    }
}

定義列舉類，來製作蛋糕的策略

/**
 * 製作蛋糕列舉
 * @author:liyajie
 * @createTime:2022/2/24 10:57
 * @version:1.0
 */
public enum CakeEnum {
    APPLE(AppleCakeHandler.class.getSimpleName(),new AppleCakeHandler()),
    BANANA(BananaCakeHandler.class.getSimpleName(),new BananaCakeHandler());

    private final String cakeType;
    private final CakeHandler cakeHandler;
    CakeEnum(String cakeType, CakeHandler cakeHandler){
        this.cakeType = cakeType;
        this.cakeHandler = cakeHandler;
    }

    // 匹配策略類
    public static CakeEnum match(String cakeType){
        CakeEnum[] values = CakeEnum.values();
        for (CakeEnum cakeEnum : values) {
            if(cakeType.equals(cakeEnum.cakeType)){
                return cakeEnum;
            }
        }
        return null;
    }

    public String getCakeType(){
        return cakeType;
    }

    public CakeHandler getCakeHandler(){
        return cakeHandler;
    }
}

定義測試類

/**
 * 測試類
 * @author:liyajie
 * @createTime:2022/2/24 11:07
 * @version:1.0
 */
public class Test {
    public static void main(String[] args) {
        String cakeType = AppleCakeHandler.class.getSimpleName();
        CakeEnum cakeEnum = CakeEnum.match(cakeType);
        CakeHandler cakeHandler = cakeEnum.getCakeHandler();
        cakeHandler.makeCake();

        cakeType = BananaCakeHandler.class.getSimpleName();
        cakeEnum = CakeEnum.match(cakeType);
        cakeHandler = cakeEnum.getCakeHandler();
        cakeHandler.makeCake();
    }
}

檢視測試結果