优化 if-else 语句的最佳方案

---

前言

尽量优化 if-else 语句，防止代码结构变得复杂混乱，防止过多的大括号，导致代码阅读困难。

---

一、条件表达式优化

1、先行计算条件表达式

一般禁止在 if 条件表达式中进行函数调用及运算，此时应先行计算出条件表达式的结果，避免 if 条件表达式过于庞大，导致代码阅读困难。允许简短的调用及运算、数据比较等单一简短的表达式。

原始代码：

if (StrUtil.equals(UserService.getUserStatus(user.getIp()), "online")) { // if (user.isConnect()) 简短的调用及运算可以允许
    // TODO
}

优化为：

String userStatus = userService.getUserStatus(user.getIp()); // 先行计算条件表达式
if (StrUtil.equals(userStatus, "online")) { // 不建议将 boolean isOnline = StrUtil.equals(userStatus, "online"); 也先行计算出，意义不大。
    // TODO
}

---

二、条件判断结构优化

1、使用三目运算符

当 if-else 语句具有两个分支，且分支业务简短时，考虑使用三目运算符。

原始代码：

public boolean isOnline() {
	if (StrUtil.equals(userStatus, "online")) {
	    return true;
	} else {
		return false;
	}
}

优化为：

public boolean isOnline() {
	return StrUtil.equals(userStatus, "online") ? true : false;
}

2、省略最后的 else

else 后面没有其他业务时，可省略最后的 else，使代码简洁。

原始代码：

public void addUser() {
	if (StrUtil.equals(userStatus, "online")) {
	    // TODO 1
	    return;
	} else {
		// TODO 2
		return;
	}
	// else 后面没有其他业务时，可省略最后的else，使代码简洁
}

优化为：

public void addUser() {
	if (StrUtil.equals(userStatus, "online")) {
	    // TODO 1
	    return;
	}
	
	// TODO 2
	return;
}

3、减少 if 语句包裹的内容

减少 if 语句包裹的内容，选择包裹代码少的条件业务，异常状态直接返回。

原始代码：

public void addUser() {
	if (StrUtil.equals(userStatus, "online")) {
	    // TODO 1.1
	    // TODO 1.2
	    return;
	}
	
	// TODO 2
	return;
}

优化为：

public void addUser() {
	if (!StrUtil.equals(userStatus, "online")) {
		// TODO 2
	    return;
	}
	
	// TODO 1.1
	// TODO 1.2
	return;
}

4、异常状态先行返回

减少 if 语句包裹的内容，选择包裹代码少的条件业务，异常状态直接返回。

原始代码：

public void editUser() {
	// 用户存在时，进行大量处理业务
	if (dbUser != null) {
	    // 处理用户数据
	    ...
	    ...
	    ...
	}

优化为：

public void editUser() {
	// 用户不存在时，直接返回
	if (dbUser == null) {
	    return;
	}
	
	// 处理用户数据
	...
	...
	...
}

5、责任链模式

当条件过滤等较多时，可封装重用时，可采用责任链模式。

原始代码：

public void addUser(User user) {
	// 1.非空校验
	if (StrUtil.isBlank(user.getUsername())) {
		throw new RuntimeException("用户名为空！");
	}
	if (StrUtil.isBlank(user.getPassword())) {
		throw new RuntimeException("密码为空！");
	}
	...
	
	// 2.格式校验
	if (!ValidUtil.isIdCardNo(user.getIdCardNo())) {
		throw new RuntimeException("身份证号格式错误！");
	}
	if (!ValidUtil.isEmail(user.getEmail())) {
		throw new RuntimeException("手机号格式错误！");
	}
	if (!ValidUtil.isEmail(user.getEmail())) {
 		throw new RuntimeException("邮箱格式错误！");
	}
   	...

	// 3.xx校验
	...
}

创建处理器链接口：

/**
 * 处理器链接口
 */
public interface UserChainHandler {
    
    /**
     * 执行责任链逻辑
     *
     * @param user 责任链执行入参
     */
    void handler(User user);
}

创建处理器1，实现处理器链接口：

/**
 * 非空校验处理器
 */
@Component
@Order(1) // 指定注入顺序
public class UserParamNullValidChainHandler implements UserChainHandler {
    
    @Override
    public void handler(User user) {
	   	// 1.非空校验
	    if (StrUtil.isBlank(user.getUsername())) {
			throw new RuntimeException("用户名为空！");
		}
		if (StrUtil.isBlank(user.getPassword())) {
			throw new RuntimeException("密码为空！");
		}
		...
    }
}

创建处理器2，实现处理器链接口：

/**
 * 格式校验处理器
 */
@Component
@Order(2) // 指定注入顺序
public class UserParamFormatValidChainHandler implements UserChainHandler {
 
    @Override
    public void handler(User user) {
	    // 2.格式校验
		if (!ValidUtil.isIdCardNo(user.getIdCardNo())) {
			throw new RuntimeException("身份证号格式错误！");
		}
		if (!ValidUtil.isEmail(user.getEmail())) {
			throw new RuntimeException("手机号格式错误！");
		}
		if (!ValidUtil.isEmail(user.getEmail())) {
	 		throw new RuntimeException("邮箱格式错误！");
		}
	   	...
    }
}

创建一个责任链上下文容器，用于存储与责任链相应的子任务：

@Component
@RequiredArgsConstructor
public class UserChainContext {
    
    private final List<UserChainHandler> userChainHandlerList; // 自动注入责任链处理器
    
    /**
     * 责任链组件执行
     *
     * @param requestParam 请求参数
     */
    public void handler(User user) {
        // 此处根据 Ordered 实际值进行排序处理
        userChainHandlerList.forEach(x -> x.handler(user));
    }
}

业务调用：

public void addUser(User user) {
	// 执行责任链
	userChainContext.handler(user);
}

---

三、使用 Optional 类

1、对象判空 - 对象为空，则抛出异常

Optional.ofNullable(T).orElseThrow(() -> new Exception(“”));

原始代码：

if (user == null) {
    throw new Exception("未查询到用户信息");
}

优化为：

Optional.ofNullable(user).orElseThrow(() -> new Exception("未查询到用户信息"));

2、对象判空 - 对象非空，则执行方法

Optional.ofNullable(T).ifPresent(x -> fun(x));

原始代码：

if (user != null) {
    update(user); // 执行方法调用
}

优化为：

Optional.ofNullable(user).ifPresent(user -> update(user));

3、更多 Optional 用法

---

总结

根据相应场景使用合适的优化方案。

Optional 类虽然简洁，不过采用这种链式编程，虽然代码优雅了。但是，逻辑性没那么明显，可读性有所降低，酌情使用。

相关文章

消除 if-else 和 switch 多分支语句的最佳方案