dev-npgsql/npgsql: Npgsql is the .NET data provider for PostgreSQL.

test/Npgsql.Benchmarks/ReadArray.cs
106 строк3 KB
Yoh Deadfall
Optimized ArrayHandler
03 май 2018, 00:45
03 май 2018, 00:4583347e8
using BenchmarkDotNet.Attributes;
using BenchmarkDotNet.Columns;
using BenchmarkDotNet.Configs;
using BenchmarkDotNet.Diagnosers;
using NpgsqlTypes;
using System.Collections.Generic;
using System.Linq;
using System.Net;
using System.Runtime.CompilerServices;

namespace Npgsql.Benchmarks
{
    [Config(typeof(ReadArrayConfig))]
    public class ReadArray
    {
        NpgsqlConnection _conn;
        NpgsqlCommand _cmd;
        NpgsqlDataReader _reader;

        [GlobalSetup(Target = nameof(ReadIntArray) + "," + nameof(ReadListOfInt))]
        public void GlobalSetupForInt()
            => GlobalSetupImpl(42);

        [GlobalSetup(Target = nameof(ReadStringArray) + "," + nameof(ReadListOfString))]
        public void GlobalSetupForString()
            => GlobalSetupImpl("The Answer to the Ultimate Question of Life, The Universe, and Everything.");

        [GlobalSetup(Target = nameof(ReadIPAddressArray)
                              + "," + nameof(ReadNpgsqlInetArray)
                              + "," + nameof(ReadListOfIPAddress)
                              + "," + nameof(ReadListOfNpgsqlInet))]
        public void GlobalSetupForInet()
            => GlobalSetupImpl(IPAddress.Loopback);

        void GlobalSetupImpl<T>(T initializationValue)
        {
            _conn = BenchmarkEnvironment.OpenConnection();
            _cmd = new NpgsqlCommand("SELECT @p1;", _conn);
            _cmd.Parameters.AddWithValue("@p1", Enumerable.Repeat(initializationValue, NumArrayElements).ToArray());
            _reader = _cmd.ExecuteReader();
            _reader.Read();
        }

        [Params(0, 10, 1000, 100000)]
        public int NumArrayElements { get; set; }

        [GlobalCleanup]
        public void Cleanup()
        {
            _reader.Dispose();
            _cmd.Dispose();
            _conn.Dispose();
        }

        [Benchmark]
        public void ReadIntArray()
            => ReadArrayImpl<int>();

        [Benchmark]
        public void ReadStringArray()
            => ReadArrayImpl<string>();

        [Benchmark]
        // ReSharper disable once InconsistentNaming
        public void ReadIPAddressArray()
            => ReadArrayImpl<IPAddress>();

        [Benchmark]
        public void ReadNpgsqlInetArray() // PSV for IPAddress
            => ReadArrayImpl<NpgsqlInet>();

        [Benchmark]
        public void ReadListOfInt()
            => ReadListImpl<int>();

        [Benchmark]
        public void ReadListOfString()
            => ReadListImpl<string>();

        [Benchmark]
        // ReSharper disable once InconsistentNaming
        public void ReadListOfIPAddress()
            => ReadListImpl<IPAddress>();

        [Benchmark]
        public void ReadListOfNpgsqlInet() // PSV for IPAddress
            => ReadListImpl<NpgsqlInet>();

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        void ReadArrayImpl<T>()
            => _reader.GetFieldValue<T[]>(0);

        [MethodImpl(MethodImplOptions.AggressiveInlining)]
        void ReadListImpl<T>()
            => _reader.GetFieldValue<List<T>>(0);

        class ReadArrayConfig : ManualConfig
        {
            public ReadArrayConfig()
            {
                Add(StatisticColumn.OperationsPerSecond);
                Add(MemoryDiagnoser.Default);
            }
        }
    }
}