使用Gson替代RabbitMq的序列化，自定义RabbitMq序列化方式

Posted 2022-11-18 donotshow

使用Gson替代RabbitMq的序列化，自定义RabbitMq序列化方式

RabbitMq默认使用了java的序列化，可读性差，以下是RabbitTamplate默认的序列化工具类实现。




所以RabbitTamplate提供了自定义序列化方式的入口setMessageConverter，同时spring.amqp提供了几种序列化方式可供选择，基本可以满足大众的需求。

在以上序列化都不满足需求的情况下可以自定义序列化和反序列化方式。
1.继承AbstractMessageConverter或者实现MessageConvert，自定义序列化工具，我这边选择了继承AbstractMessageConverter，重写createMessage和fromMessage方法。

@Component
public class GsonMqMessageConverter extends AbstractMessageConverter 
    private static ClassMapper classMapper = new RabbitMqClassMapper();

    @Override
    protected Message createMessage(Object o, MessageProperties messageProperties) 
        String jsonString = HtGson.gson().toJson(o);
        byte[] bytes = jsonString.getBytes(StandardCharsets.UTF_8);
        messageProperties.setContentType(MessageProperties.CONTENT_TYPE_JSON);
        messageProperties.setContentEncoding("UTF8");
        messageProperties.setContentLength(bytes.length);
        classMapper.fromClass(o.getClass(), messageProperties);
        return new Message(bytes, messageProperties);
    

    @Override
    public Object fromMessage(Message message) throws MessageConversionException 
        MessageProperties properties = message.getMessageProperties();
        if (properties != null) 
            String contentType = properties.getContentType();
            if (contentType != null && contentType.contains("json")) 
                String encoding = properties.getContentEncoding();
                if (encoding == null) 
                    encoding = "UTF8";
                
                try 
                    Class<?> targetClass = classMapper.toClass(
                            message.getMessageProperties());
                    String contentAsString = new String(message.getBody(), encoding);
                    return HtGson.gson().fromJson(contentAsString, targetClass);
                 catch (IOException e) 
                    throw new MessageConversionException(
                            "Failed to convert Message content", e);
                
            
        
        return message.getBody();
    

2.Provider端通过RabbitTamplate指定我们自定义的序列化工具。

@Bean(name = "rabbitTemplate")
    @Primary
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) 
        RabbitTemplate rabbitTemplate = new RabbitTemplate(connectionFactory);
        rabbitTemplate.setMandatory(true);
        rabbitTemplate.setMessageConverter(new GsonMqMessageConverter());//指定我们自定义的序列化工具
        return rabbitTemplate;
    

3.Consumer指定反序列化
通过SimpleRabbitListenerContainerFactory全局指定。

@Bean(value = "rabbitListenerContainerFactory")//bean的name必须是rabbitListenerContainerFactory
    @Primary
    public SimpleRabbitListenerContainerFactory simpleRabbitListenerContainerFactory(ConnectionFactory connectionFactory,
                                                                                     @Qualifier("rabbitRetryTemplate") RetryTemplate rabbitRetryTemplate) 
        SimpleRabbitListenerContainerFactory factory = new SimpleRabbitListenerContainerFactory();
        factory.setConnectionFactory(connectionFactory);
        factory.setMessageConverter(new GsonMqMessageConverter());
        return factory;
    

通过@RabbitListener单独指定

@RabbitListener(queues = HtRabbitMqQueue.RECEIVE_LOCATION, messageConverter = "gsonMqMessageConverter")

以上就可以根据需求自定义RabbitMq的序列化和反序列化了，我这边主要使用到Gson，后续会讲到对于Gson的一些使用。