110 lines
3.2 KiB
C#
Executable File
110 lines
3.2 KiB
C#
Executable File
using BenchmarkDotNet.Attributes;
|
|
using BenchmarkDotNet.Configs;
|
|
using BenchmarkDotNet.Jobs;
|
|
using ZB.MOM.WW.CBDD.Bson;
|
|
using ZB.MOM.WW.CBDD.Core;
|
|
using ZB.MOM.WW.CBDD.Core.Collections;
|
|
using ZB.MOM.WW.CBDD.Core.Storage;
|
|
using ZB.MOM.WW.CBDD.Core.Transactions;
|
|
using System.IO;
|
|
|
|
namespace ZB.MOM.WW.CBDD.Tests.Benchmark;
|
|
|
|
[SimpleJob]
|
|
[InProcess]
|
|
[MemoryDiagnoser]
|
|
[GroupBenchmarksBy(BenchmarkLogicalGroupRule.ByCategory)]
|
|
[HtmlExporter]
|
|
[JsonExporterAttribute.Full]
|
|
public class ReadBenchmarks
|
|
{
|
|
private const int DocCount = 1000;
|
|
|
|
private string _docDbPath = null!;
|
|
private string _docDbWalPath = null!;
|
|
|
|
private StorageEngine _storage = null!;
|
|
private BenchmarkTransactionHolder _transactionHolder = null!;
|
|
private DocumentCollection<Person> _collection = null!;
|
|
|
|
private ObjectId[] _ids = null!;
|
|
private ObjectId _targetId;
|
|
|
|
[GlobalSetup]
|
|
public void Setup()
|
|
{
|
|
var temp = AppContext.BaseDirectory;
|
|
var id = Guid.NewGuid().ToString("N");
|
|
_docDbPath = Path.Combine(temp, $"bench_read_docdb_{id}.db");
|
|
_docDbWalPath = Path.ChangeExtension(_docDbPath, ".wal");
|
|
|
|
if (File.Exists(_docDbPath)) File.Delete(_docDbPath);
|
|
if (File.Exists(_docDbWalPath)) File.Delete(_docDbWalPath);
|
|
|
|
_storage = new StorageEngine(_docDbPath, PageFileConfig.Default);
|
|
_transactionHolder = new BenchmarkTransactionHolder(_storage);
|
|
_collection = new DocumentCollection<Person>(_storage, _transactionHolder, new PersonMapper());
|
|
|
|
_ids = new ObjectId[DocCount];
|
|
for (int i = 0; i < DocCount; i++)
|
|
{
|
|
var p = CreatePerson(i);
|
|
_ids[i] = _collection.Insert(p);
|
|
}
|
|
_transactionHolder.CommitAndReset();
|
|
|
|
_targetId = _ids[DocCount / 2];
|
|
}
|
|
|
|
[GlobalCleanup]
|
|
public void Cleanup()
|
|
{
|
|
_transactionHolder?.Dispose();
|
|
_storage?.Dispose();
|
|
|
|
if (File.Exists(_docDbPath)) File.Delete(_docDbPath);
|
|
if (File.Exists(_docDbWalPath)) File.Delete(_docDbWalPath);
|
|
}
|
|
|
|
private Person CreatePerson(int i)
|
|
{
|
|
var p = new Person
|
|
{
|
|
Id = ObjectId.NewObjectId(),
|
|
FirstName = $"First_{i}",
|
|
LastName = $"Last_{i}",
|
|
Age = 20 + (i % 50),
|
|
Bio = null,
|
|
CreatedAt = DateTime.UtcNow,
|
|
Balance = 1000.50m * (i + 1),
|
|
HomeAddress = new Address
|
|
{
|
|
Street = $"{i} Main St",
|
|
City = "Tech City",
|
|
ZipCode = "12345"
|
|
}
|
|
};
|
|
|
|
// Add 10 work history items
|
|
for (int j = 0; j < 10; j++)
|
|
{
|
|
p.EmploymentHistory.Add(new WorkHistory
|
|
{
|
|
CompanyName = $"TechCorp_{i}_{j}",
|
|
Title = "Developer",
|
|
DurationYears = j,
|
|
Tags = new List<string> { "C#", "BSON", "Performance", "Database", "Complex" }
|
|
});
|
|
}
|
|
|
|
return p;
|
|
}
|
|
|
|
[Benchmark(Baseline = true, Description = "CBDD FindById")]
|
|
[BenchmarkCategory("Read_Single")]
|
|
public Person? DocumentDb_FindById()
|
|
{
|
|
return _collection.FindById(_targetId);
|
|
}
|
|
}
|