使用Kryo替换spring amqp的Java序列化

spring amqp的原生并没有对Kryo加以支持,Kryo的优点就不多说了。

git地址:https://github.com/addozhang/spring-kryo-messaeg-converter

public class KryoMessageConverter extends AbstractMessageConverter {

    public static final String CONTENT_TYPE = "application/x-kryo";
    public static final String DEFAULT_CHARSET = "UTF-8";
    private String defaultCharset = DEFAULT_CHARSET;
    private KryoFactory kryoFactory = new DefaultKryoFactory();

    /**
     * Crate a message from the payload object and message properties provided. The message id will be added to the
     * properties if necessary later.
     *
     * @param object            the payload
     * @param messageProperties the message properties (headers)
     * @return a message
     */
    @Override
    protected Message createMessage(Object object, MessageProperties messageProperties) {
        byte[] bytes = null;
        Kryo kryo = kryoFactory.create();
        Output output = new ByteBufferOutput(4096, 1024 * 1024);
        try {
            kryo.writeClassAndObject(output, object);
            bytes = output.toBytes();
        } finally {
            output.close();
        }
        messageProperties.setContentType(CONTENT_TYPE);
        if (messageProperties.getContentEncoding() == null) {
            messageProperties.setContentEncoding(defaultCharset);
        }
        return new Message(bytes, messageProperties);
    }

    @Override
    public Object fromMessage(Message message) throws MessageConversionException {
        Object content = null;
        MessageProperties properties = message.getMessageProperties();
        if (properties != null) {
            if (properties.getContentType() != null && properties.getContentType().contains("x-kryo")) {
                Kryo kryo = kryoFactory.create();
                content = kryo.readClassAndObject(new ByteBufferInput(message.getBody()));
            } else {
                throw new MessageConversionException("Converter not applicable to this message");
            }
        }
        return content;
    }

    private class DefaultKryoFactory implements KryoFactory {

        @Override
        public Kryo create() {
            Kryo kryo = new Kryo();
            return kryo;
        }
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Spring Cloud为开发人员提供了快速构建分布式系统中一些常见模式的工具(例如配置管理,服务发现,断路器,智...
    卡卡罗2017阅读 134,837评论 18 139
  • Spring Boot 参考指南 介绍 转载自:https://www.gitbook.com/book/qbgb...
    毛宇鹏阅读 46,925评论 6 342
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 172,860评论 25 708
  • 愿今生相知,允来世相守
    百川西归阅读 201评论 2 1
  • 窗外万家灯火 窗内滴答和我 你不问为何我要开灯睡觉 你不问为何我不敢燃起炉灶 却得知我正游戏于网络而心里恼火 我已...
    牌牌阅读 206评论 0 0