OLD | NEW |
| (Empty) |
1 // Protocol Buffers - Google's data interchange format | |
2 // Copyright 2009 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 package com.google.protocolbuffers; | |
32 | |
33 import java.io.ByteArrayInputStream; | |
34 import java.io.ByteArrayOutputStream; | |
35 import java.io.File; | |
36 import java.io.FileOutputStream; | |
37 import java.io.FileNotFoundException; | |
38 import java.io.IOException; | |
39 import java.io.RandomAccessFile; | |
40 import java.lang.reflect.Method; | |
41 | |
42 import com.google.protobuf.ByteString; | |
43 import com.google.protobuf.CodedInputStream; | |
44 import com.google.protobuf.CodedOutputStream; | |
45 import com.google.protobuf.Message; | |
46 | |
47 public class ProtoBench { | |
48 | |
49 private static final long MIN_SAMPLE_TIME_MS = 2 * 1000; | |
50 private static final long TARGET_TIME_MS = 30 * 1000; | |
51 | |
52 private ProtoBench() { | |
53 // Prevent instantiation | |
54 } | |
55 | |
56 public static void main(String[] args) { | |
57 if (args.length < 2 || (args.length % 2) != 0) { | |
58 System.err.println("Usage: ProtoBench <descriptor type name> <input data>"
); | |
59 System.err.println("The descriptor type name is the fully-qualified messag
e name,"); | |
60 System.err.println("e.g. com.google.protocolbuffers.benchmark.Message1"); | |
61 System.err.println("(You can specify multiple pairs of descriptor type nam
e and input data.)"); | |
62 System.exit(1); | |
63 } | |
64 boolean success = true; | |
65 for (int i = 0; i < args.length; i += 2) { | |
66 success &= runTest(args[i], args[i + 1]); | |
67 } | |
68 System.exit(success ? 0 : 1); | |
69 } | |
70 | |
71 /** | |
72 * Runs a single test. Error messages are displayed to stderr, and the return
value | |
73 * indicates general success/failure. | |
74 */ | |
75 public static boolean runTest(String type, String file) { | |
76 System.out.println("Benchmarking " + type + " with file " + file); | |
77 final Message defaultMessage; | |
78 try { | |
79 Class<?> clazz = Class.forName(type); | |
80 Method method = clazz.getDeclaredMethod("getDefaultInstance"); | |
81 defaultMessage = (Message) method.invoke(null); | |
82 } catch (Exception e) { | |
83 // We want to do the same thing with all exceptions. Not generally nice, | |
84 // but this is slightly different. | |
85 System.err.println("Unable to get default message for " + type); | |
86 return false; | |
87 } | |
88 | |
89 try { | |
90 final byte[] inputData = readAllBytes(file); | |
91 final ByteArrayInputStream inputStream = new ByteArrayInputStream(inputDat
a); | |
92 final ByteString inputString = ByteString.copyFrom(inputData); | |
93 final Message sampleMessage = defaultMessage.newBuilderForType().mergeFrom
(inputString).build(); | |
94 FileOutputStream devNullTemp = null; | |
95 CodedOutputStream reuseDevNullTemp = null; | |
96 try { | |
97 devNullTemp = new FileOutputStream("/dev/null"); | |
98 reuseDevNullTemp = CodedOutputStream.newInstance(devNullTemp); | |
99 } catch (FileNotFoundException e) { | |
100 // ignore: this is probably Windows, where /dev/null does not exist | |
101 } | |
102 final FileOutputStream devNull = devNullTemp; | |
103 final CodedOutputStream reuseDevNull = reuseDevNullTemp; | |
104 benchmark("Serialize to byte string", inputData.length, new Action() { | |
105 public void execute() { sampleMessage.toByteString(); } | |
106 }); | |
107 benchmark("Serialize to byte array", inputData.length, new Action() { | |
108 public void execute() { sampleMessage.toByteArray(); } | |
109 }); | |
110 benchmark("Serialize to memory stream", inputData.length, new Action() { | |
111 public void execute() throws IOException { | |
112 sampleMessage.writeTo(new ByteArrayOutputStream()); | |
113 } | |
114 }); | |
115 if (devNull != null) { | |
116 benchmark("Serialize to /dev/null with FileOutputStream", inputData.leng
th, new Action() { | |
117 public void execute() throws IOException { | |
118 sampleMessage.writeTo(devNull); | |
119 } | |
120 }); | |
121 benchmark("Serialize to /dev/null reusing FileOutputStream", inputData.l
ength, new Action() { | |
122 public void execute() throws IOException { | |
123 sampleMessage.writeTo(reuseDevNull); | |
124 reuseDevNull.flush(); // force the write to the OutputStream | |
125 } | |
126 }); | |
127 } | |
128 benchmark("Deserialize from byte string", inputData.length, new Action() { | |
129 public void execute() throws IOException { | |
130 defaultMessage.newBuilderForType().mergeFrom(inputString).build(); | |
131 } | |
132 }); | |
133 benchmark("Deserialize from byte array", inputData.length, new Action() { | |
134 public void execute() throws IOException { | |
135 defaultMessage.newBuilderForType() | |
136 .mergeFrom(CodedInputStream.newInstance(inputData)).build(); | |
137 } | |
138 }); | |
139 benchmark("Deserialize from memory stream", inputData.length, new Action()
{ | |
140 public void execute() throws IOException { | |
141 defaultMessage.newBuilderForType() | |
142 .mergeFrom(CodedInputStream.newInstance(inputStream)).build(); | |
143 inputStream.reset(); | |
144 } | |
145 }); | |
146 System.out.println(); | |
147 return true; | |
148 } catch (Exception e) { | |
149 System.err.println("Error: " + e.getMessage()); | |
150 System.err.println("Detailed exception information:"); | |
151 e.printStackTrace(System.err); | |
152 return false; | |
153 } | |
154 } | |
155 | |
156 private static void benchmark(String name, long dataSize, Action action) throw
s IOException { | |
157 // Make sure it's JITted "reasonably" hard before running the first progress
test | |
158 for (int i=0; i < 100; i++) { | |
159 action.execute(); | |
160 } | |
161 | |
162 // Run it progressively more times until we've got a reasonable sample | |
163 int iterations = 1; | |
164 long elapsed = timeAction(action, iterations); | |
165 while (elapsed < MIN_SAMPLE_TIME_MS) { | |
166 iterations *= 2; | |
167 elapsed = timeAction(action, iterations); | |
168 } | |
169 | |
170 // Upscale the sample to the target time. Do this in floating point arithmet
ic | |
171 // to avoid overflow issues. | |
172 iterations = (int) ((TARGET_TIME_MS / (double) elapsed) * iterations); | |
173 elapsed = timeAction(action, iterations); | |
174 System.out.println(name + ": " + iterations + " iterations in " | |
175 + (elapsed/1000f) + "s; " | |
176 + (iterations * dataSize) / (elapsed * 1024 * 1024 / 1000f) | |
177 + "MB/s"); | |
178 } | |
179 | |
180 private static long timeAction(Action action, int iterations) throws IOExcepti
on { | |
181 System.gc(); | |
182 long start = System.currentTimeMillis(); | |
183 for (int i = 0; i < iterations; i++) { | |
184 action.execute(); | |
185 } | |
186 long end = System.currentTimeMillis(); | |
187 return end - start; | |
188 } | |
189 | |
190 private static byte[] readAllBytes(String filename) throws IOException { | |
191 RandomAccessFile file = new RandomAccessFile(new File(filename), "r"); | |
192 byte[] content = new byte[(int) file.length()]; | |
193 file.readFully(content); | |
194 return content; | |
195 } | |
196 | |
197 /** | |
198 * Interface used to capture a single action to benchmark. | |
199 */ | |
200 interface Action { | |
201 void execute() throws IOException; | |
202 } | |
203 } | |
OLD | NEW |