美文网首页spring征服SpringSpringFramework
使用Kryo替换spring amqp的Java序列化

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

作者: 云原生指北 | 来源:发表于2016-06-29 13:30 被阅读513次

    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;
            }
        }
    }
    

    相关文章

      网友评论

      • sys:支持原创

      本文标题:使用Kryo替换spring amqp的Java序列化

      本文链接:https://www.haomeiwen.com/subject/khhxjttx.html