1.RabbitAutoConfiguration類解析
image.png
1.1類注解解析
@Configuration
@ConditionalOnClass({ RabbitTemplate.class, Channel.class })
@EnableConfigurationProperties(RabbitProperties.class)
@Import(RabbitAnnotationDrivenConfiguration.class)
@ConditionalOnClass判斷類路徑中存在RabbitTemplate類和Channel類
@EnableConfigurationProperties導入rabbitmq的配置類，具體就是在application.properties中的配置
@Import導入RabbitAnnotationDrivenConfiguration類，見下文2.0的分析
1.2構造rabbitmq的連接工廠類
image.png
類圖如下：
image.png
這里比較簡單，就是判斷bean不存在，就注冊為bean到spring容器中
1.3 構造rabbitmqTemplate
@Configuration
//`@Import`導入`RabbitConnectionFactoryCreator`類，也就是1.2中注入的bean
@Import(RabbitConnectionFactoryCreator.class)
protected static class RabbitTemplateConfiguration {
……
@Bean
@ConditionalOnSingleCandidate(ConnectionFactory.class)
@ConditionalOnMissingBean
public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
PropertyMapper map = PropertyMapper.get();
//設置連接工廠
RabbitTemplate template = new RabbitTemplate(connectionFactory);
//設置消息轉換器
MessageConverter messageConverter = this.messageConverter.getIfUnique();
if (messageConverter != null) {
template.setMessageConverter(messageConverter);
}
template.setMandatory(determineMandatoryFlag());
RabbitProperties.Template properties = this.properties.getTemplate();
if (properties.getRetry().isEnabled()) {
//設置重試策略
template.setRetryTemplate(createRetryTemplate(properties.getRetry()));
}
map.from(properties::getReceiveTimeout).whenNonNull().as(Duration::toMillis)
.to(template::setReceiveTimeout);
map.from(properties::getReplyTimeout).whenNonNull().as(Duration::toMillis)
.to(template::setReplyTimeout);
map.from(properties::getExchange).to(template::setExchange);
map.from(properties::getRoutingKey).to(template::setRoutingKey);
return template;
}
1.3.1 注入連接工廠
1.3.2 注入消息轉換器
MessageConverter messageConverter = this.messageConverter.getIfUnique();
if (messageConverter != null) {
template.setMessageConverter(messageConverter);
}
會判斷當前環境中是否存在自定義消息轉換器的實現
自定義實現例子為：
@Configuration
public class MyAMQPConfig {
@Bean
public MessageConverter messageConverter(){
return new Jackson2JsonMessageConverter();
}
}
public class RabbitTemplate extends RabbitAccessor implements BeanFactoryAware, RabbitOperations, MessageListener,
ListenerContainerAware, PublisherCallbackChannel.Listener, Lifecycle, BeanNameAware {
……
// 默認的消息轉換器
private volatile MessageConverter messageConverter = new SimpleMessageConverter();
……
1.3.3 重試策略的設置
- RabbitAnnotationDrivenConfiguration類
@Configuration
@ConditionalOnClass(EnableRabbit.class)
class RabbitAnnotationDrivenConfiguration {
……
//注冊默認的容器工廠類
@Bean(name = "rabbitListenerContainerFactory")
@ConditionalOnMissingBean(name = "rabbitListenerContainerFactory")
@ConditionalOnProperty(prefix = "spring.rabbitmq.listener", name = "type", havingValue = "simple", matchIfMissing = true)
public SimpleRabbitListenerContainerFactory simpleRabbitListenerContainerFactory(
SimpleRabbitListenerContainerFactoryConfigurer configurer,
ConnectionFactory connectionFactory) {
SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
configurer.configure(factory, connectionFactory);
return factory;
}
……
@Configuration
//進行bean的導入
@EnableRabbit
@ConditionalOnMissingBean(name = RabbitListenerConfigUtils.RABBIT_LISTENER_ANNOTATION_PROCESSOR_BEAN_NAME)
protected static class EnableRabbitConfiguration {
}
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Import(RabbitBootstrapConfiguration.class)
public @interface EnableRabbit {
}
@Configuration
public class RabbitBootstrapConfiguration {
//注冊bean名字為：org.springframework.amqp.rabbit.config.internalRabbitListenerAnnotationProcessor
//這個類就極為關鍵，真正注冊@RabbitListener的邏輯就在這里
@Bean(name = RabbitListenerConfigUtils.RABBIT_LISTENER_ANNOTATION_PROCESSOR_BEAN_NAME)
@Role(BeanDefinition.ROLE_INFRASTRUCTURE)
public RabbitListenerAnnotationBeanPostProcessor rabbitListenerAnnotationProcessor() {
return new RabbitListenerAnnotationBeanPostProcessor();
}
//注冊bean名字為：org.springframework.amqp.rabbit.config.internalRabbitListenerEndpointRegistry
@Bean(name = RabbitListenerConfigUtils.RABBIT_LISTENER_ENDPOINT_REGISTRY_BEAN_NAME)
public RabbitListenerEndpointRegistry defaultRabbitListenerEndpointRegistry() {
return new RabbitListenerEndpointRegistry();
}
}
- RabbitListenerAnnotationBeanPostProcessor類
public class RabbitListenerAnnotationBeanPostProcessor
implements BeanPostProcessor, Ordered, BeanFactoryAware, BeanClassLoaderAware, EnvironmentAware,
SmartInitializingSingleton {
……
//實現接口的執行順序是：
setBeanClassLoader ->
setBeanFactory ->
setEnvironment ->
postProcessAfterInitialization(每一個spring容器中的bean都會執行這個方法) ->
afterSingletonsInstantiated
setBeanClassLoader .png
setBeanFactory .png
setEnvironment .png
postProcessAfterInitialization .png
afterSingletonsInstantiated.png
@Override
public void afterSingletonsInstantiated() {
// private final RabbitListenerEndpointRegistrar registrar = new RabbitListenerEndpointRegistrar();
this.registrar.setBeanFactory(this.beanFactory);
if (this.beanFactory instanceof ListableBeanFactory) {
Map<String, RabbitListenerConfigurer> instances =
((ListableBeanFactory) this.beanFactory).getBeansOfType(RabbitListenerConfigurer.class);
for (RabbitListenerConfigurer configurer : instances.values()) {
configurer.configureRabbitListeners(this.registrar);
}
}
if (this.registrar.getEndpointRegistry() == null) {
if (this.endpointRegistry == null) {
Assert.state(this.beanFactory != null,
"BeanFactory must be set to find endpoint registry by bean name");
this.endpointRegistry = this.beanFactory.getBean(
RabbitListenerConfigUtils.RABBIT_LISTENER_ENDPOINT_REGISTRY_BEAN_NAME,
RabbitListenerEndpointRegistry.class);
}
this.registrar.setEndpointRegistry(this.endpointRegistry);
}
if (this.containerFactoryBeanName != null) {
this.registrar.setContainerFactoryBeanName(this.containerFactoryBeanName);
}
// Set the custom handler method factory once resolved by the configurer
MessageHandlerMethodFactory handlerMethodFactory = this.registrar.getMessageHandlerMethodFactory();
if (handlerMethodFactory != null) {
this.messageHandlerMethodFactory.setMessageHandlerMethodFactory(handlerMethodFactory);
}
// Actually register all listeners
//關鍵方法，
this.registrar.afterPropertiesSet();
// clear the cache - prototype beans will be re-cached.
this.typeCache.clear();
}
@Override
public void afterPropertiesSet() {
registerAllEndpoints();
}
protected void registerAllEndpoints() {
synchronized (this.endpointDescriptors) {
//遍歷每一個@RabbitListener
for (AmqpListenerEndpointDescriptor descriptor : this.endpointDescriptors) {
this.endpointRegistry.registerListenerContainer(
descriptor.endpoint, resolveContainerFactory(descriptor));
}
this.startImmediately = true; // trigger immediate startup
}
}
public void registerListenerContainer(RabbitListenerEndpoint endpoint, RabbitListenerContainerFactory<?> factory,
boolean startImmediately) {
Assert.notNull(endpoint, "Endpoint must not be null");
Assert.notNull(factory, "Factory must not be null");
String id = endpoint.getId();
Assert.hasText(id, "Endpoint id must not be empty");
synchronized (this.listenerContainers) {
Assert.state(!this.listenerContainers.containsKey(id),
"Another endpoint is already registered with id '" + id + "'");
//這里進行監聽器容器的配置設定，以及初始化，有點不懂了，后續再來擼
MessageListenerContainer container = createListenerContainer(endpoint, factory);
……
protected MessageListenerContainer createListenerContainer(RabbitListenerEndpoint endpoint,
RabbitListenerContainerFactory<?> factory) {
//工廠類創建MessageListenerContainer
//見SimpleRabbitListenerContainerFactory類的createContainerInstance方法，將SimpleRabbitListenerContainerFactory工廠類與SimpleRabbitListenerContainer實體類關聯起來
//@Override
//protected SimpleMessageListenerContainer createContainerInstance() {
// return new SimpleMessageListenerContainer();
//}
MessageListenerContainer listenerContainer = factory.createListenerContainer(endpoint);
if (listenerContainer instanceof InitializingBean) {
try {
((InitializingBean) listenerContainer).afterPropertiesSet();
}
catch (Exception ex) {
throw new BeanInitializationException("Failed to initialize message listener container", ex);
}
}
……
工廠類與實體類的類圖
SimpleRabbitListenerContainerFactory.png
SimpleRabbitListenerContainer.png
