|
|
51CTO旗下网站
|
|
移动端

彻底消灭if-else,你需要这8种方案!

代码中如果 if-else 比较多,阅读起来比较困难,维护起来也比较困难,很容易出 Bug,接下来,本文将介绍优化 if-else 代码的八种方案。

作者:捡田螺的小男孩来源:捡田螺的小男孩|2020-03-11 08:00

代码中如果 if-else 比较多,阅读起来比较困难,维护起来也比较困难,很容易出 Bug,接下来,本文将介绍优化 if-else 代码的八种方案。

图片来自 Pexels

优化方案 1:提前 return,去除不必要的 else

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

优化前:

  1. if(condition){ 
  2.     //doSomething 
  3. }else
  4.     return ; 

优化后:

  1. if(!condition){ 
  2.     return ; 
  3. //doSomething 

优化方案 2:使用条件三目运算符

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

优化前:

  1. int  price ; 
  2. if(condition){ 
  3.     price = 80; 
  4. }else
  5.     price = 100; 

优化后:

  1. int price = condition?80:100; 

优化方案 3:使用枚举

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

优化前:

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

优化后:(先定义一个枚举)

  1. public enum OrderStatusEnum { 
  2.     UN_PAID(0,"订单未支付"),PAIDED(1,"订单已支付"),SENDED(2,"已发货"),; 
  3.  
  4.     private int index
  5.     private String desc
  6.  
  7.     public int getIndex() { 
  8.         return index
  9.     } 
  10.  
  11.     public String getDesc() { 
  12.         return desc
  13.     } 
  14.  
  15.     OrderStatusEnum(int index, String desc){ 
  16.         this.index = index
  17.         this.desc =desc
  18.     } 
  19.  
  20.     OrderStatusEnum of(int orderStatus) { 
  21.         for (OrderStatusEnum temp : OrderStatusEnum.values()) { 
  22.             if (temp.getIndex() == orderStatus) { 
  23.                 return temp
  24.             } 
  25.         } 
  26.         return null
  27.     } 

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

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

优化方案 4:合并条件表达式

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

优化前:

  1. double getVipDiscount() { 
  2.         if(age<18){ 
  3.             return 0.8; 
  4.         } 
  5.         if("深圳".equals(city)){ 
  6.             return 0.8; 
  7.         } 
  8.         if(isStudent){ 
  9.             return 0.8; 
  10.         } 
  11.         //do somethig 
  12.     } 

优化后:

  1. double getVipDiscount(){ 
  2.         if(age<18|| "深圳".equals(city)||isStudent){ 
  3.             return 0.8; 
  4.         } 
  5.         //doSomthing 
  6.     } 

优化方案 5:使用 Optional

有时候 if-else 比较多,是因为非空判断导致的,这时候你可以使用 java8 的 Optional 进行优化。

优化前:

  1. String str = "jay@huaxiao"
  2. if (str != null) { 
  3.     System.out.println(str); 
  4. else { 
  5.     System.out.println("Null"); 

优化后:

  1. Optional<String> strOptional = Optional.of("jay@huaxiao"); 
  2. strOptional.ifPresentOrElse(System.out::println, () -> System.out.println("Null")); 

优化方案 6:表驱动法

表驱动法,又称之为表驱动、表驱动方法。表驱动方法是一种使你可以在表中查找信息,而不必用很多的逻辑语句(if 或 case)来把它们找出来的方法。

以下的 demo,把 map 抽象成表,在 map 中查找信息,而省去不必要的逻辑语句。

优化前:

  1. if (param.equals(value1)) { 
  2.     doAction1(someParams); 
  3. else if (param.equals(value2)) { 
  4.     doAction2(someParams); 
  5. else if (param.equals(value3)) { 
  6.     doAction3(someParams); 
  7. // ... 

优化后:

  1. Map<?, Function<?> action> actionMappings = new HashMap<>(); // 这里泛型 ? 是为方便演示,实际可替换为你需要的类型 
  2.  
  3. // 初始化 
  4. actionMappings.put(value1, (someParams) -> { doAction1(someParams)}); 
  5. actionMappings.put(value2, (someParams) -> { doAction2(someParams)}); 
  6. actionMappings.put(value3, (someParams) -> { doAction3(someParams)}); 
  7.  
  8. // 省略多余逻辑语句 
  9. actionMappings.get(param).apply(someParams); 

优化方案 7:优化逻辑结构,让正常流程走主干

优化前:

  1. public double getAdjustedCapital(){ 
  2.     if(_capital <= 0.0 ){ 
  3.         return 0.0; 
  4.     } 
  5.     if(_intRate > 0 && _duration >0){ 
  6.         return (_income / _duration) *ADJ_FACTOR; 
  7.     } 
  8.     return 0.0; 

优化后:

  1. public double getAdjustedCapital(){ 
  2.     if(_capital <= 0.0 ){ 
  3.         return 0.0; 
  4.     } 
  5.     if(_intRate <= 0 || _duration <= 0){ 
  6.         return 0.0; 
  7.     } 
  8.  
  9.     return (_income / _duration) *ADJ_FACTOR; 

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

优化方案 8:策略模式+工厂方法消除 if else

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

  1. String medalType = "guest"
  2. if ("guest".equals(medalType)) { 
  3.     System.out.println("嘉宾勋章"); 
  4.  } else if ("vip".equals(medalType)) { 
  5.     System.out.println("会员勋章"); 
  6. else if ("guard".equals(medalType)) { 
  7.     System.out.println("展示守护勋章"); 
  8. ... 

首先,我们把每个条件逻辑代码块,抽象成一个公共的接口,可以得到以下代码:

  1. //勋章接口 
  2. public interface IMedalService { 
  3.     void showMedal(); 

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

  1. //守护勋章策略实现类 
  2. public class GuardMedalServiceImpl implements IMedalService { 
  3.     @Override 
  4.     public void showMedal() { 
  5.         System.out.println("展示守护勋章"); 
  6.     } 
  7. //嘉宾勋章策略实现类 
  8. public class GuestMedalServiceImpl implements IMedalService { 
  9.     @Override 
  10.     public void showMedal() { 
  11.         System.out.println("嘉宾勋章"); 
  12.     } 
  13. //VIP勋章策略实现类 
  14. public class VipMedalServiceImpl implements IMedalService { 
  15.     @Override 
  16.     public void showMedal() { 
  17.         System.out.println("会员勋章"); 
  18.     } 

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

  1. //勋章服务工产类 
  2. public class MedalServicesFactory { 
  3.  
  4.     private static final Map<String, IMedalService> map = new HashMap<>(); 
  5.     static { 
  6.         map.put("guard", new GuardMedalServiceImpl()); 
  7.         map.put("vip", new VipMedalServiceImpl()); 
  8.         map.put("guest", new GuestMedalServiceImpl()); 
  9.     } 
  10.     public static IMedalService getMedalService(String medalType) { 
  11.         return map.get(medalType); 
  12.     } 

使用了策略+工厂模式之后,代码变得简洁多了,如下:

  1. public class Test { 
  2.     public static void main(String[] args) { 
  3.         String medalType = "guest"
  4.         IMedalService medalService = MedalServicesFactory.getMedalService(medalType); 
  5.         medalService.showMedal(); 
  6.     } 

参考与感谢:

  • 6 个实例详解如何把 if-else 代码重构成高质量代码
  • 如何 “干掉” if...else

【编辑推荐】

  1. 程序员如何优化自己的职业程序?
  2. Spring Boot项目优化和JVM调优
  3. 这35个Java代码优化细节,你用了吗?
  4. 几个 JavaScript 性能优化小 Tip
  5. 前端性能优化到底在优化什么?怎么优化
【责任编辑:武晓燕 TEL:(010)68476606】

点赞 0
分享:
大家都在看
猜你喜欢

订阅专栏+更多

网络排障一点通

网络排障一点通

网络排障及优化调整案例
共20章 | 捷哥CCIE

279人订阅学习

VMware NSX 入门到实战

VMware NSX 入门到实战

网络虚拟化革命性技术
共16章 | Cloud袁

208人订阅学习

消息队列Kafka运维实践攻略

消息队列Kafka运维实践攻略

入门级消息队列
共3章 | 独行侠梦

111人订阅学习

订阅51CTO邮刊

点击这里查看样刊

订阅51CTO邮刊

51CTO服务号

51CTO官微