BenchmarkHelper.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;
usingSystem.Collections.Concurrent;
usingSystem.Collections.Generic;
usingSystem.IO;
usingSystem.Linq;
usingMongoDB.Bson;
usingMongoDB.Driver;

namespaceMongoDB.Benchmarks
{
publicstaticclassBenchmarkHelper
{
publicconststringDataFolderPath="../../../../../../../data/";

publicstaticvoidAddFilesToQueue(ConcurrentQueue<(string,int)>filesQueue,stringdirectoryPath,stringfileNamePrefix,intfileCount)
{
varaddingLDJSONfiles=fileNamePrefix=="ldjson";
for(inti=0;i<fileCount;i++)
{
varfileName=addingLDJSONfiles?$"{fileNamePrefix}{i:D3}.txt":$"{fileNamePrefix}{i:D2}.txt";
filesQueue.Enqueue(($"{directoryPath}/{fileName}",i));// enqueue complete filepath and filenumber
}
}

publicstaticdoubleCalculateCompositeScore(IEnumerable<BenchmarkResult>benchmarkResults,stringbenchmarkCategory)
{
varidentifiedBenchmarksScores=benchmarkResults
.Where(benchmark =>benchmark.Categories.Contains(benchmarkCategory))
.Select(benchmark =>benchmark.Score).ToArray();

if(identifiedBenchmarksScores.Any())
{
returnidentifiedBenchmarksScores.Average();
}

return0;
}

publicstaticvoidCreateEmptyDirectory(stringpath)
{
if(Directory.Exists(path))
{
Directory.Delete(path,true);
}
Directory.CreateDirectory(path);
}

publicstaticBsonDocumentReadExtendedJson(stringresourcePath)
{
varextendedJson=File.ReadAllText(DataFolderPath+resourcePath);
returnBsonDocument.Parse(extendedJson);
}

publicstaticbyte[]ReadExtendedJsonToBytes(stringresourcePath)
{
varextendedJson=File.ReadAllText(DataFolderPath+resourcePath);
vardocument=BsonDocument.Parse(extendedJson);
returndocument.ToBson();
}

publicstaticclassMongoConfiguration
{
publicconststringPerfTestDatabaseName="perftest";
publicconststringPerfTestCollectionName="corpus";

publicstaticIMongoClientCreateClient()
{
varmongoUri=Environment.GetEnvironmentVariable("MONGODB_URI");
varsettings=mongoUri!=null?MongoClientSettings.FromConnectionString(mongoUri):new();
settings.ClusterSource=DisposingClusterSource.Instance;

varclient=newMongoClient(settings);
client.DropDatabase(PerfTestDatabaseName);

returnclient;
}
}
}
}