代碼中如果 if-else 比較多,閱讀起來比較困難,維護起來也比較困難,很容易出 Bug,接下來,本文將介紹優化 if-else 代碼的八種方案。
圖片來自 Pexels
優化方案 1:提前 return,去除不必要的 else
如果 if-else 代碼塊包含 return 語句,可以考慮通過提前 return,把多餘 else 幹掉,使代碼更加優雅。
優化前:
<code>if(condition){
//doSomething
}else{
return ;
} /<code>
優化後:
<code>if(!condition){
return ;
}
//doSomething /<code>
優化方案 2:使用條件三目運算符
使用條件三目運算符可以簡化某些 if-else,使代碼更加簡潔,更具有可讀性。
優化前:
<code>int price ;
if(condition){
price = 80;
}else{
price = 100;
} /<code>
優化後:
<code>int price = condition?80:100; /<code>
優化方案 3:使用枚舉
在某些時候,使用枚舉也可以優化 if-else 邏輯分支,按個人理解,它也可以看作一種表驅動方法。
優化前:
<code>String OrderStatusDes;
if(orderStatus==0){
OrderStatusDes ="訂單未支付";
}else if(OrderStatus==1){
OrderStatusDes ="訂單已支付";
}else if(OrderStatus==2){
OrderStatusDes ="已發貨";
}
... /<code>
優化後:(先定義一個枚舉)
<code>public enum OrderStatusEnum {
UN_PAID(0,"訂單未支付"),PAIDED(1,"訂單已支付"),SENDED(2,"已發貨"),;
private int index;
private String desc;
public int getIndex() {
return index;
}
public String getDesc() {
return desc;
}
OrderStatusEnum(int index, String desc){
this.index = index;
this.desc =desc;
}
OrderStatusEnum of(int orderStatus) {
for (OrderStatusEnum temp : OrderStatusEnum.values()) {
if (temp.getIndex() == orderStatus) {
return temp;
}
}
return null;
}
} /<code>
有了枚舉之後,以上 if-else 邏輯分支,可以優化為一行代碼:
<code>String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc(); /<code>
優化方案 4:合併條件表達式
如果有一系列條件返回一樣的結果,可以將它們合併為一個條件表達式,讓邏輯更加清晰。
優化前:
<code>double getVipDiscount() {
if(age<18){
return 0.8;
}
if("深圳".equals(city)){
return 0.8;
}
if(isStudent){
return 0.8;
}
//do somethig
} /<code>
優化後:
<code>double getVipDiscount(){
if(age<18|| "深圳".equals(city)||isStudent){
return 0.8;
}
//doSomthing
} /<code>
優化方案 5:使用 Optional
有時候 if-else 比較多,是因為非空判斷導致的,這時候你可以使用 java8 的 Optional 進行優化。
優化前:
<code>String str = "jay@huaxiao";
if (str != null) {
System.out.println(str);
} else {
System.out.println("Null");
} /<code>
優化後:
<code>Optional<string> strOptional = Optional.of("jay@huaxiao");
strOptional.ifPresentOrElse(System.out::println, () -> System.out.println("Null")); /<string>/<code>
優化方案 6:表驅動法
表驅動法,又稱之為表驅動、表驅動方法。表驅動方法是一種使你可以在表中查找信息,而不必用很多的邏輯語句(if 或 case)來把它們找出來的方法。
以下的 demo,把 map 抽象成表,在 map 中查找信息,而省去不必要的邏輯語句。
優化前:
<code>if (param.equals(value1)) {
doAction1(someParams);
} else if (param.equals(value2)) {
doAction2(someParams);
} else if (param.equals(value3)) {
doAction3(someParams);
}
// ... /<code>
優化後:
<code>Map, Function> action> actionMappings = new HashMap<>(); // 這裡泛型 ? 是為方便演示,實際可替換為你需要的類型
// 初始化
actionMappings.put(value1, (someParams) -> { doAction1(someParams)});
actionMappings.put(value2, (someParams) -> { doAction2(someParams)});
actionMappings.put(value3, (someParams) -> { doAction3(someParams)});
// 省略多餘邏輯語句
actionMappings.get(param).apply(someParams); /<code>
優化方案 7:優化邏輯結構,讓正常流程走主幹
優化前:
<code>public double getAdjustedCapital(){
if(_capital <= 0.0 ){
return 0.0;
}
if(_intRate > 0 && _duration >0){
return (_income / _duration) *ADJ_FACTOR;
}
return 0.0;
} /<code>
優化後:
<code>public double getAdjustedCapital(){
if(_capital <= 0.0 ){
return 0.0;
}
if(_intRate <= 0 || _duration <= 0){
return 0.0;
}
return (_income / _duration) *ADJ_FACTOR;
} public double getAdjustedCapital(){
if(_capital <= 0.0 ){
return 0.0;
}
if(_intRate <= 0 || _duration <= 0){
return 0.0;
}
return (_income / _duration) *ADJ_FACTOR;
} /<code>
將條件反轉使異常情況先退出,讓正常流程維持在主幹流程,可以讓代碼結構更加清晰。
優化方案 8:策略模式+工廠方法消除 if else
假設需求為,根據不同勳章類型,處理相對應的勳章服務,優化前有以下代碼:
<code>String medalType = "guest";
if ("guest".equals(medalType)) {
System.out.println("嘉賓勳章");
} else if ("vip".equals(medalType)) {
System.out.println("會員勳章");
} else if ("guard".equals(medalType)) {
System.out.println("展示守護勳章");
}
... /<code>
首先,我們把每個條件邏輯代碼塊,抽象成一個公共的接口,可以得到以下代碼:
<code>//勳章接口
public interface IMedalService {
void showMedal();
} /<code>
我們根據每個邏輯條件,定義相對應的策略實現類,可得以下代碼:
<code>//守護勳章策略實現類
public class GuardMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("展示守護勳章");
}
}
//嘉賓勳章策略實現類
public class GuestMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("嘉賓勳章");
}
}
//VIP勳章策略實現類
public class VipMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("會員勳章");
}
} /<code>
接下來,我們再定義策略工廠類,用來管理這些勳章實現策略類,如下:
<code>//勳章服務工產類
public class MedalServicesFactory {
private static final Map<string> map = new HashMap<>();
static {
map.put("guard", new GuardMedalServiceImpl());
map.put("vip", new VipMedalServiceImpl());
map.put("guest", new GuestMedalServiceImpl());
}
public static IMedalService getMedalService(String medalType) {
return map.get(medalType);
}
} /<string>/<code>
使用了策略+工廠模式之後,代碼變得簡潔多了,如下:
<code>public class Test {
public static void main(String[] args) {
String medalType = "guest";
IMedalService medalService = MedalServicesFactory.getMedalService(medalType);
medalService.showMedal();
}
} /<code>
參考與感謝:
- 6 個實例詳解如何把 if-else 代碼重構成高質量代碼
- 如何 “幹掉” if...else
閱讀更多 51CTO 的文章