关于SpringBoot 中 SecurityUtils在 RabbitMq 中无法获取的问题!

1.自定义CorrelationData

@Getter
@Setter
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class MyCorrelationData extends CorrelationData {

    /**
     * 内容
     * @author LXC
     * @date 2023/2/4 15:47
     */
    private String body;

    /**
     * 交换机名
     * @author LXC
     * @date 2023/2/4 15:47
     */
    private String exchangeName;

    /**
     * 路由钥匙
     * @author LXC
     * @date 2023/2/4 15:47
     */
    private String routingKey;

}

2.自定义RabbitTemplate

public class MyRabbitTemplate extends RabbitTemplate{

    public MyRabbitTemplate(ConnectionFactory connectionFactory) {
        super(connectionFactory);
    }


    public void convertAndSend(MyCorrelationData myCorrelationData){

        String messageId = UUID.randomUUID().toString().replace("-","");

        MessagePostProcessor messagePostProcessor = message -> {
            MessageProperties messageProperties = new MessageProperties();
            messageProperties.setMessageId(messageId);
            messageProperties.setReceivedExchange(myCorrelationData.getExchangeName());
            messageProperties.setReceivedRoutingKey(myCorrelationData.getRoutingKey());
            return new Message(myCorrelationData.getBody().getBytes(),messageProperties);
        };

        myCorrelationData.setId(messageId);

        convertAndSend(myCorrelationData.getExchangeName(),myCorrelationData.getRoutingKey(),
                myCorrelationData.getBody(),messagePostProcessor, myCorrelationData);
    }

}

3.自定义rabbitmqconfig


@Configuration
@Slf4j
public class RabbitConfig {


    @Bean(name = "myRabbitTemplate")
    @Primary
    public MyRabbitTemplate rabbitTemplate(CachingConnectionFactory connectionFactory) {
        MyRabbitTemplate rabbitTemplate = new MyRabbitTemplate(connectionFactory);

        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setUsePublisherConnection(true);
        rabbitTemplate.setUserCorrelationId(true);

        return rabbitTemplate;
    }


}

4.配置请求头的问题在关于SpringBoot 中 SecurityUtils在 Thread 中无法获取的问题!_活在梦里鑫的博客-CSDN博客

5.拦截AOP  RabbitListener


@Aspect
@Component
public class RabbitMQInterceptor {

    @Around("@annotation(org.springframework.amqp.rabbit.annotation.RabbitListener)")
    public Object around(ProceedingJoinPoint point) throws Throwable{

        Object[] args = point.getArgs();

        if(ObjectUtils.isEmpty(args)){
            return point.proceed();
        }

        for (Object obj:args){

            if(!(obj instanceof Message)){
                continue;
            }

            Message message = (Message) obj;

            Map<String, Object> headers = message.getMessageProperties().getHeaders();

            for (String key:headers.keySet()){

                if(!(headers.get(key) instanceof String)){
                    continue;
                }

                ThreadLocalUtils.addHeader(key, String.valueOf(headers.get(key)));
            }

            break;
        }

        return point.proceed();
    }
}

6.完美解决!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值