前言

隨著項(xiàng)目的迭代，代碼中存在的分支判斷可能會(huì)越來越多，當(dāng)里面涉及到的邏輯比較復(fù)雜或者分支數(shù)量實(shí)在是多的難以維護(hù)的時(shí)候，我們就要考慮下，有辦法能讓這些代碼變得更優(yōu)雅嗎？

創(chuàng)新互聯(lián)長(zhǎng)期為1000多家客戶提供的網(wǎng)站建設(shè)服務(wù)，團(tuán)隊(duì)從業(yè)經(jīng)驗(yàn)10年，關(guān)注不同地域、不同群體，并針對(duì)不同對(duì)象提供差異化的產(chǎn)品和服務(wù)；打造開放共贏平臺(tái)，與合作伙伴共同營(yíng)造健康的互聯(lián)網(wǎng)生態(tài)環(huán)境。為郊區(qū)企業(yè)提供專業(yè)的做網(wǎng)站、網(wǎng)站建設(shè)，郊區(qū)網(wǎng)站改版等技術(shù)服務(wù)。擁有10多年豐富建站經(jīng)驗(yàn)和眾多成功案例,為您定制開發(fā)。

正文

使用枚舉

這里我們簡(jiǎn)單的定義一個(gè)表示狀態(tài)的枚舉。

public enum Status {
    NEW(0),RUNNABLE(1),RUNNING(2),BLOCKED(3),DEAD(4);

    public int statusCode;

    Status(int statusCode){
        this.statusCode = statusCode;
    }
}

那么我們?cè)谑褂玫臅r(shí)候就可以直接通過枚舉調(diào)用了。

int statusCode = Status.valueOf("NEW").statusCode;

優(yōu)雅的解決了下面代碼賦值的方式

if(param.equals("NEW")){
    statusCode = 0;
}else if(param.equals("RUNNABLE")){
    statusCode = 1;
}
...

善用 Optional

在項(xiàng)目中，總少不了一些非空的判斷,可能大部分人還是如下的用法

if(null == user){
    //action1
}else{
    //action2
}

這時(shí)候該掏出Optional這個(gè)秘密武器了，它可以讓非空校驗(yàn)更加優(yōu)雅，間接的減少if操作。沒了解過Optional的同學(xué)可自行Google，這里就不再贅述。

Optional userOptional = Optional.ofNullable(user);
userOptional.map(action1).orElse(action2);

上面的代碼跟第一段是等效的，通過一些新特性讓代碼更加緊湊。

表驅(qū)動(dòng)法

來自Google的解釋：表驅(qū)動(dòng)法是一種編程模式，它的本質(zhì)是，從表里查詢信息來代替邏輯語句(if,case)。下面看一個(gè)案例，通過月份來獲取當(dāng)月的天數(shù)（僅作為案例演示，獲取2月份的數(shù)據(jù)不嚴(yán)謹(jǐn)），普通做法：

int getMonthDays(int month){
    switch(month){
        case 1:return 31;break;
        case 2:return 29;break;
        case 3:return 31;break;
        case 4:return 30;break;
        case 5:return 31;break;
        case 6:return 30;break;
        case 7:return 31;break;
        case 8:return 31;break;
        case 9:return 30;break;
        case 10:return 31;break;
        case 11:return 30;break;
        case 12:return 31;break;
        default：return 0;
    }
}

表驅(qū)動(dòng)法實(shí)現(xiàn)方式

int monthDays[12] = {31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31};
int getMonthDays(int month){
    return monthDays[--month];
}

其實(shí)這里的表就是數(shù)組而已，通過直接查詢數(shù)組來獲得需要的數(shù)據(jù)，那么同理，Map之類的容器也可以成為我們編程概念中的表。

Map action> actionsMap = new HashMap<>();

// 初試配置對(duì)應(yīng)動(dòng)作
actionsMap.put(value1, (someParams) -> { doAction1(someParams)});
actionsMap.put(value2, (someParams) -> { doAction2(someParams)});
actionsMap.put(value3, (someParams) -> { doAction3(someParams)});

// 省略 null 判斷
actionsMap.get(param).apply(someParams);

通過Java8的lambda表達(dá)式，我們把需要執(zhí)行東西存進(jìn)value中，調(diào)用的時(shí)候通過匹配key的方式進(jìn)行。

提前判斷返回

在之前的文章《優(yōu)化代碼中的“壞味道”》里也有提過，如下語句

if(condition){
   //dost
}else{
   return ;
}

改為

if(!condition){
   return ;
}
//dost

避免一些不必要的分支，讓代碼更精煉。

其他方法

除了上面提到的方法，我們還可以通過一些設(shè)計(jì)模式，例如策略模式，責(zé)任鏈模式等來優(yōu)化存在大量if,case的情況，其原理會(huì)和表驅(qū)動(dòng)的模式比較相似，大家可以自己動(dòng)手實(shí)現(xiàn)一下，例如我們?cè)贜etty的使用過程中，可能會(huì)出現(xiàn)需要大量判斷不同的命令去執(zhí)行對(duì)應(yīng)動(dòng)作的場(chǎng)景。

ServerHandler.java

if(command.equals("login")){
    //執(zhí)行登錄
}else if(command.equals("chat")){
    //聊天
}else if(command.equals("broadcast")){
    //廣播信息
}
....

該如何處理呢？這里先賣個(gè)關(guān)子，大家可以先思考一下，筆記后續(xù)會(huì)寫一些關(guān)于Netty實(shí)現(xiàn)IM的文章，到時(shí)候會(huì)詳細(xì)介紹。

結(jié)語

最后要明確一點(diǎn)，不是所有的if/else,switch/case都需要優(yōu)化，當(dāng)我們發(fā)現(xiàn)有“痛點(diǎn)”或者“聞到代碼有壞味道”再來優(yōu)化才是最好的，不然你可能會(huì)寫了一個(gè)從不擴(kuò)展的可擴(kuò)展代碼，所有的優(yōu)化都是為了更好的迭代項(xiàng)目，更好的服務(wù)于業(yè)務(wù)，而不是為了優(yōu)化而優(yōu)化。

公眾號(hào)博文同步Github倉(cāng)庫(kù)，有興趣的朋友可以幫忙給個(gè)Star哦，碼字不易，感謝支持。

https://github.com/PeppaLittlePig/blog-wechat