using BenchmarkDotNet.Attributes;

using Npgsql.Internal.TypeHandling;

using Npgsql.TypeMapping;

using NpgsqlTypes;

namespace Npgsql.Benchmarks;

[MemoryDiagnoser]

public class ResolveHandler

{

NpgsqlDataSource? _dataSource;

TypeMapper _typeMapper = null!;

[Params(0, 1, 2)]

public int NumPlugins { get; set; }

[GlobalSetup]

public void Setup()

{

var dataSourceBuilder = new NpgsqlDataSourceBuilder();

if (NumPlugins > 0)

dataSourceBuilder.UseNodaTime();

if (NumPlugins > 1)

dataSourceBuilder.UseNetTopologySuite();

_dataSource = dataSourceBuilder.Build();

_typeMapper = _dataSource.TypeMapper;

}

[GlobalCleanup]

public void Cleanup() => _dataSource?.Dispose();

[Benchmark]

public NpgsqlTypeHandler ResolveOID()

=> _typeMapper.ResolveByOID(23); // int4

[Benchmark]

public NpgsqlTypeHandler ResolveNpgsqlDbType()

=> _typeMapper.ResolveByNpgsqlDbType(NpgsqlDbType.Integer);

[Benchmark]

public NpgsqlTypeHandler ResolveDataTypeName()

=> _typeMapper.ResolveByDataTypeName("integer");

[Benchmark]

public NpgsqlTypeHandler ResolveClrTypeNonGeneric()

=> _typeMapper.ResolveByValue((object)8);

[Benchmark]

public NpgsqlTypeHandler ResolveClrTypeGeneric()

=> _typeMapper.ResolveByValue(8);

}