Map+函數(shù)式接口如何完美的解決if-else問題？

需求

最近寫了一個服務(wù)：根據(jù)優(yōu)惠券的類型resourceType和編碼resourceId來 查詢 發(fā)放方式grantType和領(lǐng)取規(guī)則

實現(xiàn)方式：

根據(jù)優(yōu)惠券類型resourceType -> 確定查詢哪個數(shù)據(jù)表

根據(jù)編碼resourceId -> 到對應(yīng)的數(shù)據(jù)表里邊查詢優(yōu)惠券的派發(fā)方式grantType和領(lǐng)取規(guī)則

優(yōu)惠券有多種類型，分別對應(yīng)了不同的數(shù)據(jù)庫表：

紅包 —— 紅包發(fā)放規(guī)則表

購物券 —— 購物券表

QQ會員

外賣會員

實際的優(yōu)惠券遠不止這些，這個需求是要我們寫一個業(yè)務(wù)分派的邏輯

第一個能想到的思路就是if-else或者switch case：

switch(resourceType){
case"紅包":
查詢紅包的派發(fā)方式
break;
case"購物券":
查詢購物券的派發(fā)方式
break;
case"QQ會員":
break;
case"外賣會員":
break;
......
default:logger.info("查找不到該優(yōu)惠券類型resourceType以及對應(yīng)的派發(fā)方式");
break;
}

如果要這么寫的話， 一個方法的代碼可就太長了，影響了可讀性。（別看著上面case里面只有一句話，但實際情況是有很多行的）

而且由于 整個 if-else的代碼有很多行，也不方便修改，可維護性低。

策略模式

策略模式是把 if語句里面的邏輯抽出來寫成一個類，如果要修改某個邏輯的話，僅修改一個具體的實現(xiàn)類的邏輯即可，可維護性會好不少。

以下是策略模式的具體結(jié)構(gòu)

策略模式在業(yè)務(wù)邏輯分派的時候還是if-else，只是說比第一種思路的if-else 更好維護一點。

switch(resourceType){
case"紅包":
StringgrantType=newContext(newRedPaper()).ContextInterface();
break;
case"購物券":
StringgrantType=newContext(newShopping()).ContextInterface();
break;

......
default:logger.info("查找不到該優(yōu)惠券類型resourceType以及對應(yīng)的派發(fā)方式");
break;

但缺點也明顯：

如果 if-else的判斷情況很多，那么對應(yīng)的具體策略實現(xiàn)類也會很多，上邊的具體的策略實現(xiàn)類還只是2個，查詢紅包發(fā)放方式寫在類RedPaper里邊，購物券寫在另一個類Shopping里邊；那資源類型多個QQ會員和外賣會員，不就得再多寫兩個類？有點麻煩了

沒法俯視整個分派的業(yè)務(wù)邏輯

Map+函數(shù)式接口

用上了Java8的新特性lambda表達式

判斷條件放在key中

對應(yīng)的業(yè)務(wù)邏輯放在value中

這樣子寫的好處是非常直觀，能直接看到判斷條件對應(yīng)的業(yè)務(wù)邏輯

需求：根據(jù)優(yōu)惠券(資源)類型resourceType和編碼resourceId查詢派發(fā)方式grantType

上代碼：

@Service
publicclassQueryGrantTypeService{

@Autowired
privateGrantTypeSerivegrantTypeSerive;
privateMap>grantTypeMap=newHashMap<>();

/**
*初始化業(yè)務(wù)分派邏輯,代替了if-else部分
*key:優(yōu)惠券類型
*value:lambda表達式,最終會獲得該優(yōu)惠券的發(fā)放方式
*/
@PostConstruct
publicvoiddispatcherInit(){
grantTypeMap.put("紅包",resourceId->grantTypeSerive.redPaper(resourceId));
grantTypeMap.put("購物券",resourceId->grantTypeSerive.shopping(resourceId));
grantTypeMap.put("qq會員",resourceId->grantTypeSerive.QQVip(resourceId));
}

publicStringgetResult(StringresourceType){
//Controller根據(jù)優(yōu)惠券類型resourceType、編碼resourceId去查詢發(fā)放方式grantType
Functionresult=getGrantTypeMap.get(resourceType);
if(result!=null){
//傳入resourceId執(zhí)行這段表達式獲得String型的grantType
returnresult.apply(resourceId);
}
return"查詢不到該優(yōu)惠券的發(fā)放方式";
}
}

如果單個 if 語句塊的業(yè)務(wù)邏輯有很多行的話，我們可以把這些 業(yè)務(wù)操作抽出來，寫成一個單獨的Service，即：

//具體的邏輯操作

@Service
publicclassGrantTypeSerive{

publicStringredPaper(StringresourceId){
//紅包的發(fā)放方式
return"每周末9點發(fā)放";
}
publicStringshopping(StringresourceId){
//購物券的發(fā)放方式
return"每周三9點發(fā)放";
}
publicStringQQVip(StringresourceId){
//qq會員的發(fā)放方式
return"每周一0點開始秒殺";
}
}

入?yún)tring resourceId是用來查數(shù)據(jù)庫的，這里簡化了，傳參之后不做處理。

用http調(diào)用的結(jié)果：

@RestController
publicclassGrantTypeController{

@Autowired
privateQueryGrantTypeServicequeryGrantTypeService;

@PostMapping("/grantType")
publicStringtest(StringresourceName){
returnqueryGrantTypeService.getResult(resourceName);
}
}

用Map+函數(shù)式接口也有弊端：

你的隊友得會lambda表達式才行啊，他不會讓他自己百度去

最后捋一捋本文講了什么

策略模式通過接口、實現(xiàn)類、邏輯分派來完成，把 if語句塊的邏輯抽出來寫成一個類，更好維護。

Map+函數(shù)式接口通過Map.get(key)來代替 if-else的業(yè)務(wù)分派，能夠避免策略模式帶來的類增多、難以俯視整個業(yè)務(wù)邏輯的問題。

審核編輯：劉清