(Java代码中的if语句优化(工厂模式+策略模式))
一、场景重现
代码中有大量的 if else if
代码:
if (e instanceof HttpMediaTypeNotSupportedException) {
e.printStackTrace();
}
else if (e instanceof MethodArgumentNotValidException) {
e.printStackTrace();
}
else if (......) {
.........
}
...........
总所周知,认知复杂度是衡量方法控制流理解难度的指标, 认知复杂度高的方法将难以维护。so,这样的代码是十分难以维护的。
二、开始优化
上述代码中我们可以看到它的基本骨架是:if(e instanceof Object){//逻辑}
,下面我们就以它为核心进行优化。
2.1 核心接口
首先我们定义一个核心接口,这个接口定义了我们要实现的方法。
public interface MyWorks {
boolean doWork(Exception e);
}
2.2 接口实现
我们定义几个实现类,来对核心接口进行实现,前面的代码中有几个 if 这里就定义几个实现,为了方便起见,我们这里定义了三个。
@Component
public class WorkersOne implements MyWorks {
private static final Logger WorkersOneLog = LoggerFactory.getLogger(WorkersOne.class);
@Override
public boolean doWork(Exception e) {
WorkersOneLog.info("我被执行了");
if(e instanceof BindException){
WorkersOneLog.info("如果是参数校验异常");
return true;
}
return false;
}
}
@Component
public class WorkersTwo implements MyWorks {
private static final Logger WorkersTwoLog = LoggerFactory.getLogger(WorkersTwo.class);
@Override
public boolean doWork(Exception e) {
WorkersTwoLog.info("我被执行了");
if(e instanceof HttpMessageNotReadableException){
WorkersTwoLog.info("请求参数异常");
return true;
}
return false;
}
}
@Component
public class WorkersThree implements MyWorks{
private static final Logger WorkersThreeLog = LoggerFactory.getLogger(WorkersThree.class);
@Override
public boolean doWork(Exception e) {
WorkersThreeLog.info("我被执行了");
if(e instanceof HttpRequestMethodNotSupportedException){
WorkersThreeLog.info("请求方式错误异常");
return true;
}
return false;
}
}
2.3 工厂实现
最后我们定义一个工厂类进行各种异常的管理和实现方法的调用
public class ExceptionFactory {
private static final Logger ExceptionFactoryLog = LoggerFactory.getLogger(ExceptionFactory.class);
private ExceptionFactory() {
throw new IllegalStateException("ExceptionFactory class");
}
private static final List<MyWorks> list = Stream.of(
new WorkersOne(),
new WorkersTwo(),
new WorkersThree()
).collect(Collectors.toList());
public static void getDoWorkers(Exception e) {
boolean status = true;
for (MyWorks works : list){
if(works.doWork(e)) {
status = false;
break;
}
}
if(status){
ExceptionFactoryLog.info("该异常未定义");
}
}
}
三、使用
一切都完成后,我们愉快的使用一下看看效果。
随便写一个 controller 类:
@RestController
public class ClientController {
private static final Logger clientControllerLog = LoggerFactory.getLogger(ClientController.class);
@PostMapping("/ex")
public void run(@RequestBody int s){
clientControllerLog.info("输出了 :{}",s+1);
}
}
启动项目,使用postman或其它工具调用 127.0.0.1:8080/ex,故意输入错误参数。
使用GET传参: 使用POST传参,但参数故意错误: 可以看到输出结果和我们使用 if 语句并无区别,但是这种的可维护性却明显提高。