Использование расширений Protobuf C++

Question

Я использую библиотеку google protobuf версии 2.61 и хочу использовать расширения,

У меня есть следующие прото файлы:

package communication;

message BaseMessage {
  required uint64 server_id     = 1;
  required string uuid          = 2;
  required uint64 message_id    = 3;

  extensions 100 to max;
}

message GetIdentify {

  extend BaseMessage {
    optional GetIdentify message = 100;
  }

  required string hostname = 1;
}

Я могу создать сообщение, используя следующий код:

communication::BaseMessage base_message;
base_message.set_message_id(123456);
base_message.set_server_id(112313123);
base_message.set_uuid("asdaskdjasd213123123asd");
base_message.MutableExtension(communication::GetIdentify::message)->set_hostname("http://test123123123ing");

Однако я хотел бы сделать обратное действие и получить сообщение с неизвестным расширением и проанализировать его и найти, какое расширение это и анализировать в соответствии с ним.

Я использовал nanopb для моего проекта c и Python. но мне действительно трудно писать код protobuf на C++, потому что я не могу найти достаточно хорошую документацию и примеры кода.

Есть ли способ сделать это без добавления дополнительной переменной типа расширения?

Также, что является самым элегантным способом сделать это в C++

Answer 1

Библиотека разбирает сообщение и все расширения. Вы можете проверить наличие расширения, используя метод HasExtension.

Реализация Java должна зарегистрировать расширения в парсере перед анализом. Но в C++ все делается автоматически. Пожалуйста, смотрите следующий код. (Я проверил это с protobuf 2.5.0)

Создать и написать сообщение:

#include <iostream>
#include <fstream>
#include <string>
#include <communication/p.pb.h> 
using namespace std;

int
main(int, char **)
{
    communication::BaseMessage base_message;
    base_message.set_message_id(123456);
    base_message.set_server_id(112313123);
    base_message.set_uuid("asdaskdjasd213123123asd");
    base_message.MutableExtension(communication::GetIdentify::message)->set_hostname("http://test123123123ing");
    base_message.SerializeToOstream(&cout);
    return 0;
}

Прочитайте сообщение, протестируйте расширение и распечатайте его:

#include <iostream>
#include <fstream>
#include <string>
#include <communication/p.pb.h>
#include <google/protobuf/text_format.h>
using namespace google::protobuf;
using namespace std;

int
main(int, char **)
{
    communication::BaseMessage base_message;
    base_message.ParseFromIstream(&cin);

    if (base_message.HasExtension(communication::GetIdentify::message)) {
        const communication::GetIdentify &ii = base_message.GetExtension(communication::GetIdentify::message);
        cout << "yes, msg has extension: " << ii.hostname() << endl << endl;
    } else {
        cout << "no, msg has no extension" << endl << endl;
    }

    string res;
    TextFormat::PrintToString(base_message, &res);

    cout << res << endl;
    return 0;
}