深圳幻海软件技术有限公司 欢迎您!

你可以这样优化if-else代码结构

2023-02-27

 前言最近部门在对以往的代码做一些优化,我在代码中看到一连串的if(){}elseif(){}的逻辑判断。这明显是有优化空间的。由于内部代码不适合分享,这里我就用<输出今天为星期几>来讲讲逻辑判断优化的一些方案。这里先声明,免有人疑惑:我们在项目中使用的很可能会有多层的嵌套,不

 

前言

最近部门在对以往的代码做一些优化,我在代码中看到一连串的 if(){}elseif(){} 的逻辑判断。这明显是有优化空间的。

由于内部代码不适合分享,这里我就用 <输出今天为星期几> 来讲讲逻辑判断优化的一些方案。

这里先声明,免有人疑惑:

我们在项目中使用的很可能会有多层的嵌套,不像我的例子只有一层。不过其优化的思想是大致相同的。由于 returnWeekday() 方法异常之简单,所以提前声明。

需求

写一个 returnWeekday() 方法返回"今天是星期*"。

接下来我们就一步一步来。

优化过程

这里我简单的分为 “初学 —> 入门 —> 中级” 这几个阶段。

初学

当我们开始拿到需求的时候,看到一系列的逻辑判断,首先想到的应该就是 if 语句了。

代码如下: 

function returnWeekday(){  
      let string = "今天是星期" 
      let date = new Date().getDay();  
      if (date === 0) {  
          string += "日";  
      } else if (date === 1) {  
          string += "一";  
      } else if (date === 2) { 
          string += "二";  
      } else if (date === 3) {  
          string += "三";  
      } else if (date === 4) {  
          string += "四";  
      } else if (date === 5) {  
          string += "五";  
      } else if (date === 6) {  
          string += "六";  
      }  
      return string  
  }  
  console.log(returnWeekday()) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.

当我们写完了这样的代码,第一感觉就是 elseif 块是不是太多了。

可是当我们还在思考如何优化的时候,产品就在钉钉发消息给你问这个需求完成的怎么样了?还带了一个坏笑的表情包。这个时候你告诉自己,这个优化后面再说吧。可是久而久之,需求加之。这个再也没有优化过了,直到下一个接手。

上面的代码确实 elseif 块太多了,看着就不舒服。

我们在看《JavaScript 高级程序设计》的时候,看到这样一句话:

switch 语句与 if 语句的关系最为密切,而且也是在其他语言中普遍使用的一种流控制语句。

所以我们是不是可以考虑使用 switch 语句来优化一下呢?

入门

这里我们使用 switch 语句优化一遍代码。

注意:switch 语句在比较值的时候使用的是全等操作符,不会有类型转换的情况。

代码如下:   

function returnWeekday(){  
       let string = "今天是星期" 
       let date = new Date().getDay();  
       switch (date) { 
           case 0 :  
               string += "日";  
               break;  
           case 1 : 
               string += "一";  
               break;  
           case 2 :  
               string += "二";  
               break;  
           case 3 :  
               string += "三";  
               break;  
           case 4 :  
               string += "四";  
               break;  
           case 5 :  
               string += "五";  
               break;  
           case 6 :  
               string += "六";  
               break;  
       }  
       return string  
   }  
   console.log(returnWeekday()) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.
  • 25.
  • 26.
  • 27.
  • 28.
  • 29.

我们在 case 里面拼接字符,以达到输出预期结果的目的。这里的结构看起来确实比 if 语句清晰了一点。可是还是有点疑惑?

假设哪一天,相关组织发现,星象有变。每周需要变成八天(产品的思维,你无法想象)。我们这个 returnWeekday() 方法就需要多加一层判断了。

我们的希望是已经封装好的方法,不要频繁的修改。可是需求的变动是你无法控制的。

所以我们继续思考该怎么优化。

中级

我们看到这里的 case 是数字,和数组的下标是一致的。

