【SpringBoot】applicationContext.getBeansOfType(class)获取某一接口所有实现类,应用于策略模式

一、问题的提出

在实际工作中,我们经常会遇到一个接口及多个实现类的情况,并且在不同的条件下会使用不同的实现类。

二、应用场景

springboot 项目中通过 ApplicationContext.getBeansOfType(class) 获取某一接口的所有实现类,并通过枚举完成策略模式,替代 if/else,使代码更加优雅易于拓展。

三、ApplicationContext.getBeansOfType(class) 介绍

java 复制代码
    <T> Map<String, T> getBeansOfType(@Nullable Class<T> var1) throws BeansException;

从上面的源码上我们可以看出来这个方法能返回一个接口的全部实现类(前提是所有实现类都必须由 Spring IoC 容器管理

java 复制代码
        Map<String, TrafficModeService> map = applicationContext.getBeansOfType(TrafficModeService.class);

从上面的代码上(下面案例中工厂类有) ,Map 中的 String 的值是各个实现类的名称 busModeServiceImpl、trainModeServiceImpl(首字母小写),Map 中的 value 是各个 key 对应的策略实现类

四、案例 demo

假设从 A 点到 B 点有多种交通方式,每种交通方式的费用不同,可以根据乘客的需要进行选择。按照该需求,设计如下:

有一个交通方式的接口,接口有两个方式,一个查询费用、一个查询该交通方式的类型,同时,我们可以用一个枚举类型类标识交通类型。

我们还需要一个工厂类来根据交通类型标识查找该交通类型的 Bean 实例,从而使用该实例,获得交通类型的详细信息及该交通类型的操作。

1、TrafficCodeEmun 枚举制定接口信息

java 复制代码
@AllArgsConstructor
public enum TrafficCodeEmun {

    TRAIN("TRAIN","火车"),
    BUS("BUS","大巴"),

    ;

    private final String code;
    private final String desc;
}

2、TrafficModeFactory 工厂类获取接口实现 bean,并存储到 ConcurrentHashMap,通过枚举获取对应的实现 bean

java 复制代码
@Component
@Slf4j
public class TrafficModeFactory implements ApplicationContextAware {

    public static final ConcurrentHashMap<TrafficCodeEmun, TrafficModeService> TRAFFIC_BEAN_MAP = new ConcurrentHashMap<>();

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        log.info("TrafficModeFactory 启动开始");
        Map<String, TrafficModeService> map = applicationContext.getBeansOfType(TrafficModeService.class);

        map.forEach((key, value) -> TRAFFIC_BEAN_MAP.put(value.getCode(), value));
        log.info("TrafficModeFactory 启动完成");
    }

    public static <T extends TrafficModeService> T getTrafficMode(TrafficCodeEmun code) {
        return (T) TRAFFIC_BEAN_MAP.get(code);
      }

}

3、定义策略接口 TrafficModeService

java 复制代码
public interface TrafficModeService {

    /**
     * 查询交通方式编码
     * @return 编码
     */
    TrafficCodeEmun getCode();
     

    /**
     * 查询交通方式的费用,单位:分
     * @return 费用
     */
    Integer getFee();

}

4、策略实现类 BusModeServiceImpl、TrainModeServiceImpl

java 复制代码
@Service
public class TrainModeServiceImpl implements TrafficModeService {

    /**
     * 查询交通方式编码
     * @return 编码
     */
    @Override
    public TrafficCodeEmun getCode() {
        return TrafficCodeEmun.TRAIN;
    }

    /**
     * 查询交通方式的费用,单位:分
     * @return 费用
     */
    @Override
    public Integer getFee() {
        return 5000;
    }

}

5、定义 controller

java 复制代码
    @PostMapping("/test3")
    public Integer test3() {
        Integer fee = TrafficModeFactory.getTrafficMode(TrafficCodeEmun.TRAIN).getFee();
        return fee;
    }

6、启动项目,访问 localhost:8080/testUtils/test3 测试即可看到对应接口返回

java 复制代码
5000

五、总结

  • 一个策略接口被多个策略实现类所实现,具体使用哪一种根据用户选择的类型来和 Map 里的 key 做匹配,获取对应的实现来调用具体的策略方法。
  • 使用 ConcurrentHashMap ,而不使用 HashMap ,是 put 的时候,键和值都不能为空,防止 key 对应的实现类没有注入进去,导致空指针的问题。

六、参考文档

使用Spring的getBeansOfType实现接口多实现类的动态调用

applicationContext.getBeansOfType(class)获取某一接口的所有实现类,应用于策略模式简单demo

相关推荐
Rysxt_7 分钟前
Spring Boot 集成 Spring AI OpenAI Starter 教程
java·spring boot·后端·ai
青云交19 分钟前
Java 大视界 -- Java 大数据在智能家居场景联动与用户行为模式挖掘中的应用
java·大数据·智能家居·边缘计算·户型适配·行为挖掘·场景联动
AAA修煤气灶刘哥26 分钟前
ES 高级玩法大揭秘:从算分骚操作到深度分页踩坑,后端 er 速进!
java·后端·elasticsearch
江团1io030 分钟前
深入解析MVCC:多版本并发控制的原理与实现
java·经验分享·mysql
树码小子39 分钟前
Java网络编程:(socket API编程:UDP协议的 socket API -- 回显程序的服务器端程序的编写)
java·网络·udp
君宝43 分钟前
Linux ALSA架构:PCM_OPEN流程 (二)
java·linux·c++
云深麋鹿1 小时前
数据链路层总结
java·网络
fire-flyer1 小时前
响应式客户端 WebClient详解
java·spring·reactor
北执南念1 小时前
基于 Spring 的策略模式框架,用于根据不同的类的标识获取对应的处理器实例
java·spring·策略模式
王道长服务器 | 亚马逊云1 小时前
一个迁移案例:从传统 IDC 到 AWS 的真实对比
java·spring boot·git·云计算·github·dubbo·aws