src/Npgsql/Internal/Converters/Geometric/PathConverter.cs - test-logs - dev-npgsql/npgsql

src/Npgsql/Internal/Converters/Geometric/PathConverter.cs
68 строк3 KB
Nino Floris
Introduce bind operation on PgConverter surface (#6566)
11 май 2026, 20:01
11 май 2026, 20:010f4cfd8
using System.Diagnostics;
using System.Threading;
using System.Threading.Tasks;
using NpgsqlTypes;

// ReSharper disable once CheckNamespace
namespace Npgsql.Internal.Converters;

sealed class PathConverter : PgStreamingConverter<NpgsqlPath>
{
    public override NpgsqlPath Read(PgReader reader)
        => Read(async: false, reader, CancellationToken.None).GetAwaiter().GetResult();

    public override ValueTask<NpgsqlPath> ReadAsync(PgReader reader, CancellationToken cancellationToken = default)
        => Read(async: true, reader, cancellationToken);

    async ValueTask<NpgsqlPath> Read(bool async, PgReader reader, CancellationToken cancellationToken)
    {
        if (reader.ShouldBuffer(sizeof(byte) + sizeof(int)))
            await reader.Buffer(async, sizeof(byte) + sizeof(int), cancellationToken).ConfigureAwait(false);

        var open = reader.ReadByte() switch
        {
            1 => false,
            0 => true,
            _ => throw new UnreachableException("Error decoding binary geometric path: bad open byte")
        };

        var numPoints = reader.ReadInt32();
        var result = new NpgsqlPath(numPoints, open);

        for (var i = 0; i < numPoints; i++)
        {
            if (reader.ShouldBuffer(sizeof(double) * 2))
                await reader.Buffer(async, sizeof(double) * 2, cancellationToken).ConfigureAwait(false);

            result.Add(new NpgsqlPoint(reader.ReadDouble(), reader.ReadDouble()));
        }

        return result;
    }

    protected override Size BindValue(in BindContext context, NpgsqlPath value, ref object? writeState)
        => 5 + value.Count * sizeof(double) * 2;

    public override void Write(PgWriter writer, NpgsqlPath value)
        => Write(async: false, writer, value, CancellationToken.None).GetAwaiter().GetResult();

    public override ValueTask WriteAsync(PgWriter writer, NpgsqlPath value, CancellationToken cancellationToken = default)
        => Write(async: true, writer, value, cancellationToken);

    async ValueTask Write(bool async, PgWriter writer, NpgsqlPath value, CancellationToken cancellationToken)
    {
        if (writer.ShouldFlush(sizeof(byte) + sizeof(int)))
            await writer.Flush(async, cancellationToken).ConfigureAwait(false);

        writer.WriteByte((byte)(value.Open ? 0 : 1));
        writer.WriteInt32(value.Count);

        foreach (var p in value)
        {
            if (writer.ShouldFlush(sizeof(double) * 2))
                await writer.Flush(async, cancellationToken).ConfigureAwait(false);
            writer.WriteDouble(p.X);
            writer.WriteDouble(p.Y);
        }
    }
}