使用protobuf 和 protostuff相互转换（基本信息）

骑猪看日落 2024-04-18 06:51 62阅读 0赞

1、导入pom

<dependency>
                <groupId>com.google.protobuf</groupId>
                <artifactId>protobuf-java</artifactId>
                <version>3.14.0</version>
            </dependency>
    
            <dependency>
                <groupId>com.googlecode.protobuf-java-format</groupId>
                <artifactId>protobuf-java-format</artifactId>
                <version>1.2</version>
            </dependency>
    
            <dependency>
                <groupId>com.dyuproject.protostuff</groupId>
                <artifactId>protostuff-core</artifactId>
                <version>1.0.9</version>
            </dependency>
    
            <dependency>
                <groupId>com.dyuproject.protostuff</groupId>
                <artifactId>protostuff-runtime</artifactId>
                <version>1.0.9</version>
            </dependency>

1、序列化工具

public final class ProtostuffUtil {
        private static Map<Class<?>, Schema<?>> cachedSchema = new ConcurrentHashMap<>();
    
        private static Objenesis objenesis = new ObjenesisStd(true);
        private static final ThreadLocal<LinkedBuffer> LOCAL_BUFFER =
                ThreadLocal.withInitial(() -> LinkedBuffer.allocate(LinkedBuffer.DEFAULT_BUFFER_SIZE));
    
        private static LinkedBuffer getBuffer() {
            return LOCAL_BUFFER.get().clear();
        }
    
        @SuppressWarnings("unchecked")
        private static <T> Schema<T> getSchema(Class<T> cls) {
            return (Schema<T>) cachedSchema.computeIfAbsent(cls, RuntimeSchema::createFrom);
        }
    
        /**
         * 序列化（对象 -> 字节数组）
         */
        @SuppressWarnings("unchecked")
        public static <T> byte[] serialize(T obj) {
            Class<T> cls = (Class<T>) obj.getClass();
            LinkedBuffer buffer = getBuffer();
            try {
                Schema<T> schema = getSchema(cls);
                return ProtobufIOUtil.toByteArray(obj, schema, buffer);
            } catch (Exception e) {
                throw new IllegalStateException(e.getMessage(), e);
            } finally {
                buffer.clear();
            }
        }
    
        /**
         * 反序列化（字节数组 -> 对象）
         */
        public static <T> T deserialize(byte[] data, Class<T> cls) {
            try {
                T message = objenesis.newInstance(cls);
                Schema<T> schema = getSchema(cls);
                ProtobufIOUtil.mergeFrom(data, message, schema);
                return message;
            } catch (Exception e) {
                throw new IllegalStateException(e.getMessage(), e);
            }
        }
    }

3、下载protoc.exe （windo工具，在下载包中找到bin/protoc.exe）

[https://github.com/protocolbuffers/protobuf/releases][https_github.com_protocolbuffers_protobuf_releases]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1NDE4NTE4_size_16_color_FFFFFF_t_70][]

4、编写java消息传递代码

@Data
    @Builder
    @ProtobufMessage
    public class User {
    
        private String name;
    
        private String address;
    
        private int age;
    
        private UserEnum userEnum; //枚举
    
        private PFMessage message; //对象
    
        private List<Integer> activitys; //集合
    
        public User(String name, String address, int age, UserEnum userEnum, PFMessage message, List<Integer> activitys) {
            this.name = name;
            this.address = address;
            this.age = age;
            this.userEnum = userEnum;
            this.message = message;
            this.activitys = activitys;
        }
    
        @Override
        public String toString() {
            return "User{" +
                    "name='" + name + '\'' +
                    ", address='" + address + '\'' +
                    ", age=" + age +
                    ", userEnum=" + userEnum +
                    ", message=" + message +
                    ", activitys=" + activitys +
                    '}';
        }
    }

5、生成.proto文件

syntax = "proto2";
    
    
    message User {
      optional string name = 1;
      optional string address = 2;
      optional int32 age = 3;
      optional UserEnum userEnum = 4;
      optional PFMessage message = 5;
      repeated int32 activitys = 6;
    }
    
    enum UserEnum {
      NUMBER_ONE = 0;
      NUMBER_tow = 1;
    }
    
    message ClusterMessage {
      optional string sessionId = 1;
      optional PFMessage msg = 2;
    }
    
    message PFMessage {
      optional int32 messageType = 1;
      optional int32 cmd = 2;
      optional bytes data = 3;
    }

6.编解码转换测试（证明是能完整解析的）

public class TestPb {
    
        public static void main(String[] args) {
            List<Integer> list = Arrays.asList(1,2,3,4,5,6);
            PFMessage message = new PFMessage(99,100,null);
            User user = new User("name","address1", 100, UserEnum.NUMBER_ONE, message, list);
    
            //Protostuff序列化
            byte[] data = ProtostuffUtil.serialize(user);
            //Protostuff反序列化
            User deserialUser = ProtostuffUtil.deserialize(data, User.class);
            System.out.println(deserialUser);
    
            //protobuf反序列化
            try{
                Proto2.User msg = Proto2.User.parseFrom(data);
                System.out.println(msg);
            }
            catch(Exception ex){
                System.out.println(ex.getMessage());
            }
        }
    }

[https_github.com_protocolbuffers_protobuf_releases]: https://github.com/protocolbuffers/protobuf/releases
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1NDE4NTE4_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/17/11a3b19dd2034447ac8080a5671a99cd.png