如何写好一个 Spring 组件

本文为原创文章。欢迎任何形式的转载，但请务必注明出处冷冷https://lltx.github.io。html

背景

Spring 框架提供了许多接口，可使用这些接口来定制化 bean ，而非简单的 getter/setter 或者构造器注入。细翻 Spring Cloud Netflix、Spring Cloud Alibaba 等这些构建在 Spring Framework 的成熟框架源码，你会发现大量的扩展 bean 例如java

Eureka 健康检查

package org.springframework.cloud.netflix.eureka;

public class EurekaHealthCheckHandler implements InitializingBean {}
复制代码

Seata Feign 配置

package com.alibaba.cloud.seata.feign;

public class SeataContextBeanPostProcessor implements BeanPostProcessor {}
复制代码

代码示例

DemoBean

@Slf4j
public class DemoBean implements InitializingBean {

	public DemoBean() {
		log.info("--> instantiate ");
	}

	@PostConstruct
	public void postConstruct() {
		log.info("--> @PostConstruct ");
	}

	@Override
	public void afterPropertiesSet() throws Exception {
		log.info("--> InitializingBean.afterPropertiesSet ");
	}

	public void initMethod() {
		log.info("--> custom initMehotd");
	}
}
复制代码

DemoBeanPostProcessor

@Configuration
public class DemoBeanPostProcessor implements BeanPostProcessor {
	@Override
	public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
		if ("demoBean".equals(beanName)){
			log.info("--> BeanPostProcessor.postProcessBeforeInitialization ");
		}
		return bean;
	}

	@Override
	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
		if ("demoBean".equals(beanName)){
			log.info("--> BeanPostProcessor.postProcessAfterInitialization ");
		}
		return bean;
	}
}
复制代码

DemoConfig

@Configuration
public class DemoConfig {

	@Bean(initMethod = "initMethod")
	public DemoBean demoBean() {
		return new DemoBean();
	}
}
复制代码

运行输出日志

整个 bean 的建立过程日志输出以下和文首图片横线以上 bean 建立周期一致

DemoBean             : --> instantiate
DemoBeanPostProcessor: --> BeanPostProcessor.postProcessBeforeInitialization
DemoBean             : --> @PostConstruct
DemoBean             : --> InitializingBean.afterPropertiesSet
DemoBean             : --> custom initMehotd
DemoBeanPostProcessor: --> BeanPostProcessor.postProcessAfterInitialization
复制代码

执行过程核心源码

AbstractAutowireCapableBeanFactory.initializeBean

protected Object initializeBean(final String beanName, final Object bean, @Nullable RootBeanDefinition mbd) {

  // 执行BeanPostProcessor.postProcessBeforeInitialization
  Object wrappedBean = wrappedBean = applyBeanPostProcessorsBeforeInitialization(wrappedBean, beanName);
      ...
      // 执行用户自定义初始化and JSR 250 定义的方法
  invokeInitMethods(beanName, wrappedBean, mbd);
      ...
  // 执行执行BeanPostProcessor.postProcessAfterInitialization
  wrappedBean = applyBeanPostProcessorsAfterInitialization(wrappedBean, beanName);

  return wrappedBean;
}
复制代码

下文就详细说明一下每一个 bean 的扩展周期的最佳使用场景

BeanPostProcessor

BeanPostProcessor 是一个能够自定义实现回调方法接口，来实现本身的实例化逻辑、依赖解决逻辑等，若是想要在 Spring 完成对象实例化、配置、初始化以后实现本身的业务逻辑，能够经过扩展实现一个或多个 BeanPostProcessor 处理。git

多用于适配器模式，能够在实现同一接口 bean 建立先后进行包装转换

// seata 上下文转换，将其余类型 wrap 成 SeataFeignContext
public class SeataContextBeanPostProcessor implements BeanPostProcessor {

	@Override
	public Object postProcessBeforeInitialization(Object bean, String beanName){
		if (bean instanceof FeignContext && !(bean instanceof SeataFeignContext)) {
			return new SeataFeignContext(getSeataFeignObjectWrapper(),
					(FeignContext) bean);
		}
		return bean;
	}
}
复制代码

自定义注解查找扩展

net.dreamlu.mica.redisson.stream.RStreamListenerDetector 查找自定义 @RStreamListener 实现 基于 Redisson 的 pub/sub

public class RStreamListenerDetector implements BeanPostProcessor {

	@Override
	public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
		Class<?> userClass = ClassUtils.getUserClass(bean);
		ReflectionUtils.doWithMethods(userClass, method -> {
			RStreamListener listener = AnnotationUtils.findAnnotation(method, RStreamListener.class);
			.... do something
		}, ReflectionUtils.USER_DECLARED_METHODS);
		return bean;
	}
}

复制代码

PostConstruct

JavaEE5 引入了@PostConstruct 做用于 Servlet 生命周期的注解，实现 Bean 初始化以前的自定义操做。github

只能有一个非静态方法使用此注解
被注解的方法不能有返回值和方法参数
被注解的方法不得抛出异常

这里须要注意的这个注解不是 Spring 定义的，而是属于 JavaEE JSR-250 规范定义的注解,当你在使用 Java11 的时候要手动引入相关 jar（由于 Java11 移除了）redis

<dependency>
    <groupId>javax.annotation</groupId>
    <artifactId>javax.annotation-api</artifactId>
</dependency>
复制代码

使用场景：在以前的版本，咱们能够在启动后经过 @PostConstruct 注解的方法执行初始化数据。但因为 Java 高版本已经移除相关 API ，咱们不推荐使用此注解，能够经过 Spring 相关 Event 回调事件处理spring

@PostConstruct 注解的方法在项目启动的时候执行这个方法，也能够理解为在 spring 容器启动的时候执行，可做为一些数据的常规化加载，好比数据字典之类的。api

InitializingBean

InitializingBean 接口方法会在容器初始化（getter/setter/构造器）完成 bean 的属性注入后执行。bash

应用场景：动态修改容器注入的 Bean 参数markdown

正经常使用户配置参数注入到 bean

security:
  oauth2:
      ignore-urls:
        - '/ws/**'

@ConfigurationProperties(prefix = "security.oauth2")
public class PermitAllUrlProperties {
	@Getter
	@Setter
	private List<String> ignoreUrls = new ArrayList<>();
}
复制代码

咱们发现此时用户配置并不完整，还有一些通用不须要用户维护，可经过实现 InitializingBean 接口回调扩展

@ConfigurationProperties(prefix = "security.oauth2.ignore")
public class PermitAllUrlProperties implements InitializingBean {

	@Getter
	@Setter
	private List<String> urls = new ArrayList<>();

	@Override
	public void afterPropertiesSet() {
		urls.add("/common/*");
	}
}
复制代码

initMethod

上文 @PostConstruct 已经不推荐你们使用，可使用 Bean(initMethod = 'initMehotd') 替代，相关的限制如上。app

@Bean(initMethod = "initMethod")
public DemoBean demoBean() {
  return new DemoBean();
}

public void initMethod() {
  log.info("--> custom initMehotd");
}
复制代码