Protocol Buffers（2）：编码与解码

栏目: 服务器 · 发布时间: 6年前

内容简介：在上一篇文章中我们提到，对于序列化后字节流，需要回答的一个重要问题是“key由wire type和FieldNumber两部分编码而成，具体地key的最低3个bit为wire type，

在上一篇文章中我们提到，对于序列化后字节流，需要回答的一个重要问题是“ 从哪里到哪里是哪个数据成员 ”。

message中每一个field的格式为：

required/optional/repeated FieldType FieldName = FieldNumber（a unique number in current message）

在序列化时，一个field对应一个 key-value 对，整个二进制文件就是一连串紧密排列的 key-value 对，key也称为tag，先上图直观感受一下，图片来自 Encoding and Evolution ：

Protocol Buffers（2）：编码与解码 — Example record encoded using Protocol Buffers

key由wire type和FieldNumber两部分编码而成，具体地 key = (field_number << 3) | wire_type ， field_number 部分指示了当前是哪个数据成员，通过它将cc和h文件中的数据成员与当前的key-value对应起来。

key的最低3个bit为wire type， 什么是wire type？ 如下表所示：

wire type被如此设计，主要是为了解决一个问题， 如何知道接下来value部分的长度（字节数）

，如果

wire type = 0、1、5，编码为 key + 数据 ，只有一个数据，可能占数个字节， 数据在编码时自带终止标记
wire type = 2，编码为 key + length + 数据 ，length指示了数据长度，可能有多个数据，顺序排在length后

解码代码一窥

接下来，我们直接看一下example.pb.cc及相关的源码，看下key-value对是如何解析的。解码过程相对简单，理解了解码过程，编码也就比较显然了。

// example.proto
package example;

message Person {
  required string name = 1;
  required int32 id = 2;
  optional string email = 3;
}

// in example.pb.cc
bool Person::MergePartialFromCodedStream(
    ::google::protobuf::io::CodedInputStream* input) {
#define DO_(EXPRESSION) if (!PROTOBUF_PREDICT_TRUE(EXPRESSION)) goto failure
  ::google::protobuf::uint32 tag;
  // @@protoc_insertion_point(parse_start:example.Person)
  for (;;) {
    ::std::pair<::google::protobuf::uint32, bool> p = input->ReadTagWithCutoffNoLastTag(127u);
    tag = p.first;
    if (!p.second) goto handle_unusual;
    switch (::google::protobuf::internal::WireFormatLite::GetTagFieldNumber(tag)) {
      // required string name = 1;
      case 1: {
        if (static_cast< ::google::protobuf::uint8>(tag) == (10 & 0xFF)) { // 10 = (1 << 3) + 2
          DO_(::google::protobuf::internal::WireFormatLite::ReadString(
                input, this->mutable_name()));
          ::google::protobuf::internal::WireFormat::VerifyUTF8StringNamedField(
            this->name().data(), static_cast<int>(this->name().length()),
            ::google::protobuf::internal::WireFormat::PARSE,
            "example.Person.name");
        } else {
          goto handle_unusual;
        }
        break;
      }

      // required int32 id = 2;
      case 2: {
        if (static_cast< ::google::protobuf::uint8>(tag) == (16 & 0xFF)) { // 16 = (2 << 8) + 0
          HasBitSetters::set_has_id(this);
          DO_((::google::protobuf::internal::WireFormatLite::ReadPrimitive<
                   ::google::protobuf::int32, ::google::protobuf::internal::WireFormatLite::TYPE_INT32>(
                 input, &id_)));
        } else {
          goto handle_unusual;
        }
        break;
      }

      // optional string email = 3;
      case 3: {
        if (static_cast< ::google::protobuf::uint8>(tag) == (26 & 0xFF)) {
          DO_(::google::protobuf::internal::WireFormatLite::ReadString(
                input, this->mutable_email()));
          ::google::protobuf::internal::WireFormat::VerifyUTF8StringNamedField(
            this->email().data(), static_cast<int>(this->email().length()),
            ::google::protobuf::internal::WireFormat::PARSE,
            "example.Person.email");
        } else {
          goto handle_unusual;
        }
        break;
      }

      default: {
      handle_unusual:
        if (tag == 0) {
          goto success;
        }
        DO_(::google::protobuf::internal::WireFormat::SkipField(
              input, tag, _internal_metadata_.mutable_unknown_fields()));
        break;
      }
    }
  }
success:
  // @@protoc_insertion_point(parse_success:example.Person)
  return true;
failure:
  // @@protoc_insertion_point(parse_failure:example.Person)
  return false;
#undef DO_
}

整段代码在循环地解析 input 流，遇到1个 tag （key），根据其wire type和数据类型调用相应的解析函数，如果是 string ，则调用 ReadString ， ReadString 会一直调用到 ReadBytesToString ，如果是 int32 ，则调用 ReadPrimitive ， ReadPrimitive 中会进一步调用 ReadVarint32 。可以看到，生成的example.pb.cc决定了遇到哪个 tag 调用哪个解析函数，从输入流中解析出值，赋给对应的成员变量，而真正进行解析的代码实际上是Protobuf的源码，如下所示：

// in wire_format_lit.cc
inline static bool ReadBytesToString(io::CodedInputStream* input,
                                     string* value) {
  uint32 length;
  return input->ReadVarint32(&length) &&
      input->InternalReadStringInline(value, length);
}

