自定义BeanPostProcessor实现自动注入标注了特定注解的Bean

定义注解

@Target({ElementType.FIELD, ElementType.PARAMETER, ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface MyAnno {
}

定义一个配置类

@Configuration
public class RestConfig {@MyAnno@Beanpublic PayDTO payDTO(){PayDTO payDTO = new PayDTO();payDTO.setPayNo("1");return payDTO;}@Beanpublic PayDTO payDTO1(){PayDTO payDTO = new PayDTO();payDTO.setPayNo("2");return payDTO;}}

这个配置类返回了两个PayDTO 类型的对象。

定义使用类

@Component
public class MyConfigTest implements InitializingBean {@MyAnno@Autowired(required = false)private List<PayDTO> payDTOList = Collections.emptyList();@Overridepublic void afterPropertiesSet() throws Exception {System.out.println("payDTO = " + payDTOList);}
}

这里想只从容器中获取标注了@MyAnno注解的PayDTO对象，也就是说应该是只能获取到一个。

定义BeanPostProcessor

@Component
public class MyAnnoPostProcesser implements BeanPostProcessor {@Autowiredprivate ApplicationContext applicationContext;@Overridepublic Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {//这里只考虑对@Autowired作用于属性上Field[] declaredFields = bean.getClass().getDeclaredFields();if (bean instanceof MyConfigTest){System.out.println("MyConfigTest");}for (Field field : declaredFields) {//判断当前属性是否标记了@MyAnno注解MyAnno annotation = field.getAnnotation(MyAnno.class);Autowired autowired = field.getAnnotation(Autowired.class);if (annotation != null && autowired != null) {field.setAccessible(true);//获取字段的类型Type genericType = field.getGenericType();if (genericType instanceof ParameterizedType) {//判断属性类型是否是一个参数化类型，例如 List<String> 这里的String就是泛型参数ParameterizedType parameterizedType = (ParameterizedType) genericType;Type rawType = parameterizedType.getRawType();//泛型原生类型是List类型if (rawType == List.class) {//获取参数化类型列表，每一个参数化列表可能又是一个泛型类型Type[] actualTypeArguments = parameterizedType.getActualTypeArguments();Type argument = actualTypeArguments[0];if (argument instanceof Class) {Class<?> clazz = (Class<?>) argument;//从容器中获取对应此参数化类型并且标注了MyAnno注解的BeanMap<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(MyAnno.class);if (beansWithAnnotation.size() > 0) {List<Object> beanList = beansWithAnnotation.entrySet().stream().filter(entry -> {Object value = entry.getValue();return value.getClass() == clazz;}).map(item -> item.getValue()).collect(Collectors.toList());//进行属性赋值try {field.set(bean, beanList);} catch (IllegalAccessException e) {throw new RuntimeException(e);}}}}}}}return bean;}
}

BeanPostProcessor实现原理：

获取Bean示例，并拿到这个Bean对象中声明的属性。
判断属性是否同时标注了MyAnno和Autowired注解。
获取属性的类型
属性的类型返回值有多种情况，可能是Class、ParameterizedType、GenericType等
如果是一个具体的类型则返回Class，如果是一个泛型类型，例如List，则返回ParameterizedType参数化类型，参数化类型的意思就是使用了泛型类型，并指定了泛型参数，这里的泛型参数就是PayDTO。
parameterizedType.getRawType(); 获取泛型类型的原生类型，也就是List.class
Type[] actualTypeArguments = parameterizedType.getActualTypeArguments(); 获取泛型类型的参数类型，例如List类的声明是
List 这里的T就是泛型参数，而parameterizedType.getActualTypeArguments();返回的是一个数组，因为可能不止一个泛型类型，例如Map<K,V> 。
获取参数化类型的第一个元素，其实这里返回的就是PayDTO.class对象。
从容器中获取所有标注了MyAnno注解的Bean对象，并且遍历出是PayDTO类型的。
通过field.set(bean, beanList);实现属性值注入。
完成基于BeanPostProcessor实现的自定义注解Bean对象注入。