优化JS条件语句

  1. 多个条件使用Array.includes

    function test(fruit){
        if(fruit === 'apple' || fruit === 'strawaberry') {
            console.log('red')
        }
    }
    

    乍一看,上面的例子看起来还不错,但是如果有许多水果需要判断,我们需要加入跟多的||,此时我们可以使用Array.includes重写上面的条件。

    function test(fruit) {
        const redFruits = ['apple', 'straweberry', 'cherry', 'cranberries']
        if(redFruits.includes(fruit)) {
            console.log('red')
        }
    }
    
  2. 更少的嵌套,尽早返回

    让我们扩展前面的例子,加入另外两个条件:

    • 如果没有提供水果名称,抛出错误。
    • 如果红色水果数量超出10个,接受并打印。
    /* return early when invalid condititions found */
    function test(fruit, quantity) {
        const redFruits = ['apple', 'straweberry', 'cherry', 'cranberries']
        
        // condition 1:throw error early
        if(!fruit) throw new Error('No fruit!')
        
        // condition 2: must be red
        if(redFruits.includes(fruit)) {
            console.log('red')
            
            // condition 3: must be big quantity
            if(quantity > 10) {
                console.log('big quantity')
            }
        }
    }
    

    这样,我们就少了一层嵌套。这种编码风格很好,尤其是当你有很长的if 语句时(想象一下,你需要滚动到最底部才能知道还有一个 else 语句,这并不酷)

    通过反转条件和提早返回,我们可以进一步减少嵌套。看看下面的条件 2,我们是怎么做的:

    /* return early when invalid conditions found */
     
    function test(fruit, quantity) {
      const redFruits = ['apple', 'strawberry', 'cherry', 'cranberries'];
     
      if (!fruit) throw new Error('No fruit!'); // condition 1: throw error early
      if (!redFruits.includes(fruit)) return; // condition 2: stop when fruit is not red
     
      console.log('red');
     
      // condition 3: must be big quantity
      if (quantity > 10) {
        console.log('big quantity');
      }
    }
    

    通过反转条件 2 的条件,我们的代码现在没有嵌套语句。当我们有很长的逻辑要处理时,这种技术是有用的,当一个条件没有满足时,我们想要停止进一步的处理。

  1. 使用默认的函数参数和结构

    我想下面的代码对您来说可能很熟悉,我们在使用 JavaScript时总是需要检查 nullundefined值并分配默认值:

    function test(fruit, quantity) {
      if (!fruit) return;
      const q = quantity || 1; // if quantity not provided, default to one
    
      console.log(`We have ${q} ${fruit}!`);
    }
    
    //test results
    test('banana'); // We have 1 banana!
    test('apple', 2); // We have 2 apple!
    

    事实上,我们可以通过指定默认的函数参数来消除变量 q。

    function test(fruit, quantity = 1) { // if quantity not provided, default to one
      if (!fruit) return;
      console.log(`We have ${quantity} ${fruit}!`);
    }
     
    //test results
    test('banana'); // We have 1 banana!
    test('apple', 2); // We have 2 apple!
    

    更简单和直观,不是吗?请注意,每个参数都可以有自己的默认函数参数。例如,我们也可以为 fruit 赋值:function test(fruit = 'unknown', quantity = 1)

    如果我们的 fruit 是一个对象呢?我们可以指定默认参数吗?

    function test(fruit) {
      // printing fruit name if value provided
      if (fruit && fruit.name)  {
        console.log (fruit.name);
      } else {
        console.log('unknown');
      }
    }
     
    //test results
    test(undefined); // unknown
    test({ }); // unknown
    test({ name: 'apple', color: 'red' }); // apple
    

    请看上面的示例,如果 fruit.name 是可用的,我们将打印该水果名称,否则我们将打印 unknown。我们可以避免使用与默认函数参数和解构对条件 fruit && fruit.name 进行检查。

    // destructing - get name property only
    // assign default empty object {}
    function test({name} = {}) {
      console.log (name || 'unknown');
    }
     
    //test results
    test(undefined); // unknown
    test({ }); // unknown
    test({ name: 'apple', color: 'red' }); // apple
    

    因为我们只需要水果中的属性 name,所以我们可以使用{name} 来解构,然后我们可以在代码中使用 name 作为变量,而不是 fruit.name

    我们还将空对象 {} 指定为默认值。如果我们不这样做,当执行 test(undefined),不能解构 undefinednull 的属性名时,您将会得到错误。因为在 undefined中没有 name 属性。

    如果您不介意使用第三方库,有一些方法可以减少 null检查:

    • 使用 Lodashget 函数
    • 使用 Facebook 的开源库 idx(以及 Babeljs)
    // Include lodash library, you will get _
    function test(fruit) {
      console.log(__.get(fruit, 'name', 'unknown'); // get property name, if not available, assign default value 'unknown'
    }
     
    //test results
    test(undefined); // unknown
    test({ }); // unknown
    test({ name: 'apple', color: 'red' }); // apple
    

    您可以在 这里 运行演示代码。此外,如果你喜欢函数式编程(FP),你可以选择使用 Lodash fp, 即 Lodash的函数式版本(方法改为 getgetOr)。

  2. 选择Map或对象字面量,而不是 Switch 语句

    让我们看看下面的例子,我们想要基于颜色打印水果名称:

    function test(color) {
      // use switch case to find fruits in color
      switch (color) {
        case 'red':
          return ['apple', 'strawberry'];
        case 'yellow':
          return ['banana', 'pineapple'];
        case 'purple':
          return ['grape', 'plum'];
        default:
          return [];
      }
    }
     
    //test results
    test(null); // []
    test('yellow'); // ['banana', 'pineapple']
    

    上面的代码似乎没有什么问题,但我发现它相当冗长。同样的结果可以通过对象字面量和更简洁的语法来实现:

    // use object literal to find fruits in color
      const fruitColor = {
        red: ['apple', 'strawberry'],
        yellow: ['banana', 'pineapple'],
        purple: ['grape', 'plum']
      };
     
    function test(color) {
      return fruitColor[color] || [];
    }
    

    或者,可以使用 Map 来实现相同的结果:

    // use Map to find fruits in color
      const fruitColor = new Map()
        .set('red', ['apple', 'strawberry'])
        .set('yellow', ['banana', 'pineapple'])
        .set('purple', ['grape', 'plum']);
     
    function test(color) {
      return fruitColor.get(color) || [];
    }
    

    Map 是 ES2015 以后可用的对象类型,允许您存储键值对。

    我们应该禁止使用 switch语句吗?不要把自己局限于此。就我个人而言,我尽可能地使用对象字面量,但是我不会设置严格的规则来阻止它,使用对您的场景有意义的任何一个。

    TL;DR; 重构的语法

    对于上面的示例,我们实际上可以重构代码,以使用 Array.filter 获得相同的结果。

    const fruits = [
       { name: 'apple', color: 'red' },
       { name: 'strawberry', color: 'red' },
       { name: 'banana', color: 'yellow' },
       { name: 'pineapple', color: 'yellow' },
       { name: 'grape', color: 'purple' },
       { name: 'plum', color: 'purple' }
    ];
     
    function test(color) {
     // use Array filter to find fruits in color
     
     return fruits.filter(f => f.color == color);
    }
    
  3. 所有或部分使用 Array.every&Array.some 的条件

    最后一个技巧是关于使用新的(但不是很新)Javascript 数组函数来减少代码行。看看下面的代码,我们想检查所有的水果是否都是红色的:

    const fruits = [
        { name: 'apple', color: 'red' },
        { name: 'banana', color: 'yellow' },
        { name: 'grape', color: 'purple' }
      ];
     
    function test() {
      let isAllRed = true;
     
      // condition: all fruits must be red
      for (let f of fruits) {
        if (!isAllRed) break;
        isAllRed = (f.color == 'red');
      }
     
      console.log(isAllRed); // false
    }
    

    代码太长了!我们可以用 Array.every来减少行数:

    const fruits = [
        { name: 'apple', color: 'red' },
        { name: 'banana', color: 'yellow' },
        { name: 'grape', color: 'purple' }
      ];
     
    function test() {
      // condition: short way, all fruits must be red
      const isAllRed = fruits.every(f => f.color == 'red');
     
      console.log(isAllRed); // false
    }
    

    现在干净多了,对吧?类似地,如果我们想用一行代码来判断任何一个水果是否为红色,我们可以使用 Array.some

    const fruits = [
        { name: 'apple', color: 'red' },
        { name: 'banana', color: 'yellow' },
        { name: 'grape', color: 'purple' }
    ];
     
    function test() {
      // condition: if any fruit is red
      const isAnyRed = fruits.some(f => f.color == 'red');
     
      console.log(isAnyRed); // true
    }
    
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,242评论 5 459
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,769评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,484评论 0 319
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,133评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,007评论 4 355
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,080评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,496评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,190评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,464评论 1 290
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,549评论 2 309
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,330评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,205评论 3 312
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,567评论 3 298
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,889评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,160评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,475评论 2 341
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,650评论 2 335

推荐阅读更多精彩内容