优化If else(简化代码)

if else 是我们写代码时,使用频率最高的关键词之一,然而有时过多的 if else 会让我们感到头晕。
当我们回过头看代码或者欣赏他人代码时,感到烦人。我总结了几种方式,可以参考呀。
1.使用 return
我们使用 return 去掉多余的 else,实现代码如下。

优化前代码:


if ("java".equals(str)) {
    // 业务代码......
} else {
    return;
}

优化后代码:

if (!"java".equals(str)) {
    return;
}
// 业务代码......

2.使用 Map
使用 Map 数组,把相关的判断信息,定义为元素信息可以直接避免 if else 判断,实现代码如下。
优化前代码

if (t == 1) {
    type = "name";
} else if (t == 2) {
    type = "id";
} else if (t == 3) {
    type = "mobile";
}

我们先定义一个 Map 数组,把相关判断信息存储起来:

Map<Integer, String> typeMap = new HashMap<>();
typeMap.put(1, "name");
typeMap.put(2, "id");
typeMap.put(3, "mobile");

之前的判断语句可以使用以下一行代码代替了:

type = typeMap.get(t);

3.使用三元运算符
三元运算符也叫三元表达式或者三目运算符/表达式,不过代表的都是一个意思,优化代码如下。
优化前代码:

Integer score = 81;
if (score > 80) {
    score = 100;
} else {
    score = 60;
}

优化后代码:

score = score > 80 ? 100 : 60;

4.合并条件表达式

String city = "唐山";
String area = "888";
String province = "唐山";
if ("唐山".equals(city)) {
    return "tang'shan";
}
if ("020".equals(area)) {
    return "tang'shan";
}
if ("唐山".equals(province)){
    return "tang'shan";
}

优化后代码:

if ("唐山".equals(city) || "888".equals(area) || "唐山".equals(province)){
    return"tang'shan";
}

5.使用枚举

Integer typeId = 0;
String type = "Name";
if ("Name".equals(type)) {
    typeId = 1;
} else if ("Age".equals(type)) {
    typeId = 2;
} else if ("Address".equals(type)) {
    typeId = 3;
}

优化时,我们先来定义一个枚举:

public enum TypeEnum {
    Name(1), Age(2), Address(3);
    public Integer typeId;
    TypeEnum(Integer typeId) {
        this.typeId = typeId;
    }
}

之前的 if else 判断就可以被如下一行代码所替代了:

typeId = TypeEnum.valueOf("Name").typeId;

6.使用 Optional

从 JDK 1.8 开始引入 Optional 类,在 JDK 9 时对 Optional 类进行了改进,增加了 ifPresentOrElse() 方法,我们可以借助它,来消除 if else 的判断,使用如下。

String str = "java";
if (str == null) {
    System.out.println("Null");
} else {
    System.out.println(str);
}

优化后代码:

Optional<String> opt = Optional.of("java");
opt.ifPresentOrElse(v -> 
	System.out.println(v), () -> System.out.println("Null"));

7.梳理优化判断逻辑

// 年龄大于 18
if (age > 18) {
    // 工资大于 5000
    if (salary > 5000) {
        // 是否漂亮
        if (pretty == true) {
            return true;
        }
    }
}
return false;

优化后代码:

if (age < 18) {
    return false;
}
if (salary < 5000) {
    return false;
}
return pretty; 

8.使用多态
继承、封装和多态是 OOP(面向对象编程)的重要思想,我们使用多态的思想,提供一种去除 if else 方法。

Integer typeId = 0;
String type = "Name";
if ("Name".equals(type)) {
    typeId = 1;
} else if ("Age".equals(type)) {
    typeId = 2;
} else if ("Address".equals(type)) {
    typeId = 3;
}

使用多态,我们先定义一个接口,在接口中声明一个公共返回 typeId 的方法,在添加三个子类分别实现这三个子类

public interface IType {
    public Integer getType();
}
 
public class Name implements IType {
    @Override
    public Integer getType() {
        return 1;
    }
}
 
public class Age implements IType {
    @Override
    public Integer getType() {
        return 2;
    }
}
 
public class Address implements IType {
    @Override
    public Integer getType() {
        return 3;
    }
}

注:为了简便我们这里把类和接口放到了一个代码块中,在实际开发中应该分别创建一个接口和三个类分别存储。

此时,我们之前的 if else 判断就可以改为如下代码:

IType itype = (IType) Class.forName("com.example." + type).newInstance();
Integer typeId = itype.getType();

9.选择性的使用 switch

if ("add".equals(cmd)) {
    result = n1 + n2;
} else if ("subtract".equals(cmd)) {
    result = n1 - n2;
} else if ("multiply".equals(cmd)) {
    result = n1 * n2;
} else if ("divide".equals(cmd)) {
    result = n1 / n2;
} else if ("modulo".equals(cmd)) {
    result = n1 % n2;
}

switch 代码:

switch (cmd) {
    case "add":
        result = n1 + n2;
        break;
    case "subtract":
        result = n1 - n2;
        break;
    case "multiply":
        result = n1 * n2;
        break;
    case "divide":
        result = n1 / n2;
        break;
    case "modulo":
        result = n1 % n2;
        break;
}

在Java更新中java14 switch可以这么用:

// java 14
switch (cmd) {
    case "add" -> {
        result = n1 + n2;
    }
    case "subtract" -> {
        result = n1 - n2;
    }
    case "multiply" -> {
        result = n1 * n2;
    }
    case "divide" -> {
        result = n1 / n2;
    }
    case "modulo" -> {
        result = n1 % n2;
    }
}

有的时候代码写多了,就可以体会到换一种方式,代码就会很清晰,遇到了一个判断条件很多,逻辑稍微复杂的东西,if else用多了,看着特别难受,多用条件return跳出来,代码会好很多,且行且体会。