一、if else 的滥用
在实际业务开发中,常常面临多个条件判断的情况,如果不加思索就写代码,很容易出现一堆if else 的代码,导致代码逻辑不清晰,可阅读性很差,笔者现在改造中的一个历史遗留项目,大片大片的if else,给后来人带来很大的理解成本。过多的if else ,这是很明显的坏代码的味道,这时就应该考虑自己的设计是否合理,有没有更好的表达方式了。
二、if else 重构技巧
1.适当提前return
这是判断条件取反的做法,代码在逻辑表达上会更清晰,先看看下面这段demo示例代码:
/**
* 判断用户余额是否足够取现
* @param userId
* @param withdrawAmount 取现金额
* @return
*/
public Boolean checkBanlance(Long userId,BigDecimal withdrawAmount) {
User user = userDao.getUser(userId);
if (user != null) {
if (withdrawAmount.compareTo(user.getBalance()) <= 0) {
return Boolean.TRUE;
} else {
return false;
}
}
return false;
}
这段就可以通过if条件判断提前return来避免if嵌套的问题,如果user对象为空直接return false,如下:
public Boolean checkBanlance2(Long userId,BigDecimal withdrawAmount) {
User user = userDao.getUser(userId);
if (user == null) {
return Boolean.FALSE;
}
if (withdrawAmount.compareTo(user.getBalance()) <= 0) {
return Boolean.TRUE;
} else {
return false;
}
return false;
}
2.用布尔表达式来代替
上面的函数返回的结果是一个Boolean,那我们可以用一个布尔表达式来表示,如下:
public Boolean checkBanlance3(Long userId,BigDecimal withdrawAmount) {
User user = userDao.getUser(userId);
if (user == null) {
return Boolean.FALSE;
}
return withdrawAmount.compareTo(user.getBalance()) <= 0;
}
checkBanlance3 相比checkBanlance2的代码又可以简化了许多!
3.Map结合反射来代替if else
以一个获取手机号的demo代码来说明这种用法,代码如下:
public class UserMobile {
public String getMobile1() {
return "114774440";
}
public String getMobile2() {
return "1448720444";
}
public String getMobile3() {
return "1447191222";
}
public String getMobiel(String name) {
if (StringUtils.equals(name, "xiaoming")) {
return this.getMobile1();
} else if (StringUtils.equals(name, "xiaohua")) {
return this.getMobile2();
} else if (StringUtils.equals(name, "xiaozhang")) {
return this.getMobile3();
}
return "";
}
}
getMobiel方法根据姓名来获取手机号,不同的人返回不同的手机号,按照现在的写法,if else 很容易被滥用,尤其是要新增其他用户的时候。这时候我们可以考虑使用map来代替if else,
思路是:将name作为key,将getMobile1/getMobile2/getMobile3这三个方法名作为value,放在map中,然后根据参数name,获取对应的方法名,然后使用反射来调用相关的方法。示例代码如下:
public class UserMobile {
static Map map = new HashMap();
static {
map.put("xiaoming", "getMobile1");
map.put("xiaohua", "getMobile2");
map.put("xiaozhang", "getMobile3");
}
public String getMobile1() {
return "114774440";
}
public String getMobile2() {
return "1448720444";
}
public String getMobile3() {
return "1447191222";
}
public String getMobiel(String name) throws Exception {
String methodName = map.get(name);
if (StringUtils.isEmpty(methodName)) {
return "";
}
Method method = this.getClass().getMethod(methodName,new Class[]{});
String mobile = (String) method.invoke(this);
return mobile;
}
public static void main(String[] args) throws Exception {
UserMobile userMobile = new UserMobile();
String mobile = userMobile.getMobiel("xiaoming");
System.out.println("mobile:"+mobile);
}
}
这种用法一般多用于需要选择不同的执行方法时可以尝试使用! 除了这种处理方式外,还可以使用枚举。
4.用枚举来代替if else
public enum UserInfo {
XIAOMING("xiaoming"){
@Override
public String getUserMobile() {
return "13652222";
}
},
XIAOHUA("xiaohua"){
@Override
public String getUserMobile() {
return "12555555896";
}
},
XIAOZHANG("xiaozhang") {
@Override
public String getUserMobile() {
return "12555555896";
}
};
private String userName;
private String mobile;
UserInfo(String userName) {
this.userName = userName;
}
public String getUserName() {
return userName;
}
public void setUserName(String userName) {
this.userName = userName;
}
public void setMobile(String mobile) {
this.mobile = mobile;
}
public String getMobile(String name) {
for (UserInfo userInfo : UserInfo.values()) {
if (StringUtils.equals(name,userInfo.getUserName())) {
userInfo.getUserMobile();
}
}
return "";
}
public abstract String getUserMobile();
}
5.应用多态来代替if else
public class RestaurantService {
public void provideService(String worker) {
if (StringUtils.equals(worker,"waiter")) {
System.out.println("拿瓶红酒");
} else if (StringUtils.equals(worker,"cook")){
System.out.println("做一盘番茄鸡蛋");
} else if (StringUtils.equals(worker,"cleaner")) {
System.out.println("打扫一下卫生");
} else {
System.out.println("sorry,没有此项服务");
}
}
}
这是以餐厅的服务场景为例一个demo,这段代码写法,很容易引发if else的滥用,如果再增加一些服务,继续这种用法,只能不断的if else,那代码的可阅读性极差,且逻辑繁杂。
我们可以考虑使用多态,将不同的服务定义为一个子类,将每种服务封装到子类中。
public interface RestService {
void provideService();
}
public class Cleaner implements RestService {
@Override
public void provideService() {
System.out.println("打扫一下卫生");
}
}
public class Cook implements RestService {
@Override
public void provideService() {
System.out.println("做一盘番茄鸡蛋");
}
}
public class Waiter implements RestService {
@Override
public void provideService() {
System.out.println("拿瓶红酒");
}
}
public class RestaurantService {
static Map registerService = new HashMap<>();
static {
registerService.put("waiter", new Waiter());
registerService.put("cook", new Cook());
registerService.put("cleaner", new Cleaner());
}
public void getService(String name) {
RestService restService = registerService.get(name);
if (restService==null) {
System.out.println("sorry,没有此项服务");
return;
}
restService.provideService();
}
public static void main(String[] args) {
RestaurantService factory = new RestaurantService();
factory.getService("cleaner");
}
}
将waiter、cook、cleaner这三个角色分别定义三个子类,通过接口来调用需要用到的服务。
RestaurantService类中,将三个服务注册到map中,根据key来获取对应的服务,这也是用map来代替if else的一种用法。
注意,本文探讨的是过多的if else的重构技巧,不要为了重构而重构。
————————————————
原文链接:https://blog.csdn.net/song19890528/article/details/79574032