作者:Jay_huaxiao來源:掘金 鏈接:https://juejin.im/post/5e5fa79de51d45271849e7bd
前言
代碼中如果if-else比較多,閱讀起來比較困難,維護起來也比較困難,很容易出bug,接下來,本文將介紹優(yōu)化if-else代碼的八種方案。
優(yōu)化方案一:提前return,去除不必要的else
如果if-else代碼塊包含return語句,可以考慮通過提前return,把多余else干掉,使代碼更加優(yōu)雅。
優(yōu)化前:
if(condition){
//doSomething
}else{
return ;
}
優(yōu)化后:
if(!condition){
return ;
}
//doSomething
優(yōu)化方案二:使用條件三目運算符
使用條件三目運算符可以簡化某些if-else,使代碼更加簡潔,更具有可讀性。
優(yōu)化前:
int price ;
if(condition){
price = 80;
}else{
price = 100;
}
優(yōu)化后:
int price = condition?80:100;
優(yōu)化方案三:使用枚舉
在某些時候,使用枚舉也可以優(yōu)化if-else邏輯分支,按個人理解,它也可以看做一種表驅(qū)動方法。
優(yōu)化前:
String OrderStatusDes;
if(orderStatus==0){
OrderStatusDes ="訂單未支付";
}else if(OrderStatus==1){
OrderStatusDes ="訂單已支付";
}else if(OrderStatus==2){
OrderStatusDes ="已發(fā)貨";
}
...
優(yōu)化后:
先定義一個枚舉
:
public enum OrderStatusEnum {
UN_PAID(0,"訂單未支付"),PAIDED(1,"訂單已支付"),SENDED(2,"已發(fā)貨"),;
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;
}
}
有了枚舉之后,以上if-else邏輯分支,可以優(yōu)化為一行代碼
String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc();
優(yōu)化方案四:合并條件表達式
如果有一系列條件返回一樣的結(jié)果,可以將它們合并為一個條件表達式,讓邏輯更加清晰。
優(yōu)化前
double getVipDiscount() {
if(age<18){
return 0.8;
}
if("深圳".equals(city)){
return 0.8;
}
if(isStudent){
return 0.8;
}
//do somethig
}
優(yōu)化后
double getVipDiscount(){
if(age<18|| "深圳".equals(city)||isStudent){
return 0.8;
}
//doSomthing
}
優(yōu)化方案五:使用 Optional
有時候if-else比較多,是因為非空判斷導致的,這時候你可以使用JAVA8的Optional進行優(yōu)化。
優(yōu)化前:
String str = "jay@huaxiao";
if (str != null) {
System.out.println(str);
} else {
System.out.println("Null");
}
優(yōu)化后:
Optional<String> strOptional = Optional.of("jay@huaxiao");
strOptional.ifPresentOrElse(System.out::println, () -> System.out.println("Null"));
優(yōu)化方案六:表驅(qū)動法
表驅(qū)動法,又稱之為表驅(qū)動、表驅(qū)動方法。表驅(qū)動方法是一種使你可以在表中查找信息,而不必用很多的邏輯語句(if或Case)來把它們找出來的方法。以下的demo,把map抽象成表,在map中查找信息,而省去不必要的邏輯語句。
優(yōu)化前:
if (param.equals(value1)) {
doAction1(someParams);
} else if (param.equals(value2)) {
doAction2(someParams);
} else if (param.equals(value3)) {
doAction3(someParams);
}
// ...
優(yōu)化后:
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);
優(yōu)化方案七:優(yōu)化邏輯結(jié)構(gòu),讓正常流程走主干
優(yōu)化前:
public double getAdjustedCapital(){
if(_capital <= 0.0 ){
return 0.0;
}
if(_intRate > 0 && _duration >0){
return (_income / _duration) *ADJ_FACTOR;
}
return 0.0;
}
優(yōu)化后:
public double getAdjustedCapital(){
if(_capital <= 0.0 ){
return 0.0;
}
if(_intRate <= 0 || _duration <= 0){
return 0.0;
}
return (_income / _duration) *ADJ_FACTOR;
}
將條件反轉(zhuǎn)使異常情況先退出,讓正常流程維持在主干流程,可以讓代碼結(jié)構(gòu)更加清晰。
優(yōu)化方案八:策略模式+工廠方法消除if else
假設需求為,根據(jù)不同勛章類型,處理相對應的勛章服務,優(yōu)化前有以下代碼:
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("展示守護勛章");
}
...
首先,我們把每個條件邏輯代碼塊,抽象成一個公共的接口,可以得出以下代碼:
//勛章接口
public interface IMedalService {
void showMedal();
}
我們根據(jù)每個邏輯條件,定義相對應的策略實現(xiàn)類,可得以下代碼:
//守護勛章策略實現(xiàn)類
public class GuardMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("展示守護勛章");
}
}
//嘉賓勛章策略實現(xiàn)類
public class GuestMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("嘉賓勛章");
}
}
//VIP勛章策略實現(xiàn)類
public class VipMedalServiceImpl implements IMedalService {
@Override
public void showMedal() {
System.out.println("會員勛章");
}
}
接下來,我們再定義策略工廠類,用來管理這些勛章實現(xiàn)策略類,如下:
//勛章服務工產(chǎn)類
public class MedalServicesFactory {
private static final Map<String, IMedalService> 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);
}
}
使用了策略+工廠模式之后,代碼變得簡潔多了,如下:
public class Test {
public static void main(String[] args) {
String medalType = "guest";
IMedalService medalService = MedalServicesFactory.getMedalService(medalType);
medalService.showMedal();
}
}