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跳出来,代码会好很多,且行且体会。