如何使用Protobuf的反射修改地图

Question

我正在使用C ++ 14项目中的ProtoBuf3。有一些功能，可以返回google :: ProtoBuf :: Message*作为RPC请求，我需要做的是设置其字段。因此，我需要使用ProtoBuf3的反射。

这是一个原始文件：

syntax="proto3";
package srv.user;

option cc_generic_services = true;
message BatchGetUserInfosRequest {
    uint64 my_uid = 1;
    repeated uint64 peer_uids = 2;
    map<string, string> infos = 3;
}

message BatchGetUserInfosResponse {
    uint64 my_uid = 1;
    string info = 2;
}

Service UserSrv {
    rpc BatchGetUserInfos(BatchGetUserInfosRequest) returns (BatchGetUserInfosResponse);
};

现在我打了一个函数，它返回google :: protobuf :: message*，指向对象batchgetuserinfosrequest，然后尝试设置其字段。

// msg is a Message*, pointing to an object of BatchGetUserInfosRequest
auto descriptor = msg->GetDescriptor();
auto reflection = msg->GetReflection();
auto field = descriptor->FindFieldByName("my_uid");
reflection->SetUInt64(msg, field, 1234);
auto field2 = descriptor->FindFieldByName("peer_uids");
reflection->GetMutableRepeatedFieldRef<uint64_t>(msg, field2).CopyFrom(peerUids); // peerUids is a std::vector<uint64_t>

如您所见，我可以设置my_uid和peer_uids如上所述，但是对于字段infos，它是google :: ProtoBuf :: MAP，我不知道如何使用反射机制设置它。

Answer 1

如果您深入研究，您会在proto3中找到map在repotedfield上实现：

  // Whether the message is an automatically generated map entry type for the
  // maps field.
  //
  // For maps fields:
  //     map<KeyType, ValueType> map_field = 1;
  // The parsed descriptor looks like:
  //     message MapFieldEntry {
  //         option map_entry = true;
  //         optional KeyType key = 1;
  //         optional ValueType value = 2;
  //     }
  //     repeated MapFieldEntry map_field = 1;
  //
  // Implementations may choose not to generate the map_entry=true message, but
  // use a native map in the target language to hold the keys and values.
  // The reflection APIs in such implementations still need to work as
  // if the field is a repeated message field.
  //
  // NOTE: Do not set the option in .proto files. Always use the maps syntax
  // instead. The option should only be implicitly set by the proto compiler
  // parser.
  optional bool map_entry = 7;

从

  BatchGetUserInfosRequest message;
  auto *descriptor = message.GetDescriptor();
  auto *reflection = message.GetReflection();

  const google::protobuf::FieldDescriptor *fd_map_string_string =
      descriptor->FindFieldByName("infos");
  const google::protobuf::FieldDescriptor *fd_map_string_string_key =
      fd_map_string_string->message_type()->map_key();
  const google::protobuf::FieldDescriptor *fd_map_string_string_value =
      fd_map_string_string->message_type()->map_value();

  const google::protobuf::MutableRepeatedFieldRef<google::protobuf::Message>
      mmf_string_string =
          reflection->GetMutableRepeatedFieldRef<google::protobuf::Message>(
              &message, fd_map_string_string);
  std::unique_ptr<google::protobuf::Message> entry_string_string(
      google::protobuf::MessageFactory::generated_factory()
          ->GetPrototype(fd_map_string_string->message_type())
          ->New(message.GetArena()));
  entry_string_string->GetReflection()->SetString(
      entry_string_string.get(), fd_map_string_string->message_type()->field(0),
      "1234");
  entry_string_string->GetReflection()->SetString(
      entry_string_string.get(), fd_map_string_string->message_type()->field(1),
      std::to_string(10));
  mmf_string_string.Add(*entry_string_string);


  std::cout << "1234: " << message.infos().at("1234") << '\n';

：

1234: 10