BsonEncodingBenchmark.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.
 */

using System.Collections.Generic;
using System.IO;
using BenchmarkDotNet.Attributes;
using MongoDB.Bson;
using MongoDB.Bson.IO;
using MongoDB.Bson.Serialization;
using MongoDB.Bson.Serialization.Serializers;
using static MongoDB.Benchmarks.BenchmarkHelper;

namespace MongoDB.Benchmarks.Bson;

[IterationTime(3000)]
[BenchmarkCategory(DriverBenchmarkCategory.BsonBench)]
public class BsonEncodingBenchmark
{
    private const int Iterations = 10_000;

    private BsonSerializationContext _context;
    private BsonDocument _document;
    private object _documentPoco;
    private MemoryStream _stream;
    private BsonBinaryWriter _writer;
    private IBsonSerializer _pocoSerializer;

    [ParamsSource(nameof(BenchmarkDataSources))]
    public BsonBenchmarkData BenchmarkData { get; set; }

    [GlobalSetup]
    public void Setup()
    {
        _stream = new MemoryStream();
        _writer = new BsonBinaryWriter(_stream);
        _context = BsonSerializationContext.CreateRoot(_writer);
        _document = ReadExtendedJson(BenchmarkData.FilePath);
        _documentPoco =  BsonSerializer.Deserialize(_document, BenchmarkData.PocoType);
        _pocoSerializer = BsonSerializer.LookupSerializer(BenchmarkData.PocoType);
    }

    [Benchmark]
    public void BsonEncoding()
    {
        for (int i = 0; i < Iterations; i++)
        {
            BsonDocumentSerializer.Instance.Serialize(_context, _document);
            _stream.Position = 0;
        }
    }

    [Benchmark]
    public void BsonEncodingPoco()
    {
        for (int i = 0; i < Iterations; i++)
        {
            _pocoSerializer.Serialize(_context, _documentPoco);
            _stream.Position = 0;
        }
    }

    [GlobalCleanup]
    public void Cleanup()
    {
        _writer.Dispose();
        _stream.Dispose();
    }

    public IEnumerable<BsonBenchmarkData> BenchmarkDataSources() =>
    [
        new("extended_bson/flat_bson.json", "Flat", 75_310_000, typeof(FlatPoco)),
        new("extended_bson/full_bson.json", "Full", 57_340_000, typeof(FullPoco)),
        new("extended_bson/deep_bson.json", "Deep", 22_840_000, typeof(DeepPocoRoot))
    ];
}