proto 3 语法详解（2）

oneof 类型

如果消息中有很多可选字段， 并且将来同时只有⼀个字段会被设置， 那么就可以使⽤ oneof 加强这个⾏为，也能有节约内存的效果。

升级通讯录⾄ 2.3 版本

通讯录 2.3 版本想新增联系⼈的其他联系⽅式，⽐如qq或者微信号⼆选⼀，我们就可以使⽤ oneof 字段来加强多选⼀这个⾏为。oneof 字段定义的格式为： oneof 字段名 { 字段 1; 字段 2; ... }

contacts.proto

syntax = "proto3";
package contacts;

import "google/protobuf/any.proto"; // 引入any.proto文件

message Address
{
  string home_address = 1; // 家庭住址
  string unit_address = 2; // 单位地址
}

// 联系人信息
message PeopleInfo 
{
  string name = 1; 
  int32 age = 2; 
  // 嵌套写法
  message Phone
  {
    string number = 1; // 电话号码
    enum PhoneType
    {
      MP = 0; // 移动电话
      TEL = 1; // 固定电话
    } 
    PhoneType type = 2; // 一个枚举类型，表示电话的类型
  }

  repeated Phone phones = 3;

  google.protobuf.Any data = 4; // 用来存地址信息

  oneof other_contact
  {
    string qq = 5; // 使用qq
    string vx = 6; // 使用微信
  }
}

// 通讯录
message Contacts
{
  repeated PeopleInfo contacts = 1;
}

注意：

可选字段中的字段编号，不能与⾮可选字段的编号冲突。

•

不能在 oneof 中使⽤ repeated 字段。

•

将来在设置 oneof 字段中值时，如果将 oneof 中的字段设置多个，那么只会保留最后⼀次设置的成

员，之前设置的 oneof 成员会⾃动清除。

 进行编译后：

 

上述的代码中，对于 oneof 字段：

•

会将 oneof 中的多个字段定义为⼀个枚举类型。

•

设置和获取：对 oneof 内的字段进⾏常规的设置和获取即可，但要注意只能设置⼀个。如果设置

多个，那么只会保留最后⼀次设置的成员。

•

清空oneof字段：clear_ ⽅法

•

获取当前设置了哪个字段：_case ⽅法

 当我们调用_case方法时，会返回一个枚举类型

我们就可以使用这个枚举类型来判断我们设置了哪个类型，或者没有设置。 

 更新 write.cc (通讯录 2.3)

 这里还是只展示新增部分的函数

void AddPeopelInfo(PeopleInfo* info)
{
    cout << "------------添加联系人-------------" << endl;
    cout << "请输入联系人姓名: " << endl;
    string name;
    getline(cin,name); 
    info->set_name(name);

    cout << "请输入联系人年龄: " << endl;
    int32_t age;
    cin >> age;
    cin.ignore(256,'\n'); // 清空一下缓冲区
    info->set_age(age);

    string numbers;
    for(int i = 1; ; ++i)
    {
        cout << "请输入联系人的第" << i << "个号码(两次回车终止输入)" << endl;
        getline(cin,numbers);
        if(numbers.empty()) break;
        cout << "请输入这个电话的类型(0/移动电话 1/固定电话): " << endl;
        int num = 0;
        cin >> num;
        cin.ignore(256,'\n'); // 清空以下缓冲区
        contacts::PeopleInfo_Phone* phone = info->add_phones();
        phone->set_number(numbers); // info->add_phones()返回的是一块开辟好空间的地址
        // 为了代码的可扩展性，这里使用switch而不是if语句
        switch(num)
        {
            case 1:
                phone->set_type(contacts::PeopleInfo_Phone_PhoneType::PeopleInfo_Phone_PhoneType_MP);
                break;
            case 2:
                phone->set_type(contacts::PeopleInfo_Phone_PhoneType::PeopleInfo_Phone_PhoneType_TEL);
                break;
            default:
                cout << "选择有误,取默认类型" << endl;
                break;
        }
    }
    
    contacts::Address address;
    cout << "请输入他/她的家庭地址: " << endl;
    string address_str;
    getline(cin,address_str);
    address.set_home_address(address_str);
    cout << "请输入他/她的单位地址: " << endl;
    getline(cin,address_str);
    address.set_unit_address(address_str);
    google::protobuf::Any* anydata = info->mutable_data();
    anydata->PackFrom(address); // 将任意的消息类型转化为Any类型

    cout << "请选择一种其它联系方式: 0/QQ , 1/微信" << endl;
    int contact_num;
    cin >> contact_num;
    cin.ignore(256,'\n');
    cout << "请输入号码: " << endl;
    string contact_nums;
    getline(cin,contact_nums);
    switch(contact_num)
    {
        case 0:
            info->set_qq(contact_nums);
            break;
        case 1:
            info->set_vx(contact_nums);
            break;
        default:
            cerr << "选择联系方式有误，设置失败" << endl;
            break;
    }
}

更新 read.cc (通讯录 2.3)

#include <iostream>
#include <fstream>
#include "contacts.pb.h"

using namespace std;
using namespace contacts;

const string path = "./contacts.bin";

int main()
{
    cout << "---------------打开文件并进行反序列化-----------" << endl;
    fstream inPut(path,ios::in | ios::binary);
    if(!inPut.is_open())
    {
        cerr << "打开文件失败: " << strerror(errno) << endl;
        exit(-1);
    }
    Contacts contacts;
    if(!contacts.ParseFromIstream(&inPut))
    {
        cerr << "反序列化解析文件失败" << endl;
        exit(1);
    }

    for(int i = 0; i < contacts.contacts_size(); ++i)
    {
        const PeopleInfo& people_info =  contacts.contacts(i); // 这种方式获取到每个元素
        cout << "---------第" << i + 1 << "个联系人----------" << endl;
        cout << "姓名: " << people_info.name()