- 定义注解
@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对象注入。
- 踩坑
MyConfigTest对象标注的@Component，如果是标注@Configuration注解，此时Spring会生成一个代理对象，通过代理对象拿不到原始对象的相关属性了。
