Marmelator/persistence-service
1
0
Fork 0
Code Issues Pull Requests 17 Actions Packages Projects Releases 1.1k Wiki Activity

RED-4512: Fixed wrong bean for setAfterReceivePostProcessors message interception

Browse Source
This commit is contained in:
deiflaender 2022-09-06 11:03:59 +02:00
parent 5df851ed06
commit a24ba6d601
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
persistence-service-v1/persistence-service-server-v1/src/main/java/com/iqser/red/service/peristence/v1/server/multitenancy/MultiTenancyMessagingConfiguration.java
View File

@ -3,6 +3,7 @@ package com.iqser.red.service.peristence.v1.server.multitenancy;
import static com.iqser.red.service.peristence.v1.server.multitenancy.TenantInterceptor.DEFAULT_TENANT; import static com.iqser.red.service.peristence.v1.server.multitenancy.TenantInterceptor.DEFAULT_TENANT;
import static com.iqser.red.service.peristence.v1.server.multitenancy.TenantInterceptor.TENANT_HEADER_NAME; import static com.iqser.red.service.peristence.v1.server.multitenancy.TenantInterceptor.TENANT_HEADER_NAME;
import org.springframework.amqp.rabbit.config.AbstractRabbitListenerContainerFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate; import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.BeansException; import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor; import org.springframework.beans.factory.config.BeanPostProcessor;
@ -24,14 +25,14 @@ public class MultiTenancyMessagingConfiguration {
if (bean instanceof RabbitTemplate) { if (bean instanceof RabbitTemplate) {
RabbitTemplate rabbitTemplate = (RabbitTemplate) bean; ((RabbitTemplate) bean).setBeforePublishPostProcessors(m -> {
rabbitTemplate.setBeforePublishPostProcessors(m -> {
m.getMessageProperties().setHeader(TENANT_HEADER_NAME, TenantContext.getTenantId()); m.getMessageProperties().setHeader(TENANT_HEADER_NAME, TenantContext.getTenantId());
return m; return m;
}); });
rabbitTemplate.addAfterReceivePostProcessors(m -> { } else if (bean instanceof AbstractRabbitListenerContainerFactory) {
((AbstractRabbitListenerContainerFactory<?>) bean).setAfterReceivePostProcessors(m -> {
String tenant = m.getMessageProperties().getHeader(TENANT_HEADER_NAME); String tenant = m.getMessageProperties().getHeader(TENANT_HEADER_NAME);
if (tenant != null) { if (tenant != null) {
@ -42,7 +43,6 @@ public class MultiTenancyMessagingConfiguration {
} }
return m; return m;
}); });
} }
return bean; return bean;
} }