|
13 | 13 | * limitations under the License.
|
14 | 14 | */
|
15 | 15 |
|
| 16 | +using System.CommandLine; |
| 17 | +using System.Linq; |
16 | 18 | using BenchmarkDotNet.Configs;
|
17 | 19 | using BenchmarkDotNet.Running;
|
18 | 20 | using MongoDB.Benchmarks.Exporters;
|
19 |
| -using NDesk.Options; |
20 | 21 |
|
21 | 22 | namespace MongoDB.Benchmarks
|
22 | 23 | {
|
23 | 24 | public class BenchmarkRunner
|
24 | 25 | {
|
25 |
| - public static void Main(string[] args) |
| 26 | + public static int Main(string[] args) |
26 | 27 | {
|
27 |
| - var executingDriverBenchmarks = false; |
28 |
| - var exportingToEvergreen = false; |
29 |
| - var evergreenOutputFile = "evergreen-results.json"; // default output file name |
| 28 | + var rootCommand = new RootCommand("CSharp Driver benchmarks runner"); |
| 29 | + rootCommand.TreatUnmatchedTokensAsErrors = false; |
| 30 | + var evergreenOption = new Option<bool>("--evergreen", () => false); |
| 31 | + rootCommand.AddOption(evergreenOption); |
| 32 | + var driverBenchmarksOption = new Option<bool>("--driverBenchmarks", () => false); |
| 33 | + rootCommand.AddOption(driverBenchmarksOption); |
| 34 | + var evergreenOutputFileOption = new Option<string>(["--o", "--output-file"], () => "evergreen-results.json"); |
| 35 | + rootCommand.AddOption(evergreenOutputFileOption); |
30 | 36 |
|
31 |
| - var parser = new OptionSet |
| 37 | + rootCommand.SetHandler(invocationContext => |
32 | 38 | {
|
33 |
| - { "evergreen", v => exportingToEvergreen = v != null }, |
34 |
| - { "driverBenchmarks", v => executingDriverBenchmarks = v != null }, |
35 |
| - { "o|output-file=", v => evergreenOutputFile = v } |
36 |
| - }; |
| 39 | + var evergreenValue = invocationContext.ParseResult.GetValueForOption(evergreenOption); |
| 40 | + var driverBenchmarksValue = invocationContext.ParseResult.GetValueForOption(driverBenchmarksOption); |
| 41 | + var evergreenOutputFileValue = invocationContext.ParseResult.GetValueForOption(evergreenOutputFileOption); |
37 | 42 |
|
38 |
| - // the parser will try to parse the options defined above and will return any extra options |
39 |
| - var benchmarkSwitcherArgs = parser.Parse(args).ToArray(); |
| 43 | + var config = DefaultConfig.Instance; |
40 | 44 |
|
41 |
| - var config = DefaultConfig.Instance; |
| 45 | + // use a modified config if running driver benchmarks |
| 46 | + if (driverBenchmarksValue) |
| 47 | + { |
| 48 | + config = config |
| 49 | + .WithOption(ConfigOptions.JoinSummary, true) |
| 50 | + .AddExporter(new LocalExporter()) |
| 51 | + .HideColumns("BenchmarkDataSetSize"); |
| 52 | + } |
42 | 53 |
|
43 |
| - // use a modified config if running driver benchmarks |
44 |
| - if (executingDriverBenchmarks) |
45 |
| - { |
46 |
| - config = config |
47 |
| - .WithOption(ConfigOptions.JoinSummary, true) |
48 |
| - .AddExporter(new LocalExporter()) |
49 |
| - .HideColumns("BenchmarkDataSetSize"); |
50 |
| - } |
| 54 | + if (evergreenValue) |
| 55 | + { |
| 56 | + config = config.AddExporter(new EvergreenExporter(evergreenOutputFileValue)); |
| 57 | + } |
51 | 58 |
|
52 |
| - if (exportingToEvergreen) |
53 |
| - { |
54 |
| - config = config.AddExporter(new EvergreenExporter(evergreenOutputFile)); |
55 |
| - } |
| 59 | + BenchmarkSwitcher.FromAssembly(typeof(BenchmarkRunner).Assembly).Run(invocationContext.ParseResult.UnmatchedTokens.ToArray(), config); |
| 60 | + }); |
56 | 61 |
|
57 |
| - BenchmarkSwitcher.FromAssembly(typeof(BenchmarkRunner).Assembly).Run(benchmarkSwitcherArgs, config); |
| 62 | + return rootCommand.Invoke(args); |
58 | 63 | }
|
59 | 64 | }
|
60 | 65 | }
|
0 commit comments