1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc. All rights reserved.
3 // https://developers.google.com/protocol-buffers/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 // * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 // * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 // * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
31 #include <fstream>
32 #include <iostream>
33 #include "benchmarks.pb.h"
34
35 using benchmarks::BenchmarkDataset;
36 using google::protobuf::Descriptor;
37 using google::protobuf::DescriptorPool;
38 using google::protobuf::Message;
39 using google::protobuf::MessageFactory;
40
41 std::set<std::string> names;
42
43 const char *file_prefix = "dataset.";
44 const char *file_suffix = ".pb";
45
WriteFileWithPayloads(const std::string & name,const std::string & message_name,const std::vector<std::string> & payload)46 void WriteFileWithPayloads(const std::string& name,
47 const std::string& message_name,
48 const std::vector<std::string>& payload) {
49 if (!names.insert(name).second) {
50 std::cerr << "Duplicate test name: " << name << "\n";
51 abort();
52 }
53
54 // First verify that this message name exists in our set of benchmark messages
55 // and that these payloads are valid for the given message.
56 const Descriptor* d =
57 DescriptorPool::generated_pool()->FindMessageTypeByName(message_name);
58
59 if (!d) {
60 std::cerr << "For dataset " << name << ", no such message: "
61 << message_name << "\n";
62 abort();
63 }
64
65 Message* m = MessageFactory::generated_factory()->GetPrototype(d)->New();
66
67 for (size_t i = 0; i < payload.size(); i++) {
68 if (!m->ParseFromString(payload[i])) {
69 std::cerr << "For dataset " << name << ", payload[" << i << "] fails "
70 << "to parse\n";
71 abort();
72 }
73 }
74
75 BenchmarkDataset dataset;
76 dataset.set_name(name);
77 dataset.set_message_name(message_name);
78 for (size_t i = 0; i < payload.size(); i++) {
79 dataset.add_payload()->assign(payload[i]);
80 }
81
82 std::ofstream writer;
83 std::string fname = file_prefix + name + file_suffix;
84 writer.open(fname.c_str());
85 dataset.SerializeToOstream(&writer);
86 writer.close();
87
88 std::cerr << "Wrote dataset: " << fname << "\n";
89 }
90
WriteFile(const std::string & name,const std::string & message_name,const std::string & payload)91 void WriteFile(const std::string& name, const std::string& message_name,
92 const std::string& payload) {
93 std::vector<std::string> payloads;
94 payloads.push_back(payload);
95 WriteFileWithPayloads(name, message_name, payloads);
96 }
97
ReadFile(const std::string & name)98 std::string ReadFile(const std::string& name) {
99 std::ifstream file(name.c_str());
100 GOOGLE_CHECK(file.is_open()) << "Couldn't find file '" << name <<
101 "', please make sure you are running "
102 "this command from the benchmarks/ "
103 "directory.\n";
104 return std::string((std::istreambuf_iterator<char>(file)),
105 std::istreambuf_iterator<char>());
106 }
107
main()108 int main() {
109 WriteFile("google_message1_proto3", "benchmarks.proto3.GoogleMessage1",
110 ReadFile("google_message1.dat"));
111 WriteFile("google_message1_proto2", "benchmarks.proto2.GoogleMessage1",
112 ReadFile("google_message1.dat"));
113
114 // Not in proto3 because it has a group, which is not supported.
115 WriteFile("google_message2", "benchmarks.proto2.GoogleMessage2",
116 ReadFile("google_message2.dat"));
117 }
118