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 语句并无区别，但是这种的可维护性却明显提高。