fastjson:实现对java.nio.ByteBuffer数据类型的支持

来源:互联网 发布:松下plc编程视频教程 编辑:程序博客网 时间:2024/05/16 16:08

一般情况下,我们都用byte[]作为保存二进制数据的数据类型,较少用ByteBuffer来表示字节数组。然而最近的工作中需要二进制数据(字节数组)即支持fastjson序列化也要能用thrift:swift序列化。
我们知道fastjson可以对byte[]进行序列化,但fastjson默认是不支持java.noi.ByteBuffer数据类型序列化的。
而thrift是跨语言的,所以它并没有byte[]这样的类型,对于二进制数据则对应的用binary类型,在java对应的是java.nio.ByteBuffer。这就尴尬了。

那么到底字节数组用什么表示?byte[] or ByteBuffer?

看来还是得用ByteBuffer,因为fastjson是个非常灵活的框架,允许使用者通过自定义对象实现对特定类型的序列化支持,只要fastjson能支持ByteBuffer的序列化,thrift也就没问题了。
如果想要实现对ByteBuffer类型的支持,需要自己实现ByteBuffer的序列化器和反序列化器(ObjectSerializer, ObjectDeserializer),以下是代码实现。
ByteBufferCodec.java

package gu.simplemq.json;import java.io.IOException;import java.lang.reflect.Type;import java.nio.ByteBuffer;import com.alibaba.fastjson.JSONException;import com.alibaba.fastjson.parser.DefaultJSONParser;import com.alibaba.fastjson.parser.JSONToken;import com.alibaba.fastjson.parser.deserializer.ObjectDeserializer;import com.alibaba.fastjson.serializer.JSONSerializer;import com.alibaba.fastjson.serializer.ObjectSerializer;import com.alibaba.fastjson.serializer.PrimitiveArraySerializer;import com.alibaba.fastjson.serializer.SerializerFeature;/** * {@link ByteBuffer} 序列化反序列化支持 * @author guyadong * */public class ByteBufferCodec implements ObjectSerializer, ObjectDeserializer {    public static ByteBufferCodec instance = new ByteBufferCodec();    @SuppressWarnings("unchecked")    @Override    public <T> T deserialze(DefaultJSONParser parser, Type type, Object fieldName) {        int token = parser.lexer.token();        if (token == JSONToken.NULL) {            parser.lexer.nextToken();            return null;        } else if(token == JSONToken.HEX  || token == JSONToken.LITERAL_STRING){            byte[] bytes = parser.lexer.bytesValue();            parser.lexer.nextToken();            return (T) ByteBuffer.wrap(bytes);        }        throw new JSONException(String.format("invalid '%s' for ByteBuffer",JSONToken.name(token)));    }    @Override    public int getFastMatchToken() {        return JSONToken.LITERAL_STRING;    }    /**     * 返回buffer中所有字节(position~limit),不改变buffer状态     * @param buffer     * @return     */    private static final byte[] getAllBytesInBuffer(ByteBuffer buffer){        int pos = buffer.position();        try{            byte[] bytes = new byte[buffer.remaining()];            buffer.get(bytes);            return bytes;        }finally{            buffer.position(pos);        }    }    /**      * 直接引用{@link PrimitiveArraySerializer}实现序列化     */    @Override    public void write(JSONSerializer serializer, Object object, Object fieldName, Type fieldType, int features)            throws IOException {        if ( (object instanceof ByteBuffer) ) {            PrimitiveArraySerializer.instance.write(serializer, getAllBytesInBuffer((ByteBuffer)object), fieldName, fieldType, features);        }else{            serializer.out.writeNull(SerializerFeature.WriteNullListAsEmpty);        }    }}

使用示例:

package gu.simplemq;import java.nio.ByteBuffer;import org.junit.Test;import com.alibaba.fastjson.JSON;import com.alibaba.fastjson.parser.ParserConfig;import com.alibaba.fastjson.serializer.SerializeConfig;import gu.simplemq.json.ByteBufferCodec;public class TestByteBuffer {    @Test    public void test() {        ByteBuffer byteBuffer =ByteBuffer.wrap(new byte[]{22,33,3,2,3,1,5,-1});        // 修改ParserConfig.global全局变量,将ByteBufferCodec加入反序列化器容器        ParserConfig.global.putDeserializer(ByteBuffer.class, ByteBufferCodec.instance);        // 修改SerializeConfig.globalInstance全局变量,将ByteBufferCodec加入序列化器容器        // 对应的Class为 java.nio.HeapByteBuffer        SerializeConfig.globalInstance.put(ByteBuffer.wrap(new byte[]{}).getClass(), ByteBufferCodec.instance);        // 待序列化的 ByteBuffer 对象        ByteBuffer byteBuffer =ByteBuffer.wrap(new byte[]{22,33,3,2,3,1,5,-1});        // 序列化        String serString = JSON.toJSONString(group.byteBuffer);        System.out.println(serString);        // 反序列化        ByteBuffer deserialedByteBuffer = JSON.parseObject(serString,ByteBuffer.class);        System.out.println(JSON.toJSONString(deserialedByteBuffer));    }}