即:['天','一','二','三','四','五','六'] 的下标。

所以我们可以考虑使用数组来优化。

代码如下:  

function returnWeekday(){  
       let string = "今天是星期" 
       let date = new Date().getDay();  
       // 使用数组  
       let dateArr = ['天','一','二','三','四','五','六'];  
       return string + dateArr[date]  
   }  
   console.log(returnWeekday()) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.

以上代码是不是比 switch 语句和 if 语句清晰多了。而且就算一周变为八天,只需要修改 dateArr 数组就好了。

倘若我们的每个 case 是不规律的字符串呢?那我们使用对象,每个 case 为一个 key

代码如下:

function returnWeekday(){    
     let string = "今天是星期" 
     let date = new Date().getDay();  
     // 使用对象  
     dateObj = {   
         0: '天',  
         1: "一",   
         2: "二",   
         3: "三",   
         4: "四",   
         5: "五",   
         6: "六",   
     };  
     return string + dateObj[date]  
 }  
 console.log(returnWeekday()) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.

以上使用数组或者对象的写法,提高代码的可读性的同时,维护起来也变得简单了。

使用 charAt 字符方法

字符串有个和使用数组下标类似的方法:   

// charAt 定位方法  
    function returnWeekday(){  
        return "今天是星期" + "日一二三四五六".charAt(new Date().getDay());  
    }  
    console.log(returnWeekday()) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.

需求变动

这个时候,有人希望 returnWeekday() 方法不仅返回今天是周几,还需要区分工作日和休息日,调用相关方法。

如果是使用 switch、 if 或 数组 维护起来就有点麻烦,需要改写的地方还挺多的。  

function returnWeekday(){  
       let string = "今天是星期" 
       let date = new Date().getDay();  
       // 使用对象  
       dateObj = {   
           0: ['天','休'],   
           1: ["一",'工'],   
           2: ["二",'工'],   
           3: ["三",'工'],   
           4: ["四",'工'],   
           5: ["五",'工'],   
           6: ["六",'休'],   
       }  
       // 类型,这里也可以对应相关方法  
       dayType = {  
           '休': function(){  
               // some code  
               console.log('为休息日')  
           },  
           '工': function(){  
               // some code  
               console.log('为工作日')  
           }  
       }  
       let returnData = {  
           'string' : string + dateObj[date][0],  
           'method' : dayType[dateObj[date][1]]  
       }  
       return returnData  
   };  
   console.log(returnWeekday().method.call(this)) 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.
  • 25.
  • 26.
  • 27.
  • 28.
  • 29.
  • 30.
  • 31.

其他常见优化手段

这里在给出一些常见的优化手段。

三元运算

适合简单的 if(){}else{} 情况。   

//滚动监听,头部固定  
    handleScroll: function () {  
        let offsetTop = this.$refs.pride_tab_fixed.getBoundingClientRect().top;  
        if( offsetTop < 0 ){  
            this.titleFixed = true  
        } else {  
            this.titleFixed = false  
        }  
        // 改成三元  
        (offsetTop < 0) ? this.titleFixed = true : this.titleFixed = false 
        // 我们发现条件块里面的赋值情况是布尔值,所以可以更简单 
        this.titleFixed = offsetTop < 0 
    } 
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.

这样在精简代码的时候,可读性也没有降低。

逻辑与运算符

有些时候我们可以使用逻辑与运算符来简化代码   

if( falg ){  
       someMethod()  
   } 
  • 1.
  • 2.
  • 3.

可以改成:

falg && someMethod(); 
  • 1.

使用 includes 处理多重条件   

if( code === '202' || code === '203' || code === '204' ){ 
       someMethod()  
   } 
  • 1.
  • 2.
  • 3.

可以改成 

if( ['202','203','204'].includes(code) ){  
      someMethod()  
  } 
  • 1.
  • 2.
  • 3.