記一次重構，策略模式替換if else

弓長日含發表於2020-09-25

場景

先說一下大概要做的事情。我需要先從資料庫A中篩選出一週的基礎資料，呼叫介面獲取資料庫B中的規則，再做一次資料篩選，然後入庫資料庫B。定時任務每週跑一次。
吶，說到這裡，也大概知道了，在資料篩選那裡，將會有業務判斷。虛擬碼就集中展示那裡。

第1版

考慮時間等各種因素，先實現再說。此處並沒有用到資料庫B中的規則，直接在專案中寫的固定值。

for(CustomerResource customerResource : customerResourceList){
	//TODO 
	//V1
	//條件1 
	if(customerResource.getFactoryPrice().intValue() < -10){
	    condition1 = true;
	}else {
	    continue;//條件1未滿足，直接忽略
	}
	//條件2 
	if(customerResource.getLastWeekSales().intValue() > 100){
	    condition2 = true;
	}else {
	    continue;//條件2未滿足，直接忽略
	}
	//條件3 
	if(factoryDay.get(customerResource.getFactoryCode()) == null ? false : factoryDay.get(customerResource.getFactoryCode()) > 3){
	    condition3 = true;
	}else{condition3 = false;}
	//條件4 
	if(factorySales.get(customerResource.getFactoryCode()) == null ? false :
	        customerResource.getLastWeekSales().divide(factorySales.get(customerResource.getFactoryCode()),2,BigDecimal.ROUND_HALF_UP).doubleValue() > 1.1){
	    condition4 = true;
	}else{condition4 = false;}
	
	//根據4個條件歸類
}

第2版

等到第1版本實現以後，篩選出的資料基本上已經是想要的資料。但是不利於規則的擴充套件。便於維護，後續在後管系統做頁面。

存在每個工廠的規則數量(目前4條)是一樣，但是規則中比較的數值存在變動，比較存在變動(符號)。

如此一來，第1版本顯然已經不符合需求了。

for(CustomerResource customerResource : customerResourceList){
	//獲取工廠的規則
    List<CustomerResourceRule> rules = factoryRulesMap.get(customerResource.getFactoryCode());
    if(rules == null){//未獲取到該工廠的規則
        continue;
    }
    for(CustomerResourceRule rule : rules){
		// V2
		//條件1 
		if(ConditionEnum.CONDITION1.getRuleCode().equals(rule.getRuleCode())){
		    switch (SymbolEnum.getBySymbol(rule.getSymbol())){
		        case GT:
		            condition1 = customerResource.getFactoryPrice().intValue() > Integer.parseInt(rule.getSetValue());
		            break;
		        case LT:
		            condition1 = customerResource.getFactoryPrice().intValue() < Integer.parseInt(rule.getSetValue());
		            break;
		        default:break;
		    }
		}
		//條件2 
		if(ConditionEnum.CONDITION2.getRuleCode().equals(rule.getRuleCode())){
		    switch (SymbolEnum.getBySymbol(rule.getSymbol())){
		        case GT:
		            condition2 = customerResource.getLastWeekSales().intValue() > Integer.parseInt(rule.getSetValue());
		            break;
		        case LT:
		            condition2 = customerResource.getLastWeekSales().intValue() > Integer.parseInt(rule.getSetValue());
		            break;
		        default:break;
		    }
		}
		//條件3 
		if(ConditionEnum.CONDITION3.getRuleCode().equals(rule.getRuleCode())){
		    List<FactoryClinkerRatio> ratios = factoryRatioMap.get(customerResource.getFactoryCode());
		    if(ratios != null){        
		        switch (SymbolEnum.getBySymbol(rule.getSymbol())){
		            case GT:
		                List<FactoryClinkerRatio> ratioList = ratios.stream().filter(s -> s.getRatios().doubleValue() > Double.parseDouble(rule.getSetValue())).collect(Collectors.toList());
		                condition3 = ratioList.size() > 3;
		                break;
		            case LT:
		                List<FactoryClinkerRatio> ratioList1 = ratios.stream().filter(s -> s.getRatios().doubleValue() < Double.parseDouble(rule.getSetValue())).collect(Collectors.toList());
		                condition3 = ratioList1.size() > 3;
		                break;
		            default:break;
		        }
		    }
		}
		//條件4 
		if(ConditionEnum.CONDITION4.getRuleCode().equals(rule.getRuleCode())){
		    BigDecimal salePlan = factorySalesMap.get(customerResource.getFactoryCode());
		    if(salePlan != null){
		        switch (SymbolEnum.getBySymbol(rule.getSymbol())){
		            case GT:
		                condition4 = customerResource.getLastWeekSales().divide(salePlan,2,BigDecimal.ROUND_HALF_UP).doubleValue() > Double.parseDouble(rule.getSetValue());
		                break;
		            case LT:
		                condition4 = customerResource.getLastWeekSales().divide(salePlan,2,BigDecimal.ROUND_HALF_UP).doubleValue() < Double.parseDouble(rule.getSetValue());
		                break;
		            default:break;
		        }
		    }
		}

		//根據4個條件歸類
	}
}

