重构——改写 if else 逻辑

繁多的 if else，if 中的 if else，else 中的 if else，这些庞大的逻辑看上去就让人云山雾绕，是时候祭出重构公式来简化它了。

思路

重构的思路为：

if 和 else 的逻辑我们只能二选一

对于 if，我们可以在 if 的逻辑块里加上一个 return，表明执行完 if 之后，不再涉及 else 的代码；对于 else，由于之前已确保 if 的逻辑执行后会返回，我们大可放心地将 else 的代码提出来，去掉 else 这个关键字

实战

一个例子

if ( m_Prop.A == 1 ) {
    if ( m_Prop.B == 1 ) {
        return "Good";
    } else if ( m_Prop.B == 2 ) {
        return "Not too bad";
    } else {
        return "Just so so";
    }
} else if ( m_Prop.A == 2 ) {
    if ( m_Prop.B == 1 ) {
        if ( m_Prop.C == 1 ) {
            return "Need improvement";
        } else {
            return "Need more improvement";
        }
    } else if ( m_Prop.B == 2 ) {
        return "Have a chance to improve";
    } else {
        return "Too bad";
    }
}

先从最外层的 else if ( m_Prop.A == 2 ) 开始修改，去掉 else

if ( m_Prop.A == 1 ) {
    if ( m_Prop.B == 1 ) {
        return "Good";
    } else if ( m_Prop.B == 2 ) {
        return "Not too bad";
    } else {
        return "Just so so";
    }
} 

if ( m_Prop.A == 2 ) {
    if ( m_Prop.B == 1 ) {
        if ( m_Prop.C == 1 ) {
            return "Need improvement";
        } else {
            return "Need more improvement";
        }
    } else if ( m_Prop.B == 2 ) {
        return "Have a chance to improve";
    } else {
        return "Too bad";
    }
}

然后以此类推，得到如下代码

if ( m_Prop.A == 1 ) {
    if ( m_Prop.B == 1 ) {
        return "Good";
    }
    
    if ( m_Prop.B == 2 ) {
        return "Not too bad";
    }
    
    return "Just so so";
} 

if ( m_Prop.A == 2 ) {
    if ( m_Prop.B == 1 ) {
        if ( m_Prop.C == 1 ) {
            return "Need improvement";
        }
        
        return "Need more improvement";
    }
    
    if ( m_Prop.B == 2 ) {
        return "Have a chance to improve";
    }
    
    return "Too bad";
}

就这样，去掉了 else，代码看起来清晰了很多，心情也变好了。

好处

1. 可读性

   当函数中出现任一一个 return，意味着分支到此为止。这样一个个 return，将一块块的逻辑分支分段。读代码的人只需专心理解每一段的逻辑，而不必耗费精力去深挖该段逻辑之后是否还有额外的处理——因为我们代码已经 return 了。

2. 避免修改

   当然，我们也可以在每个分支的结尾，将返回值赋给一个变量，在函数的末位返回这个变量。这样保证了函数只有一次返回，并且该处还可以做一些诸如异常捕捉的统一处理。

   但是，这样就没法保证，这个作为返回值的变量，不会再后续的处理中被意外地修改。俗话说：夜长梦多。还是该出手时就出手，能返回就返回。

参考来源

• 维护一个五六百行的程序就已经力不从心了怎么办？