java SPI实现类中注入spring bean对象

在项目中,用到了SPI来扩展一些功能,发现很多实现类中用到了bean对象,并且都是通过getBean的方式每次都去拿,感觉不是很方便,而且速度也没有直接使用对象快。

正好安排的工作就是优化那一块的代码,所以就改造了一下,让SPI实现类中支持注入spring bean依赖,下面的类名有所改变,还有使用到的方法,大家可以按需改造。

  1. 首先,为了不影响之前的逻辑,定义了一个接口BeanResource,需要自动注入bean的类,就实现这个接口;

    java 复制代码
    /**
     * 实现类中,实现了这个接口就会自动注入bean,配合注入注解等使用
     */
    public interface BeanResource {
    }
  2. 通过ApplicationContext类的getAutowireCapableBeanFactory()方法拿到AutowireCapableBeanFactory工厂类;

    可以通过hutool工具包的SpringUtil.getApplicationContext().getAutowireCapableBeanFactory()方式拿到,也可以自己实现一个ApplicationContextAware接口,然后在构造器中获取ApplicationContext对象。

  3. 通过autowireBeanProperties方法注入bean对象
    autowireBeanProperties方法有三个形参,第一个是要注入bean依赖的对象,第二个是注入的方式(根据类型或名称),第三个参数是是否依赖检查,这里我们用true,这样注入bean依赖有问题会报错。

    java 复制代码
    // 首先通过拿到所有的子类
    final ServiceLoader<XXXXX> loader = ServiceLoader.load(XXXXX.class);
    final AutowireCapableBeanFactory beanFactory = SpringUtil.getApplicationContext().getAutowireCapableBeanFactory();
    for (XXXXX xxxxx : loader) {
        // 如果需要注入bean对象
        if (xxxxx instanceof BeanResource) {
            try {
                // 给xxxxx注入bean对象,支持@Resource和@Autowired注入,还有set注入; 按类型注入(其实只要你用平常的注入方式BY NAME也可以,除了构造器注入,因为对象是已经创建好了的);进行依赖检查,如果有问题直接报错
                beanFactory.autowireBeanProperties(xxxxx, AutowireCapableBeanFactory.AUTOWIRE_BY_TYPE, true);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
  4. 最后在扩展类中使用,除了构造器注入,其他常用的注入方式都行。

    java 复制代码
    @AutoService(XXXXX.class)
    public class Sub implements XXXXX, BeanResource {
        @Resource
        private A a;
    	@Autowired
    	private B b;
    	@Setter
    	private C c;
    }
相关推荐
布朗克16812 分钟前
34 JVM深入理解
java·jvm
Flittly21 分钟前
【AgentScope Java新手村系列】(4)结构化输出
java·spring boot·spring·ai
wzg19690226wzg22 分钟前
rust 学习 泛型
开发语言·学习·rust
techdashen24 分钟前
Rust 基础设施团队 2025 Q4 回顾与 2026 Q1 计划
开发语言·后端·rust
红宝村村长24 分钟前
torch.autograd.Function.apply()
开发语言·python
AI科技星25 分钟前
《数术工坊:非欧射影录》类型:硬核光影·几何本源
c语言·开发语言·网络·量子计算·agi
何以解忧,唯有..33 分钟前
Python 中的继承机制:从基础到高级用法详解
java·开发语言·python
Yiyaoshujuku38 分钟前
化合物数据集API接口(数据结构及样例)
java·网络·数据结构
plainGeekDev1 小时前
算法刷题笔记:一维DP没那么难,状态想清楚就赢了一半
java·算法·面试
IceBing1 小时前
还在一个个连接 Arthas?这个开源平台支持批量诊断 JVM
java