netty之Protostuff序列化协议

• 前言
• 依赖
• 定义编解码器
• • 工具类
  • 编码器
  • 解码器
• 整合到netty
• 注意事项

比起上文的Protobuf协议，Protostuff的好处就是不用去编辑proto文件，然后生成对应的java文件，直接可以利用现有的实体类来通信，而且性能性能损失也是很少的。

github官方链接地址

	
	     io.protostuff
	     protostuff-core
	     1.7.2
	 
	
	 
	     io.protostuff
	     protostuff-runtime
	     1.7.2
	 

Protobuf有netty提供已经封装好的编解码器，但是Protostuff需要自己定义。

public class ProtostuffUtils {

    
    private static Map, Schema> schemaCache = new ConcurrentHashMap<>();

    
    @SuppressWarnings("unchecked")
    public static  byte[] serialize(T obj) {
        Class clazz = (Class) obj.getClass();
        Schema schema = getSchema(clazz);
        LinkedBuffer buffer = LinkedBuffer.allocate(LinkedBuffer.DEFAULT_BUFFER_SIZE);
        byte[] data;
        try {
            data = ProtobufIOUtil.toByteArray(obj, schema, buffer);
        } finally {
            buffer.clear();
        }

        return data;
    }

    
    public static  T deserialize(byte[] data, Class clazz) {
        Schema schema = getSchema(clazz);
        T obj = schema.newMessage();
        ProtobufIOUtil.mergeFrom(data, obj, schema);
        return obj;
    }

    @SuppressWarnings("unchecked")
    private static  Schema getSchema(Class clazz) {
        Schema schema = (Schema) schemaCache.get(clazz);
        if (Objects.isNull(schema)) {
            //这个schema通过RuntimeSchema进行懒创建并缓存
            //所以可以一直调用RuntimeSchema.getSchema(),这个方法是线程安全的
            schema = RuntimeSchema.getSchema(clazz);
            if (Objects.nonNull(schema)) {
                schemaCache.put(clazz, schema);
            }
        }
        return schema;
    }
}

这里采用分隔符的方式来解决拆包粘包的问题，那么分隔符实验了多次，如果用换行符n，会报错，错误信息如下：

Protocol message contained an invalid tag (zero)

好像是因为的问题，所以只能采用不带的分隔符了，这里选用@来作为分隔符。

public class ProtostuffEncoder extends MessageToByteEncoder {

    //分隔符@，放在配置文件，方便修改配置
    private String delimiter;

    public ProtostuffEncoder(String delimiter) {
        this.delimiter = delimiter;
    }

    @Override
    protected void encode(ChannelHandlerContext ctx, Object msg, ByteBuf out) throws Exception {
        if (msg instanceof ComputeResult) {
            byte[] bytes = ProtostuffUtils.serialize(msg);
            byte[] delimiterBytes = delimiter.getBytes();
            byte[] total = new byte[bytes.length + delimiterBytes.length];
            System.arraycopy(bytes, 0, total, 0, bytes.length);
            System.arraycopy(delimiterBytes, 0, total, bytes.length, delimiterBytes.length);
            out.writeBytes(total);
        }
    }
}

public class ProtostuffDecoder extends ByteToMessageDecoder {

	// 写在这里，方便扩展
    private Class clazz;

    public ProtostuffDecoder(Class clazz) {
        this.clazz = clazz;
    }

    @Override
    protected void decode(ChannelHandlerContext ctx, ByteBuf in, List