第3版

第二版的程式碼已經基本實現了功能。那時專案經理告訴我，還有更好的寫法，去了解一下策略模式。

就開始了策略模式的學習之路，邊學習邊實踐。(關於設計模式，後續有空會在設計模式專欄中更新，此處就先不介紹了)。
下圖為本次專案中的策略類圖：
在這裡插入圖片描述

重構步驟

輔助類

列舉

/**
 * 條件列舉
 * Created by zhanghan_a on 2020/6/10.
 */
public enum ConditionEnum {
    CONDITION1("1"),
    CONDITION2("2"),
    CONDITION3("3"),
    CONDITION4("4");

    private String ruleCode;

    ConditionEnum(String ruleCode){
        this.ruleCode = ruleCode;
    }

    public String getRuleCode(){
        return ruleCode;
    }

    public static ConditionEnum getConditionByCode(String ruleCode){
        for(ConditionEnum conditionEnum : values()){
            if(conditionEnum.getRuleCode().equals(ruleCode)){
                return conditionEnum;
            }
        }
        return null;
    }
}

/**
 * 符號列舉
 * Created by zhanghan_a on 2020/6/10.
 */
public enum SymbolEnum {
    GT(">"),
    LT("<"),
    WRONG("wrong");

    private String symbol;

    SymbolEnum(String symbol){
        this.symbol = symbol;
    }

    public String getSymbol(){
        return this.symbol;
    }

    public static SymbolEnum getBySymbol(String symbol){
        for(SymbolEnum symbolEnum : values()){
            if(symbolEnum.getSymbol().equals(symbol)){
                return symbolEnum;
            }
        }
        return WRONG;//返回錯誤的比較符號
    }
}

自定義註解

/**
 * Created by zhanghan_a on 2020/6/12.
 */
@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface ConditionTypeAnnotation {
    ConditionEnum type();
}

在專案中新建策略包：strategy

策略類介面

/**
 * Created by zhanghan_a on 2020/6/12.
 */
public interface ConditionStrategy {
    String check(ConditionParams conditionParams);
}

策略實現類

@Service
@ConditionTypeAnnotation(type = ConditionEnum.CONDITION1)
public class FirstCondition implements ConditionStrategy{
    @Override
    public String check(ConditionParams conditionParams) {
        //判斷
        boolean condition = false;
        //TODO 條件1的判斷
        return condition ? ConditionEnum.CONDITION1.getRuleCode() : "";
    }
}

@Service
@ConditionTypeAnnotation(type = ConditionEnum.CONDITION2)
public class SecondCondition implements ConditionStrategy {
    @Override
    public String check(ConditionParams conditionParams) {
        //判斷
        boolean condition = false;
        //TODO 條件2的判斷
        return condition ? ConditionEnum.CONDITION2.getRuleCode() : "";
    }
}

@Service
@ConditionTypeAnnotation(type = ConditionEnum.CONDITION3)
public class ThirdCondition implements ConditionStrategy {
    @Override
    public String check(ConditionParams conditionParams) {
        //判斷
        boolean condition = false;
        //TODO 條件3的判斷
        return condition ? ConditionEnum.CONDITION3.getRuleCode() : "";
    }
}

@Service
@ConditionTypeAnnotation(type = ConditionEnum.CONDITION4)
public class ForthCondition implements ConditionStrategy {
    @Override
    public String check(ConditionParams conditionParams) {
        //判斷
        boolean condition = false;
        //TODO 條件4的判斷
        return condition ? ConditionEnum.CONDITION4.getRuleCode() : "";
    }
}

在config資料夾下或util資料夾下新增類

新增StrategyContext類

/**
 * Created by zhanghan_a on 2020/6/12.
 */
public class StrategyContext {
    private Map<ConditionEnum,Class> strategyMap;

    public StrategyContext(Map<ConditionEnum, Class> strategyMap) {
        this.strategyMap = strategyMap;
    }

    public ConditionStrategy getStrategy(ConditionEnum conditionEnum){
        if(conditionEnum == null){
            return null;
        }
        if(strategyMap == null){
            return null;
        }

        Class clazz = strategyMap.get(conditionEnum);
        if(clazz == null){
            return null;
        }
        return (ConditionStrategy)BeanTools.getBean(clazz);
    }
}

新增StrategyProcessor類

此處用到了反射工具包。請在pom.xml中新增依賴

<dependency>
    <groupId>org.reflections</groupId>
    <artifactId>reflections</artifactId>
    <version>0.9.10</version>
</dependency>

注意新增@Component 註解

/**
 * Created by zhanghan_a on 2020/6/12.
 */
@Component
public class StrategyProcessor implements BeanFactoryPostProcessor {