// in wire_format_lit.h
template <>
inline bool WireFormatLite::ReadPrimitive<int32, WireFormatLite::TYPE_INT32>(
    io::CodedInputStream* input,
    int32* value) {
  uint32 temp;
  if (!input->ReadVarint32(&temp)) return false;
  *value = static_cast<int32>(temp);
  return true;
}

// in coded_stream.h
inline bool CodedInputStream::ReadVarint32(uint32* value) {
  uint32 v = 0;
  if (PROTOBUF_PREDICT_TRUE(buffer_ < buffer_end_)) {
    v = *buffer_;
    if (v < 0x80) {
      *value = v;
      Advance(1);
      return true;
    }
  }
  int64 result = ReadVarint32Fallback(v);
  *value = static_cast<uint32>(result);
  return result >= 0;
}

可以看到，如果遇到 int32 的 tag ，直接读取接下来的数据，如果遇到 string 的 tag ，会先读一个Varint32的 length ，然后再读 length 个字节的数据。

这里频繁出现了varint， length 是varint，存储的 int32 数据也是varint，那varint是什么？

varint

varint是一种 可变长编码 ，使用1个或多个字节对整数进行编码，可编码任意大的整数， 小整数占用的字节少，大整数占用的字节多 ，如果小整数更频繁出现，则通过varint可实现压缩存储。

varint中每个字节的最高位bit称之为 most significant bit (MSB) ，如果该bit为0意味着这个字节为表示当前整数的最后一个字节，如果为1则表示后面还有至少1个字节，可见， varint的终止位置其实是自解释的 。

在Protobuf中，tag和length都是使用varint编码的。 length 和 tag 中的field_number都是正整数 int32 ，这里提一下 tag ，它的低3位bit为wire type，如果只用1个字节表示的话，最高位bit为0，则留给field_number只有4个bit位，1到15，如果field_number大于等于16，就需要用2个字节，所以对于频繁使用的field其field_number应设置为1到15。

比如正整数150，其使用varint编码如下（小端存储）：

// proto file
message Test1 {
  optional int32 a = 1;
}

// c++ file
// set a = 150

// binary file, in hex
// 08 96 01

其中 08 为key， 96 01 为150的varint编码，解释如下

有关varint的更多内容，可以参见 wiki Variable-length quantity

。

至此，key-value的编码方式我们已经解决了一半，还剩value部分没有解决，接下来看看Protobuf数据部分是如何编码的。

Protobuf中的整数和浮点数

Protobuf中整数也是通过varint进行编码，移除每个字节的MSB，然后拼接在一起，可以得到一个含有数个字节的buffer，这个buffer该怎么解释还需要 参考具体的数据类型 。

对于 int32 或 int64 ，正数直接按varint编码，数据类型为 int32 或 int64 的负数统一被编码为10个字节长的varint（补码）。

如果是 sint32 或 sint64 ，则采用ZigZag方式进行编码，如下表所示：

sint32 n 被编码为 (n << 1) ^ (n >> 31) 对应的varint， sint64 n 被编码为 (n << 1) ^ (n >> 63) 对应的varint，这样， 绝对值较小的整数只需要较少的字节就可以表示

。

至于浮点数，对应的wire type为1或5，直接按小端存储。

Length-delimited相关类型

主要有3类：string、嵌套message以及packed repeated fields。它们的编码方式统一为 tag + length + 数据 ，只是数据部分有所差异。

string的编码为 key + length + 字符 ，参看开篇的图片已经很清晰了。

嵌套message也很简单，直接将嵌套message部分的编码接在 length 后即可，如下所示：

// proto file
message Test1 {
  optional int32 a = 1;
}
message Test3 {
  optional Test1 c = 3;
}

// cpp file
// set a = 150

// message Test3 binary file, in hex
// 1a 03 08 96 01

其中， 1a 为 c 的key， 03 为 c 的长度，接下来的 08 96 01 为 a 的key+value。

packed repeated fields，指的是proto2中声明了 [packed=true] 的repeated varint、32bit or 64bit数据，proto3中repeated默认packed，如下所示

// in proto2
message Test4 {
  repeated int32 d = 4 [packed=true];
}

// in proto3
message Test4 {
  repeated int32 d = 4;
}

// 3, 270, 86942压缩存储如下，in hex
22        // key (field number 4, wire type 2), 0x22 = 34 = (4 << 3) + 2
06        // payload size (6 bytes), length
03        // first element (varint 3)
8E 02     // second element (varint 270)
9E A7 05  // third element (varint 86942)

6个字节根据varint的MSB可自动分割成3个数据。对这种packed repeated fields，在Protobuf中会以 RepeatedField 对象承载，支持get-by-index、set-by-index和add（添加元素）操作。

小结

至此，二进制文件中key-value对的编码方式已基本介绍完毕，后面将通过一个相对复杂的例子，将这些琐碎的编码方式串起来，以加深理解。

参考

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。

码农书籍

MySQL性能调优与架构设计

简朝阳 / 2009-6 / 59.80元

《MySQL性能调优与架构设计》以 MySQL 数据库的基础及维护为切入点，重点介绍了 MySQL 数据库应用系统的性能调优，以及高可用可扩展的架构设计。全书共分3篇，基础篇介绍了MySQL软件的基础知识、架构组成、存储引擎、安全管理及基本的备份恢复知识。性能优化篇从影响 MySQL 数据库应用系统性能的因素开始，针对性地对各个影响因素进行调优分析。如 MySQL Schema 设计的技巧......一起来看看《MySQL性能调优与架构设计》这本书的介绍吧!

码农工具