BenchmarkRunner.cs

/* Copyright 2010-present MongoDB Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

usingSystem.CommandLine;
usingSystem.Linq;
usingBenchmarkDotNet.Configs;
usingBenchmarkDotNet.Running;
usingMongoDB.Benchmarks.Exporters;

namespaceMongoDB.Benchmarks
{
publicclassBenchmarkRunner
{
publicstaticintMain(string[]args)
{
varrootCommand=newRootCommand("CSharp Driver benchmarks runner");
rootCommand.TreatUnmatchedTokensAsErrors=false;
varevergreenOption=newOption<bool>("--evergreen",()=>false);
rootCommand.AddOption(evergreenOption);
vardriverBenchmarksOption=newOption<bool>("--driverBenchmarks",()=>false);
rootCommand.AddOption(driverBenchmarksOption);
varevergreenOutputFileOption=newOption<string>(["--o","--output-file"],()=>"evergreen-results.json");
rootCommand.AddOption(evergreenOutputFileOption);

rootCommand.SetHandler(invocationContext =>
{
varevergreenValue=invocationContext.ParseResult.GetValueForOption(evergreenOption);
vardriverBenchmarksValue=invocationContext.ParseResult.GetValueForOption(driverBenchmarksOption);
varevergreenOutputFileValue=invocationContext.ParseResult.GetValueForOption(evergreenOutputFileOption);

varconfig=DefaultConfig.Instance;

// use a modified config if running driver benchmarks
if(driverBenchmarksValue)
{
config=config
.WithOption(ConfigOptions.JoinSummary,true)
.AddExporter(newLocalExporter())
.HideColumns("BenchmarkDataSetSize");
}

if(evergreenValue)
{
config=config.AddExporter(newEvergreenExporter(evergreenOutputFileValue));
}

BenchmarkSwitcher.FromAssembly(typeof(BenchmarkRunner).Assembly).Run(invocationContext.ParseResult.UnmatchedTokens.ToArray(),config);
});

returnrootCommand.Invoke(args);
}
}
}