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

test/Npgsql.Benchmarks/CopyImport.cs

43 строки1 KB

Shay Rojansky

Convert codebase to use C# 8 nullability

05 май 2019, 21:55

05 май 2019, 21:55f2dd3f7

using BenchmarkDotNet.Attributes;
using NpgsqlTypes;

namespace Npgsql.Benchmarks
{
    public class CopyImport
    {
        NpgsqlConnection _conn = default!;
        NpgsqlCommand _truncateCmd = default!;
        const int Rows = 1000;

        [GlobalSetup]
        public void Setup()
        {
            _conn = BenchmarkEnvironment.OpenConnection();
            using (var cmd = new NpgsqlCommand("CREATE TEMP TABLE data (i1 INT, i2 INT, i3 INT, i4 INT, i5 INT, i6 INT, i7 INT, i8 INT, i9 INT, i10 INT)", _conn))
                cmd.ExecuteNonQuery();

            _truncateCmd = new NpgsqlCommand("TRUNCATE data", _conn);
            _truncateCmd.Prepare();
        }

        [GlobalCleanup]
        public void Cleanup() => _conn.Dispose();

        [IterationCleanup]
        public void IterationCleanup() => _truncateCmd.ExecuteNonQuery();

        [Benchmark]
        public void Import()
        {
            using (var importer = _conn.BeginBinaryImport("COPY data FROM STDIN (FORMAT BINARY)"))
            {
                for (var row = 0; row < Rows; row++)
                {
                    importer.StartRow();
                    for (var col = 0; col < 10; col++)
                        importer.Write(col, NpgsqlDbType.Integer);
                }
            }
        }
    }
}