对普通bean进行Autowired字段注入

Posted 2022-04-22 justry_deng

对普通bean进行Autowired字段注入

• 对普通bean进行Autowired字段注入
  • 工具类
  • 使用测试

工具类

import lombok.Getter;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.SmartInitializingSingleton;
import org.springframework.beans.factory.config.AutowireCapableBeanFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.lang.NonNull;
import org.springframework.stereotype.Component;

import java.util.Objects;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;

/**
 * 对普通bean(即:非spring-bean)的autowired注入支持
 *
 * @author JustryDeng
 * @since 2020/7/5 13:20:19
 */
@Slf4j
public abstract class ExtAutowiredInjector 
    
    public ExtAutowiredInjector() 
        Object obj = applyAutowired();
        ExtAutowiredInjector.inject(obj);
    
    
    /**
     * 对返回的object进行autowired注入处理
     *
     * @return 要进行注入处理的Object
     */
    protected Object applyAutowired() 
        return this;
    
    
    /**
     * 对目标对象中的被@Auwired标注了的字段进行注入
     *
     * @param obj
     *            目标obj
     */
    public static void inject(Object obj) 
        if (obj == null) 
            log.warn("obj is null, skip inject.");
            return;
        
        if (ExtAutowiredInjectorProcessor.applicationContext != null) 
            ExtAutowiredInjectorProcessor.applicationContext.publishEvent(new AutowiredInjectEvent(obj));
         else 
            ExtAutowiredInjectorProcessor.queue.add(obj);
        
    
    
    /**
     * Autowired注入支持事件
     *
     * @author JustryDeng
     * @since 2020/7/5 15:02:19
     */
    public static class AutowiredInjectEvent extends ApplicationEvent 
        
        @Getter
        private final Object bean;
        
        /**
         * Create a new ApplicationEvent.
         *
         * @param source
         *         the object on which the event initially occurred (never @code null)
         */
        public AutowiredInjectEvent(Object source) 
            super(source);
            Objects.requireNonNull(source, "source cannot be null.");
            this.bean = source;
        
    
    
    /**
     * @link ExtAutowiredInjector处理器
     *
     * @author JustryDeng
     * @since 2020/7/5 13:46:25
     */
    @Slf4j
    @Component
    public static class ExtAutowiredInjectorProcessor implements SmartInitializingSingleton, ApplicationListener<AutowiredInjectEvent> 
        
        static ApplicationContext applicationContext;
        
        static Queue<Object> queue = new ConcurrentLinkedQueue<>();
        
        private final AutowireCapableBeanFactory autowireCapableBeanFactory;
        
        public ExtAutowiredInjectorProcessor(AutowireCapableBeanFactory autowireCapableBeanFactory,
                                             ApplicationContext applicationContext) 
            this.autowireCapableBeanFactory = autowireCapableBeanFactory;
            ExtAutowiredInjectorProcessor.applicationContext = applicationContext;
        
        
        @Override
        public void onApplicationEvent(@NonNull AutowiredInjectEvent event) 
            Object bean = event.getBean();
            autowireCapableBeanFactory.autowireBean(bean);
            afterSingletonsInstantiated();
        
        
        @Override
        public void afterSingletonsInstantiated() 
            Object obj = queue.poll();
            while (obj != null) 
                autowireCapableBeanFactory.autowireBean(obj);
                obj = queue.poll();
            
        
    

使用测试

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.stereotype.Component;

/**
 * start class
 *
 * @author JustryDeng
 * @since 2022/4/21 11:41
 */
@SpringBootApplication
public class DemoApplication implements ApplicationRunner 
    
    public static void main(String[] args) 
        SpringApplication.run(DemoApplication.class, args);
    
    
    
    @Override
    public void run(ApplicationArguments args) throws Exception 
        // 启动springboot项目， 观察输入
        
        // 测试一，输出：hello 邓二洋~
        new TestClassOne().who();
        
        
        // 测试二，输出：hello 邓帅~
        TestClassTwo testClassTwo = new TestClassTwo();
        ExtAutowiredInjector.inject(testClassTwo);
        testClassTwo.who();
        // 重复注入也没关系
        //ExtAutowiredInjector.inject(testClassTwo);
        //ExtAutowiredInjector.inject(testClassTwo);
        //ExtAutowiredInjector.inject(testClassTwo);
        //testClassTwo.who();
    
    
    /**
     * 测试用法一：继承使用，自动完成注入
     */
    public static class TestClassOne extends ExtAutowiredInjector 
        
        @Autowired
        @Qualifier("demoApplication.DengErYang")
        private Mine mine;
        
        public void who() 
            System.err.println(mine.hello());
        
        
        @Override
        protected Object applyAutowired() 
            return this;
        
    
    
    /**
     * 测试用法一：不继承， 直接用@link ExtAutowiredInjector#inject(Object)注入
     */
    public static class TestClassTwo 
        
        @Autowired
        @Qualifier("demoApplication.DengShuai")
        private Mine mine;
        
        public void who() 
            System.err.println(mine.hello());
        
    
    
    public interface Mine 
        String hello();
    
    
    @Component(value = "demoApplication.DengErYang")
    public static class DengErYang implements Mine 
        @Override
        public String hello() 
            return "hello 邓二洋~";
        
    
    
    @Component(value = "demoApplication.DengShuai")
    public static class DengShuai implements Mine 
        @Override
        public String hello() 
            return "hello 邓帅~";
        
    
    

启动程序，控制台输出：

  .   ____          _            __ _ _
 /\\\\ / ___'_ __ _ _(_)_ __  __ _ \\ \\ \\ \\
( ( )\\___ | '_ | '_| | '_ \\/ _` | \\ \\ \\ \\
 \\\\/  ___)| |_)| | | | | || (_| |  ) ) ) )
  '  |____| .__|_| |_|_| |_\\__, | / / / /
 =========|_|==============|___/=/_/_/_/
 :: Spring Boot ::                (v2.6.6)

2022-04-21 13:47:39.176  INFO 15784 --- [           main] com.example.demo.DemoApplication         : Starting DemoApplication using Java 1.8.0_291 on DESKTOP-K08MNSB with PID 15784 (E:\\working\\demo\\demo\\target\\classes started by 邓沙利文 in E:\\java\\flex-server)
2022-04-21 13:47:39.179  INFO 15784 --- [           main] com.example.demo.DemoApplication         : No active profile set, falling back to 1 default profile: "default"
2022-04-21 13:47:39.480  INFO 15784 --- [           main] com.example.demo.DemoApplication         : Started DemoApplication in 0.509 seconds (JVM running for 0.9)
hello 邓二洋~
hello 邓帅~

Process finished with exit code 0

相关资料

• 本文已被收录进《程序员成长笔记》 ，笔者JustryDeng