0%

简化代码的小技巧

简化代码的小技巧

今天重构之前写的一段对账代码,基本逻辑是先校验,然后更新一下记录状态,然后处理对账主体的数据修改。抽象出来就是要三步,每一步出错或者校验不通过,就直接返回错误信息。抽象代码如下

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public class Check{
public boolean first(Map<String,String> message){
...
//出错或者校验不通过返回false,同时将错误信息保存到message中。
...
return true;
}

public boolean second(Map<String,String> message){
...
//出错或者校验不通过返回false,同时将错误信息保存到message中。
...
return true;
}

public boolean last(Map<String,String> message){
...
//出错或者校验不通过返回false,同时将错误信息保存到message中。
...
return true;
}
}

最直接写法

如果不考虑其他,第一反应肯定是这么写

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public class Test{

public static void main(String[] args){
Check check = new Check();
Map<String,String> message = new HashMap<>();
if(check.first(message)){
if(check.second(message)){
if(check.last(message)){
//commit;
}else{
//rollback;
}
} else{
//rollback;
}
}else{
//rollback;
}
System.out.println(message.get("msg"));
}
}

没错,项目中的代码就是这么写的,多层if嵌套,抽象出来还能看懂,添加上大段的业务逻辑,看起来就费劲了。

第一步优化

先来第一步优化,把多层if嵌套去掉。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
public class Test{

public static void main(String[] args){
Check check = new Check();
Map<String,String> message = new HashMap<>();
if(!check.first(message)){
System.out.println(message.get("msg"));
//rollback;
return;
}
if(!check.second(message)){
System.out.println(message.get("msg"));
//rollback;
return;
}
if(!check.last(message)){
System.out.println(message.get("msg"));
//rollback;
return;
}
//commit;
}
}

通过return提前返回,避免多层if嵌套。

再次精简

我觉得写代码能有上面的觉悟已经不错了。不过无意中发现还有更简单的,但是好像不容易理解。代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
public class Test{

public static void main(String[] args){
Check check = new Check();
Map<String,String> message = new HashMap<>();
if(!check.first(message) || !check.second(message) || !check.last(message)){
System.out.println(message.get("msg"));
//rollback;
return;
}
//commit;
}
}

利用判断条件的或,合并代码。原理是或判断如果前面的条件为真,就不再判断后面的条件。也就是前面如果报错,取反为真,则不判断后面的条件,直接进入if条件里面,也就不执行后面的代码而直接返回错误信息。只有当所有的操作都不报错,才会到最后的commit操作。