借助spring的IOC能力消除条件判断

news2024/11/17 9:26:01

shigen坚持更新文章的博客写手，记录成长，分享认知，留住感动。个人IP：shigen

在前边讲到了如何借助HashMap、枚举类、switch-case消除条件判断，这里讲到我们最常见的用spring的IOC能力来消除代码中的逻辑判断。其实大部分的设计模式用的方法也和接下来的这种类似。

我们先分析下最初的代码：

public String handleQuestion(String type, String detail) {
    if ("A".equals(type)) {
        return handleQuestionA(detail);
    } else if ("B".equals(type)) {
        return handleQuestionB(detail);
    } else if ("C".equals(type)) {
        return handleQuestionC(detail);
    } else {
        throw new IllegalArgumentException("invalid type: " + type);
    }
}

其实就是想用给定的问题类型处理对应的问题。抽取出对应的要素，我们生成一个抽象类：

生成抽象类

interface QuestionHandler {
    /** 处理问题 */
    String handle(String detail);
    /** 获得问题类型 */
    String getType();
}

这里抽象出QuestionHandler这个类，对应的两个方法：处理问题和获得问题类型

具体实现

对于各种问题场景，只需要实现QuestionHandler接口即可。

@Service
class QuestionHandlerA implements QuestionHandler {

    @Override
    public String handle(String detail) {
        return handleQuestionA(detail);
    }

    @Override
    public String getType() {
        return "A";
    }
}

@Service
class QuestionHandlerB implements QuestionHandler {
    @Override
    public String handle(String detail) {
        return handleQuestionB(detail);
    }

    @Override
    public String getType() {
        return "B";
    }
}

关键的一步- IOC注入

还是先来看看代码的设计：

@Resource
private List<QuestionHandler> questionHandlers;

public String handleQuestionV5(String type, String detail) {
    return questionHandlers.stream()
            .filter(handler -> handler.getType().equals(type)).findFirst()
            .map(handler -> handler.handle(detail))
            .orElseThrow(() -> new IllegalArgumentException("invalid type: " + type));
}