Extract RabbitTemplate auto-configuration in a configurer
This commit movers the auto-configuration of RabbitTemplate to a dedicated class that can be reused to create additional template with similar settings. CLoses gh-19440pull/19476/head
parent
015714c1d6
commit
bb5e09882a
@ -0,0 +1,102 @@
|
||||
/*
|
||||
* Copyright 2012-2019 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot.autoconfigure.amqp;
|
||||
|
||||
import java.time.Duration;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
|
||||
import org.springframework.amqp.rabbit.core.RabbitTemplate;
|
||||
import org.springframework.amqp.support.converter.MessageConverter;
|
||||
import org.springframework.boot.context.properties.PropertyMapper;
|
||||
|
||||
/**
|
||||
* Configure {@link RabbitTemplate} with sensible defaults.
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @since 2.3.0
|
||||
*/
|
||||
public class RabbitTemplateConfigurer {
|
||||
|
||||
private MessageConverter messageConverter;
|
||||
|
||||
private List<RabbitRetryTemplateCustomizer> retryTemplateCustomizers;
|
||||
|
||||
private RabbitProperties rabbitProperties;
|
||||
|
||||
/**
|
||||
* Set the {@link MessageConverter} to use or {@code null} if the out-of-the-box
|
||||
* converter should be used.
|
||||
* @param messageConverter the {@link MessageConverter}
|
||||
*/
|
||||
protected void setMessageConverter(MessageConverter messageConverter) {
|
||||
this.messageConverter = messageConverter;
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the {@link RabbitRetryTemplateCustomizer} instances to use.
|
||||
* @param retryTemplateCustomizers the retry template customizers
|
||||
*/
|
||||
protected void setRetryTemplateCustomizers(List<RabbitRetryTemplateCustomizer> retryTemplateCustomizers) {
|
||||
this.retryTemplateCustomizers = retryTemplateCustomizers;
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the {@link RabbitProperties} to use.
|
||||
* @param rabbitProperties the {@link RabbitProperties}
|
||||
*/
|
||||
protected void setRabbitProperties(RabbitProperties rabbitProperties) {
|
||||
this.rabbitProperties = rabbitProperties;
|
||||
}
|
||||
|
||||
protected final RabbitProperties getRabbitProperties() {
|
||||
return this.rabbitProperties;
|
||||
}
|
||||
|
||||
/**
|
||||
* Configure the specified {@link RabbitTemplate}. The template can be further tuned
|
||||
* and default settings can be overridden.
|
||||
* @param template the {@link RabbitTemplate} instance to configure
|
||||
* @param connectionFactory the {@link ConnectionFactory} to use
|
||||
*/
|
||||
public void configure(RabbitTemplate template, ConnectionFactory connectionFactory) {
|
||||
PropertyMapper map = PropertyMapper.get();
|
||||
template.setConnectionFactory(connectionFactory);
|
||||
if (this.messageConverter != null) {
|
||||
template.setMessageConverter(this.messageConverter);
|
||||
}
|
||||
template.setMandatory(determineMandatoryFlag());
|
||||
RabbitProperties.Template templateProperties = this.rabbitProperties.getTemplate();
|
||||
if (templateProperties.getRetry().isEnabled()) {
|
||||
template.setRetryTemplate(new RetryTemplateFactory(this.retryTemplateCustomizers)
|
||||
.createRetryTemplate(templateProperties.getRetry(), RabbitRetryTemplateCustomizer.Target.SENDER));
|
||||
}
|
||||
map.from(templateProperties::getReceiveTimeout).whenNonNull().as(Duration::toMillis)
|
||||
.to(template::setReceiveTimeout);
|
||||
map.from(templateProperties::getReplyTimeout).whenNonNull().as(Duration::toMillis)
|
||||
.to(template::setReplyTimeout);
|
||||
map.from(templateProperties::getExchange).to(template::setExchange);
|
||||
map.from(templateProperties::getRoutingKey).to(template::setRoutingKey);
|
||||
map.from(templateProperties::getDefaultReceiveQueue).whenNonNull().to(template::setDefaultReceiveQueue);
|
||||
}
|
||||
|
||||
private boolean determineMandatoryFlag() {
|
||||
Boolean mandatory = this.rabbitProperties.getTemplate().getMandatory();
|
||||
return (mandatory != null) ? mandatory : this.rabbitProperties.isPublisherReturns();
|
||||
}
|
||||
|
||||
}
|
Loading…
Reference in New Issue