org.springframework.cloud.stream.config.MessageSourceCustomizer Java Examples

@Bean
RabbitMessageChannelBinder rabbitMessageChannelBinder(
		@Nullable ListenerContainerCustomizer<AbstractMessageListenerContainer> listenerContainerCustomizer,
		@Nullable MessageSourceCustomizer<AmqpMessageSource> sourceCustomizer,
		@Nullable ProducerMessageHandlerCustomizer<AmqpOutboundEndpoint> producerMessageHandlerCustomizer,
		@Nullable ConsumerEndpointCustomizer<AmqpInboundChannelAdapter> consumerCustomizer,
		List<DeclarableCustomizer> declarableCustomizers) {

	RabbitMessageChannelBinder binder = new RabbitMessageChannelBinder(
			this.rabbitConnectionFactory, this.rabbitProperties,
			provisioningProvider(declarableCustomizers), listenerContainerCustomizer, sourceCustomizer);
	binder.setAdminAddresses(
			this.rabbitBinderConfigurationProperties.getAdminAddresses());
	binder.setCompressingPostProcessor(gZipPostProcessor());
	binder.setDecompressingPostProcessor(deCompressingPostProcessor());
	binder.setNodes(this.rabbitBinderConfigurationProperties.getNodes());
	binder.setExtendedBindingProperties(this.rabbitExtendedBindingProperties);
	binder.setProducerMessageHandlerCustomizer(producerMessageHandlerCustomizer);
	binder.setConsumerEndpointCustomizer(consumerCustomizer);
	return binder;
}
 

public KafkaMessageChannelBinder(
		KafkaBinderConfigurationProperties configurationProperties,
		KafkaTopicProvisioner provisioningProvider,
		ListenerContainerCustomizer<AbstractMessageListenerContainer<?, ?>> containerCustomizer,
		MessageSourceCustomizer<KafkaMessageSource<?, ?>> sourceCustomizer,
		KafkaBindingRebalanceListener rebalanceListener,
		DlqPartitionFunction dlqPartitionFunction) {

	super(headersToMap(configurationProperties), provisioningProvider,
			containerCustomizer, sourceCustomizer);
	this.configurationProperties = configurationProperties;
	String txId = configurationProperties.getTransaction().getTransactionIdPrefix();
	if (StringUtils.hasText(txId)) {
		this.transactionManager = new KafkaTransactionManager<>(getProducerFactory(
				txId, new ExtendedProducerProperties<>(configurationProperties
						.getTransaction().getProducer().getExtension()), txId + ".producer"));
		this.transactionTemplate = new TransactionTemplate(this.transactionManager);
	}
	else {
		this.transactionManager = null;
		this.transactionTemplate = null;
	}
	this.rebalanceListener = rebalanceListener;
	this.dlqPartitionFunction = dlqPartitionFunction != null
			? dlqPartitionFunction
			: null;
}
 

public RabbitMessageChannelBinder(ConnectionFactory connectionFactory,
		RabbitProperties rabbitProperties,
		RabbitExchangeQueueProvisioner provisioningProvider,
		ListenerContainerCustomizer<AbstractMessageListenerContainer> containerCustomizer,
		MessageSourceCustomizer<AmqpMessageSource> sourceCustomizer) {

	super(new String[0], provisioningProvider, containerCustomizer, sourceCustomizer);
	Assert.notNull(connectionFactory, "connectionFactory must not be null");
	Assert.notNull(rabbitProperties, "rabbitProperties must not be null");
	this.connectionFactory = connectionFactory;
	this.rabbitProperties = rabbitProperties;
}
 

public AbstractMessageChannelBinder(String[] headersToEmbed, PP provisioningProvider,
		@Nullable ListenerContainerCustomizer<?> containerCustomizer,
		@Nullable MessageSourceCustomizer<?> sourceCustomizer) {

	SimpleModule module = new SimpleModule();
	module.addSerializer(Expression.class, new ExpressionSerializer(Expression.class));
	objectMapper.registerModule(module);

	this.headersToEmbed = headersToEmbed == null ? new String[0] : headersToEmbed;
	this.provisioningProvider = provisioningProvider;
	this.containerCustomizer = containerCustomizer == null ? (c, q, g) -> {
	} : containerCustomizer;
	this.sourceCustomizer = sourceCustomizer == null ? (s, q, g) -> {
	} : sourceCustomizer;
}
 

@Bean
public MessageSourceCustomizer<KafkaMessageSource<?, ?>> sourceCustomizer() {
	return (s, q, g) -> s.setBeanName("setByCustomizer:" + q);
}
 

@Bean
public MessageSourceCustomizer<AmqpMessageSource> sourceCustomizer() {
	return (s, q, g) -> s.setBeanName("setByCustomizer:" + g);
}
 

@SuppressWarnings("unchecked")
protected <S> MessageSourceCustomizer<S> getMessageSourceCustomizer() {
	return (MessageSourceCustomizer<S>) this.sourceCustomizer;
}