"Fossies" - the Fresh Open Source Software Archive

Member "protobuf-3.21.1/examples/add_person.cc" (27 May 2022, 2800 Bytes) of package /linux/misc/protobuf-all-3.21.1.tar.gz:


As a special service "Fossies" has tried to format the requested source page into HTML format using (guessed) C and C++ source code syntax highlighting (style: standard) with prefixed line numbers and code folding option. Alternatively you can here view or download the uninterpreted source code file.

    1 // See README.txt for information and build instructions.
    2 
    3 #include <ctime>
    4 #include <fstream>
    5 #include <google/protobuf/util/time_util.h>
    6 #include <iostream>
    7 #include <string>
    8 
    9 #include "addressbook.pb.h"
   10 
   11 using namespace std;
   12 
   13 using google::protobuf::util::TimeUtil;
   14 
   15 // This function fills in a Person message based on user input.
   16 void PromptForAddress(tutorial::Person* person) {
   17   cout << "Enter person ID number: ";
   18   int id;
   19   cin >> id;
   20   person->set_id(id);
   21   cin.ignore(256, '\n');
   22 
   23   cout << "Enter name: ";
   24   getline(cin, *person->mutable_name());
   25 
   26   cout << "Enter email address (blank for none): ";
   27   string email;
   28   getline(cin, email);
   29   if (!email.empty()) {
   30     person->set_email(email);
   31   }
   32 
   33   while (true) {
   34     cout << "Enter a phone number (or leave blank to finish): ";
   35     string number;
   36     getline(cin, number);
   37     if (number.empty()) {
   38       break;
   39     }
   40 
   41     tutorial::Person::PhoneNumber* phone_number = person->add_phones();
   42     phone_number->set_number(number);
   43 
   44     cout << "Is this a mobile, home, or work phone? ";
   45     string type;
   46     getline(cin, type);
   47     if (type == "mobile") {
   48       phone_number->set_type(tutorial::Person::MOBILE);
   49     } else if (type == "home") {
   50       phone_number->set_type(tutorial::Person::HOME);
   51     } else if (type == "work") {
   52       phone_number->set_type(tutorial::Person::WORK);
   53     } else {
   54       cout << "Unknown phone type.  Using default." << endl;
   55     }
   56   }
   57   *person->mutable_last_updated() = TimeUtil::SecondsToTimestamp(time(NULL));
   58 }
   59 
   60 // Main function:  Reads the entire address book from a file,
   61 //   adds one person based on user input, then writes it back out to the same
   62 //   file.
   63 int main(int argc, char* argv[]) {
   64   // Verify that the version of the library that we linked against is
   65   // compatible with the version of the headers we compiled against.
   66   GOOGLE_PROTOBUF_VERIFY_VERSION;
   67 
   68   if (argc != 2) {
   69     cerr << "Usage:  " << argv[0] << " ADDRESS_BOOK_FILE" << endl;
   70     return -1;
   71   }
   72 
   73   tutorial::AddressBook address_book;
   74 
   75   {
   76     // Read the existing address book.
   77     fstream input(argv[1], ios::in | ios::binary);
   78     if (!input) {
   79       cout << argv[1] << ": File not found.  Creating a new file." << endl;
   80     } else if (!address_book.ParseFromIstream(&input)) {
   81       cerr << "Failed to parse address book." << endl;
   82       return -1;
   83     }
   84   }
   85 
   86   // Add an address.
   87   PromptForAddress(address_book.add_people());
   88 
   89   {
   90     // Write the new address book back to disk.
   91     fstream output(argv[1], ios::out | ios::trunc | ios::binary);
   92     if (!address_book.SerializeToOstream(&output)) {
   93       cerr << "Failed to write address book." << endl;
   94       return -1;
   95     }
   96   }
   97 
   98   // Optional:  Delete all global objects allocated by libprotobuf.
   99   google::protobuf::ShutdownProtobufLibrary();
  100 
  101   return 0;
  102 }