怎么写出更好的代码-优化if-else

代码中如果 if-else 比较多，阅读起来比较困难，维护起来也比较困难，很容易出 Bug。

我们要怎么优化呢？接下来，本文将介绍优化 if-else 代码的几种方案：

1：提前 return，去除不必要的 else

如果 if-else 代码块包含 return 语句，可以考虑通过提前 return，把多余 else 干掉，使代码更加优雅。

优化前：

if(condition){ 
  doSomething()
}else{    
  return ;
}

优化后：

if(!condition){    
  return ;
}
doSomething()

2：使用条件三目运算符

使用条件三目运算符可以简化某些 if-else，使代码更加简洁，更具有可读性。

三目运算：对于条件表达式b ? x : y，先计算条件b，然后进行判断。如果b的值为true，计算x的值，运算结果为x的值；否则，计算y的值，运算结果为y的值。

一个条件表达式绝不会既计算x，又计算y。

条件运算符是右结合的，也就是说，从右向左分组计算。

例如，a ? b : c ? d : e将按a ? b : (c ? d : e)执行。

优化前：

int  price ;
if(condition){    
  price = 80;
}else{    
  price = 100;
}

优化后：

int price = condition?80:100;

3：使用枚举

在某些时候，使用枚举也可以优化 if-else 逻辑分支，按个人理解，它也可以看作一种表驱动方法。

优化前：

String OrderStatusDes;
if(orderStatus==0){    
  OrderStatusDes ="订单未支付";
}else if(OrderStatus==1){    
  OrderStatusDes ="订单已支付";
}else if(OrderStatus==2){   
  OrderStatusDes ="已发货"; 
}
...

优化后：（先定义一个枚举）

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;
    }
}

有了枚举之后，以上 if-else 逻辑分支，可以优化为一行代码：

String OrderStatusDes = OrderStatusEnum.0f(orderStatus).getDesc();

4：合并条件表达式

如果有一系列条件返回一样的结果，可以将它们合并为一个条件表达式，让逻辑更加清晰。

优化前：

getVipDiscount() {
        if(age<18){
            return 0.8;
        }
        if("深圳".equals(city)){
            return 0.8;
        }
        if(isStudent){
            return 0.8;
        }
        //do somethig
    }

优化后：

 getVipDiscount(){
        if(age<18|| "深圳".equals(city)||isStudent){
            return 0.8;
        }
        //doSomthing
    }

5：优化逻辑结构，让正常流程走主干

优化前：

public function getAdjustedCapital(){
    if(_capital <= 0.0 ){
        return 0.0;
    }
    if(_intRate > 0 && _duration >0){
        return (_income / _duration) *ADJ_FACTOR;
    }
    return 0.0;
}

优化后：

public function getAdjustedCapital(){
    if(_capital <= 0.0 ){
        return 0.0;
    }
    if(_intRate <= 0 || _duration <= 0){
        return 0.0;
    }

    return (_income / _duration) *ADJ_FACTOR;
}

将条件反转使异常情况先退出，让正常流程维持在主干流程，可以让代码结构更加清晰。

6：策略模式+工厂方法消除 if else

假设需求为，根据不同勋章类型，处理相对应的勋章服务，优化前有以下代码：

 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();
}

我们根据每个逻辑条件，定义相对应的策略实现类，可得以下代码：

//守护勋章策略实现类
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("会员勋章");
    }
}

接下来，我们再定义策略工厂类，用来管理这些勋章实现策略类，如下：

//勋章服务工产类
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();
    }
}