Benchmarking, dumping and munging

1 files changed, 90 insertions, 0 deletions

diff --git a/src/ProtoBench/Program.cs b/src/ProtoBench/Program.cs
new file mode 100644
index 00000000..9b65c0b1
--- /dev/null
+++ b/src/ProtoBench/Program.cs
@@ -0,0 +1,90 @@
+using System;
+using System.Diagnostics;
+using System.IO;
+using Google.ProtocolBuffers;
+
+namespace ProtoBench {
+  /// <summary>
+  /// Simple benchmarking of arbitrary messages.
+  /// </summary>
+  public sealed class Program {
+
+    private static readonly TimeSpan MinSampleTime = TimeSpan.FromSeconds(2);
+    private static readonly TimeSpan TargetTime = TimeSpan.FromSeconds(30);
+
+    // Avoid a .NET 3.5 dependency
+    delegate void Action();
+
+    public static int Main(string[] args) {
+      if (args.Length != 2) {
+        Console.Error.WriteLine("Usage: ProtoBecnh <descriptor type name> <input data>");
+        Console.Error.WriteLine("The descriptor type name is the fully-qualified message name,");
+        Console.Error.WriteLine("including assembly - e.g. Google.ProtocolBuffers.BenchmarkProtos.Message1,ProtoBench");
+        return 1;
+      }
+      IMessage defaultMessage;
+      try {
+        defaultMessage = MessageUtil.GetDefaultMessage(args[0]);
+      } catch (ArgumentException e) {
+        Console.Error.WriteLine(e.Message);
+        return 1;
+      }
+      try {
+        IBuilder builder = defaultMessage.WeakCreateBuilderForType();
+        byte[] inputData = File.ReadAllBytes(args[1]);
+        ByteString inputString = ByteString.CopyFrom(inputData);
+        IMessage sampleMessage = defaultMessage.WeakCreateBuilderForType().WeakMergeFrom(inputString).WeakBuild();
+        sampleMessage.ToByteString();
+        Console.WriteLine("Benchmarking {0} with file {1}", sampleMessage.GetType().Name, args[1]);
+        Benchmark("Serialize to byte string", inputData.Length, () => sampleMessage.ToByteString());
+        Benchmark("Serialize to byte array", inputData.Length, () => sampleMessage.ToByteArray());
+        Benchmark("Serialize to memory stream", inputData.Length, () => sampleMessage.WriteTo(new MemoryStream()));
+        Benchmark("Deserialize from byte array", inputData.Length, () =>
+          defaultMessage.WeakCreateBuilderForType()
+              .WeakMergeFrom(CodedInputStream.CreateInstance(inputData))
+              .WeakBuild()
+        );
+        Benchmark("Deserialize from byte array", inputData.Length, () =>
+          defaultMessage.WeakCreateBuilderForType()
+              .WeakMergeFrom(inputString)
+              .WeakBuild()
+        );
+        return 0;
+      } catch (Exception e) {
+        Console.Error.WriteLine("Error: {0}", e.Message);
+        Console.Error.WriteLine();
+        Console.Error.WriteLine("Detailed exception information: {0}", e);
+        return 1;
+      }
+    }
+
+    private static void Benchmark(string name, int dataSize, Action action) {
+      // Make sure it's JITted
+      action();
+      // Run it progressively more times until we've got a reasonable sample
+
+      int iterations = 1;
+      TimeSpan elapsed = TimeAction(action, iterations);
+      while (elapsed < MinSampleTime) {
+        iterations *= 2;
+        elapsed = TimeAction(action, iterations);
+      }
+      // Upscale the sample to the target time. Do this in floating point arithmetic
+      // to avoid overflow issues.
+      iterations = (int) ((TargetTime.Ticks / (double)elapsed.Ticks) * iterations);
+      elapsed = TimeAction(action, iterations);
+      Console.WriteLine("{0}: {1} iterations in {2:f3}s; {3:f3}MB/s",
+        name, iterations, elapsed.TotalSeconds,
+        (iterations * dataSize) / (elapsed.TotalSeconds * 1024 * 1024));
+    }
+
+    private static TimeSpan TimeAction(Action action, int iterations) {
+      Stopwatch sw = Stopwatch.StartNew();
+      for (int i = 0; i < iterations; i++) {
+        action();
+      }
+      sw.Stop();
+      return sw.Elapsed;
+    }
+  }
+}