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

test/Npgsql.Benchmarks/WriteVaryingNumberOfParameters.cs

49 строк1 KB

Shay Rojansky

Convert codebase to use C# 8 nullability

05 май 2019, 21:55

05 май 2019, 21:55f2dd3f7

using System.Linq;
using BenchmarkDotNet.Attributes;
using NpgsqlTypes;

namespace Npgsql.Benchmarks.Types
{
    public class WriteVaryingNumberOfParameters
    {
        NpgsqlConnection _conn = default!;
        NpgsqlCommand _cmd = default!;

        [Params(10)]
        public int NumParams { get; set; }

        [GlobalSetup]
        public void Setup()
        {
            _conn = BenchmarkEnvironment.OpenConnection();

            var funcParams = string.Join(",",
                Enumerable.Range(0, NumParams)
                .Select(i => $"IN p{i} int4")
            );
            using (var cmd = new NpgsqlCommand($"CREATE FUNCTION pg_temp.swallow({funcParams}) RETURNS void AS 'BEGIN END;' LANGUAGE 'plpgsql'", _conn))
                cmd.ExecuteNonQuery();

            var cmdParams = string.Join(",", Enumerable.Range(0, NumParams).Select(i => $"@p{i}"));
            _cmd = new NpgsqlCommand($"SELECT pg_temp.swallow({cmdParams})", _conn);
            for (var i = 0; i < NumParams; i++)
                _cmd.Parameters.Add(new NpgsqlParameter("p" + i, NpgsqlDbType.Integer));
            _cmd.Prepare();
        }

        [GlobalCleanup]
        public void Cleanup()
        {
            _cmd.Unprepare();
            _conn.Close();
        }

        [Benchmark]
        public void WriteParameters()
        {
            for (var i = 0; i < NumParams; i++)
                _cmd.Parameters[i].Value = i;
            _cmd.ExecuteNonQuery();
        }
    }
}