利用策略模式優化過多 if else 程式碼

crossoverJie發表於2019-01-30

前言

不出意外,這應該是年前最後一次分享,本次來一點實際開發中會用到的小技巧。

比如平時大家是否都會寫類似這樣的程式碼:

if(a){
	//dosomething
}else if(b){
	//doshomething
}else if(c){
	//doshomething
} else{
	////doshomething
}
複製程式碼

條件少還好,一旦 else if 過多這裡的邏輯將會比較混亂,並很容易出錯。

比如這樣:

利用策略模式優化過多 if else 程式碼

摘自 cim 中的一個客戶端命令的判斷條件。

剛開始條件較少,也就沒管那麼多直接寫的;現在功能多了導致每次新增一個 else 條件我都得仔細核對,生怕影響之前的邏輯。

這次終於忍無可忍就把他重構了,重構之後這裡的結構如下:

利用策略模式優化過多 if else 程式碼

最後直接變為兩行程式碼,簡潔了許多。

而之前所有的實現邏輯都單獨抽取到其他實現類中。

利用策略模式優化過多 if else 程式碼
利用策略模式優化過多 if else 程式碼

這樣每當我需要新增一個 else 邏輯,只需要新增一個類實現同一個介面便可完成。每個處理邏輯都互相獨立互不干擾。

實現

利用策略模式優化過多 if else 程式碼

按照目前的實現畫了一個草圖。

整體思路如下:

  • 定義一個 InnerCommand 介面,其中有一個 process 函式交給具體的業務實現。
  • 根據自己的業務,會有多個類實現 InnerCommand 介面;這些實現類都會註冊到 Spring Bean 容器中供之後使用。
  • 通過客戶端輸入命令,從 Spring Bean 容器中獲取一個 InnerCommand 例項。
  • 執行最終的 process 函式。

主要想實現的目的就是不在有多個判斷條件,只需要根據當前客戶端的狀態動態的獲取 InnerCommand 例項。

從原始碼上來看最主要的就是 InnerCommandContext 類,他會根據當前客戶端命令動態獲取 InnerCommand 例項。

利用策略模式優化過多 if else 程式碼

  • 第一步是獲取所有的 InnerCommand 例項列表。
  • 根據客戶端輸入的命令從第一步的例項列表中獲取類型別。
  • 根據類型別從 Spring 容器中獲取具體例項物件。

因此首先第一步需要維護各個命令所對應的類型別。

利用策略模式優化過多 if else 程式碼

所以在之前的列舉中就維護了命令和類型別的關係,只需要知道命令就能知道他的類型別。

這樣才能滿足只需要兩行程式碼就能替換以前複雜的 if else,同時也能靈活擴充套件。

InnerCommand instance = innerCommandContext.getInstance(msg);
instance.process(msg) ;
複製程式碼

總結

當然還可以做的更靈活一些,比如都不需要顯式的維護命令和類型別的對應關係。

只需要在應用啟動時掃描所有實現了 InnerCommand 介面的類即可,在 cicada 中有類似實現,感興趣的可以自行檢視

這樣一些小技巧希望對你有所幫助。

以上所有原始碼可以在這裡檢視:

github.com/crossoverJi…

你的點贊與分享是對我最大的支援

利用策略模式優化過多 if else 程式碼

相關文章