Use oakton to issue different commands for benchmark project (relates…

… to #9)

src/Benchmark/Benchmark.csproj

@@ -8,6 +8,7 @@
 
   <ItemGroup>
     <PackageReference Include="BenchmarkDotNet" Version="0.12.0" />
+    <PackageReference Include="Oakton" Version="2.0.3" />
   </ItemGroup>
 
   <ItemGroup>

src/Benchmark/Commands/BenchCommand.cs

@@ -0,0 +1,14 @@
+using BenchmarkDotNet.Running;
+using Oakton;
+
+namespace Benchmark.Commands
+{
+    public class BenchCommand : OaktonCommand<BenchInput>
+    {
+        public override bool Execute(BenchInput input)
+        {
+            BenchmarkRunner.Run<PingPongBenchmark>();
+            return true;
+        }
+    }
+}

src/Benchmark/Commands/BenchInput.cs

@@ -0,0 +1,6 @@
+namespace Benchmark.Commands
+{
+    public class BenchInput
+    {
+    }
+}

src/Benchmark/Commands/ProfileCommand.cs

@@ -0,0 +1,31 @@
+using System;
+using Oakton;
+
+namespace Benchmark.Commands
+{
+    public class ProfileCommand : OaktonCommand<ProfileInput>
+    {
+        public override bool Execute(ProfileInput input)
+        {
+            var benchmark = new PingPongBenchmark();
+            try
+            {
+                Console.WriteLine("Starting setup");
+                benchmark.GlobalSetup();
+                benchmark.IterationSetup();
+
+                Console.WriteLine("Starting profiling");
+                benchmark.Run_ping_pong_benchmark();
+
+                Console.WriteLine("Press any key to exit");
+                Console.ReadKey();
+            }
+            finally
+            {
+                benchmark.GlobalCleanup();
+            }
+
+            return true;
+        }
+    }
+}

src/Benchmark/Commands/ProfileInput.cs

@@ -0,0 +1,7 @@
+namespace Benchmark.Commands
+{
+    public class ProfileInput
+    {
+
+    }
+}

src/Benchmark/Program.cs

@@ -1,9 +1,21 @@
+using System.Reflection;
 using BenchmarkDotNet.Running;
+using Oakton;
 
 namespace Benchmark
 {
     internal class Program
     {
-        private static void Main() => BenchmarkRunner.Run<PingPongBenchmark>();
+        private static int Main(string[] args)
+        {
+            var executor = CommandExecutor.For(_ =>
+            {
+                // Find and apply all command classes discovered
+                // in this assembly
+                _.RegisterCommands(typeof(Program).GetTypeInfo().Assembly);
+            });
+
+            return executor.Execute(args);
+        }
     }
 }