    private static final String BASE_PACKAGE = "com.**.**.strategy";//此處填寫對應專案的包名即可

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
        Map<ConditionEnum,Class> strategyMap = new HashMap<>();
        Reflections reflections = new Reflections(BASE_PACKAGE);//掃描策略包
        reflections.getTypesAnnotatedWith(ConditionTypeAnnotation.class).forEach(clazz -> {//尋找ConditionTypeAnnotation註解的類
            ConditionEnum conditionEnum = clazz.getAnnotation(ConditionTypeAnnotation.class).type();//獲取對應的條件
            strategyMap.put(conditionEnum,clazz);//對映條件對應的策略實現類
        });
        StrategyContext strategyContext = new StrategyContext(strategyMap);
        configurableListableBeanFactory.registerSingleton(StrategyContext.class.getName(),strategyContext);//註冊為單例
    }
}

在之前的業務實現類中新增strategyContext

@Autowired
private StrategyContext strategyContext;

替換掉第二版的程式碼

for(CustomerResource customerResource : customerResourceList){
	//獲取工廠的規則
    List<CustomerResourceRule> rules = factoryRulesMap.get(customerResource.getFactoryCode());
    if(rules == null){//未獲取到該工廠的規則
        continue;
    }
    for(CustomerResourceRule rule : rules){
		// V3
		ConditionEnum conditionEnum = ConditionEnum.getConditionByCode(rule.getRuleCode());//根據規則程式碼獲取對應的列舉型別
		if(conditionEnum == null){
            continue;
        }
        conditionParams.setRule(rule);//設定條件引數
        ConditionStrategy conditionStrategy = strategyContext.getStrategy(conditionEnum);//根據列舉型別獲取對應的實現類
        resultList.add(conditionStrategy.check(conditionParams));//每次篩選的條件記錄在resultList
		//根據4個條件歸類
	}
}

這樣一來，簡潔了很多，也便於後續的擴充套件。

附

小編工作兩年有餘，知識的廣度與深度均不夠，尚在學習中。如果您在閱讀中發現各種錯誤，還望指出及時糾正。
本篇只是提供一個思路，畢竟每個專案中的業務場景不同。有了這個思路就足夠了，剩下的只需要你去找幾篇類似的部落格，結合自己的需求，大膽嘗試。
當初參考來自於簡書的一篇文章《還在業務中用if else，策略模式瞭解一下》，大概是作者刪除或者設為私密，我留下的連結已經失效，就不在此處分享。還是非常感謝作者給了案例作為參考。再者也感謝專案經理的提點，才有了新的收穫。
最後，程式碼的重構優化也是從第一步先實現，再到抽象，封裝一步步的來的。沒有最優，只有更優。逐漸讓程式碼越來越靈活。

if else與策略模式
2023-03-08
模式
策略模式+工廠模式取代if{}else{}
2024-04-03
模式
使用列舉ENUM替換Switch或If-Else
2019-03-03
記一次專案重構
2020-08-10
記一次程式碼重構
2018-08-09
記一次正規表示式替換，使用 idea
2020-03-28
Idea
專案重構之路——Fresco非侵入式替換Glide
2017-06-25
IDE
利用策略模式優化過多 if else 程式碼
2019-01-30
模式優化
使用策略模式重構電商折扣和支付場景
2021-11-05
模式
記一次 Vue 專案的重構
2017-05-31
Vue
還在使用 if else 寫程式碼？試試 “策略模式” 吧！
2020-05-08
模式
Swift 中單例模式的替換
2019-02-20
Swift單例模式
LRU快取替換策略及C#實現
2023-04-05
快取C#
過多if - else 的問題, 以及策略模式 + 反射解決方法
2021-03-10
模式反射
記一次使用策略模式優化程式碼的經歷
2020-11-21
模式優化
bash 引數替換中的模式匹配
2020-04-05
模式
結構型：策略模式
2023-03-27
模式
實際業務中使用策略模式對程式碼進行重構
2023-05-20
模式
重構與模式
2010-01-27
模式
webpack入門筆記——熱替換
2018-05-13
Web筆記
委派模式與策略模式記錄
2020-10-07
模式
設計模式例項講解 - 里氏替換
2020-01-04
設計模式
專案中的if else太多了，該怎麼重構？
2020-01-11
HTML 替換元素與非替換元素
2018-11-16
HTML
Spring 實現策略模式--自定義註解方式解耦if...else
2021-05-22
Spring模式解耦
記一次百萬行WPF專案程式碼的重構記錄
2022-05-13
設計模式筆記：策略模式（Strategy）
2013-12-30
設計模式筆記
重構 - 觀察者模式
2018-05-30
模式
Find and Replace Pattern（C++查詢和替換模式）
2021-01-03
C++模式
替換tabBar
2016-04-09
tabBar
替換(substitution)
2008-04-27
重構一次外包專案
2017-07-07
嘻哈說：設計模式之里氏替換原則
2019-01-19
設計模式
使用策略模式和簡單工廠模式重寫支付模組
2021-03-18
模式
重構模式（一）---- 介紹
2004-10-23
模式
js replace替換字串，同時替換多個方法
2018-11-26
JS字串
SQL Server 替換
2021-01-01
SQLServer
08:字元替換
2017-02-28
字元

記一次重構，策略模式替換if else

場景

第1版

第2版

第3版

重構步驟

附

相關文章